File size: 501 Bytes
58bb4c7 54c1271 a8baed5 54c1271 a8baed5 54c1271 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
#import pytest
from fr_toolbelt.api_requests import get_documents_by_date
from pandas import DataFrame
from modules.get_rules_in_window import (
format_documents,
)
START = "2024-03-01"
END = "2024-04-30"
#@pytest.fixture
def temp_documents():
return get_documents_by_date(START, end_date=END, document_types=("RULE", ))[0]
TEMP_DOCUMENTS = temp_documents()
def test_format_documents():
formatted = format_documents(TEMP_DOCUMENTS)
assert isinstance(formatted, DataFrame)
|