def test_client_passed_to_objects(self, mock_user_agent):
client = NetworkClient(user_agent=mock_user_agent)
company = filings(cik_lookup="aapl",
filing_type=FilingType.FILING_10Q,
client=client)
daily = filings(start_date=date(2021, 1, 1),
end_date=date(2021, 1, 1),
client=client)
quarterly = filings(start_date=date(2020, 1, 1),
end_date=date(2020, 3, 31),
client=client)
assert company.client == client
assert daily.client == client
assert quarterly.client == client
def test_exact_quarter_returns_quarterly_filings(self, mock_user_agent):
f = filings(start_date=date(2020, 1, 1),
end_date=date(2020, 3, 31),
user_agent=mock_user_agent)
assert isinstance(f, QuarterlyFilings)
assert f.year == 2020
assert f.quarter == 1
def test_mismatch_of_quarters_returns_combo_filings(self):
start_date = date(2020, 1, 1)
end_date = date(2020, 5, 1)
f = filings(start_date=start_date, end_date=end_date)
assert isinstance(f, ComboFilings)
assert f.start_date == start_date
assert f.end_date == end_date
def test_same_date_returns_daily_filings(self, mock_user_agent):
day = date(2020, 1, 1)
f = filings(start_date=day, end_date=day, user_agent=mock_user_agent)
assert isinstance(f, DailyFilings)
assert f.date == day
assert f.year == day.year
assert f.quarter == day.month // 3 + 1
def test_no_end_date_no_cik_lookp_returns_daily_filings(self):
day = date(2020, 1, 1)
f = filings(start_date=day, end_date=date(2020, 1, 1))
assert isinstance(f, DailyFilings)
assert f.date == day
assert f.year == day.year
assert f.quarter == day.month // 3 + 1
def test_bad_args_combination_raises_error(self, kwargs, error):
with pytest.raises(error):
filings(**kwargs)
def test_count_not_implemented(self, mock_user_agent):
with pytest.raises(NotImplementedError):
filings(start_date=date(2010, 1, 1),
end_date=date(2020, 1, 1),
user_agent=mock_user_agent,
count=10)
def test_user_agent_passed_to_objects(self, mock_user_agent):
f = filings(cik_lookup="aapl",
filing_type=FilingType.FILING_10Q,
user_agent=mock_user_agent)
assert f.client.user_agent == mock_user_agent
def test_bad_filing_type(self):
with pytest.raises(FilingTypeError):
filings(cik_lookup='aapl', filing_type='10-k')
