diff --git a/tests/test_company.py b/tests/test_company.py index 59680cad..8acb54dc 100644 --- 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 index 59824767..ccf69b8f 100644 --- 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()