Minor test changes for get_filings

dgunning · Oct 4, 2023 · 5a94cb7 · 5a94cb7
1 parent e34939b
commit 5a94cb7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/test_company.py b/tests/test_company.py
@@ -1,7 +1,7 @@
 import json
 from functools import lru_cache
 from pathlib import Path
-
+import datetime
 import humanize
 import pyarrow.compute as pc
 from rich import print
@@ -367,6 +367,9 @@ def test_search_for_company():
 
 
 def test_search():
+    is_start_of_quarter = datetime.datetime.now().month in [1, 4, 7, 10] and datetime.datetime.now().day <= 3
+    if is_start_of_quarter:
+        return
     text_index = CompanySearchIndex(get_filings()
                                     .to_pandas('company', 'cik'))
     res = text_index.similar('Tesla')

diff --git a/tests/test_filing.py b/tests/test_filing.py
@@ -600,8 +600,10 @@ def test_get_filings_for_future_period(capsys):
 
 
 def test_get_filings_default():
+
     filings = get_filings()
     assert not filings.empty
+
     filings = get_filings(form="8-K")
     assert not filings.empty
     print()