#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)