Update test_process_data.py
Browse files- tests/test_process_data.py +15 -1
tests/test_process_data.py
CHANGED
|
@@ -1,9 +1,23 @@
|
|
| 1 |
import pytest
|
|
|
|
|
|
|
| 2 |
|
| 3 |
from modules.get_rules_in_window import (
|
| 4 |
format_documents,
|
| 5 |
)
|
| 6 |
|
| 7 |
|
| 8 |
-
|
|
|
|
| 9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import pytest
|
| 2 |
+
from fr_toolbelt.api_requests import get_documents_by_date
|
| 3 |
+
from pandas import DataFrame
|
| 4 |
|
| 5 |
from modules.get_rules_in_window import (
|
| 6 |
format_documents,
|
| 7 |
)
|
| 8 |
|
| 9 |
|
| 10 |
+
START = "2024-03-01"
|
| 11 |
+
END = "2024-04-30"
|
| 12 |
|
| 13 |
+
|
| 14 |
+
#@pytest.fixture
|
| 15 |
+
def temp_documents():
|
| 16 |
+
return get_documents_by_date(START, end_date=END, document_types=("RULE", ))[0]
|
| 17 |
+
TEMP_DOCUMENTS = temp_documents()
|
| 18 |
+
|
| 19 |
+
|
| 20 |
+
def test_format_documents():
|
| 21 |
+
|
| 22 |
+
formatted = format_documents(TEMP_DOCUMENTS)
|
| 23 |
+
assert isinstance(formatted, DataFrame)
|