mfebrizio commited on
Commit
54c1271
·
1 Parent(s): a8baed5

Update test_process_data.py

Browse files
Files changed (1) hide show
  1. 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
- #pytest.fixture
 
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)