def test_get_filing_info_bad_form():
forms = ['4', '10-Z']
try:
get_filing_info(forms=forms, year=2018, quarter=4)
assert False
except InvalidInputException:
assert True
def test_get_filing_info_cik_filter():
cik = '1698878'
filing_infos = get_filing_info(cik=cik, year=2018, quarter=4)
assert filing_infos is not None
assert type(filing_infos) is list
for filing_info in filing_infos:
assert filing_info.cik == cik
def test_get_filing_info_forms_filter():
forms = ['4', '10-Q']
filing_infos = get_filing_info(forms=forms, year=2018, quarter=4)
assert filing_infos is not None
assert type(filing_infos) is list
for filing_info in filing_infos:
assert filing_info.form in forms
def test_get_filing_info():
filing_infos = get_filing_info(year=2018, quarter=4)
# print(filing_infos)
assert filing_infos is not None
assert type(filing_infos) is list
for filing_info in filing_infos:
validate_filing_info(filing_info)
def test_get_filing_info_bad_quarter():
try:
get_filing_info(year=2018, quarter=5)
assert False
except InvalidInputException:
assert True
try:
get_filing_info(year=2018, quarter=-1)
assert False
except InvalidInputException:
assert True
try:
get_filing_info(year=2018, quarter=0.25)
assert False
except InvalidInputException:
assert True
def test_get_filing_info_bad_year():
try:
get_filing_info(year=20181, quarter=4)
assert False
except InvalidInputException:
assert True
try:
get_filing_info(year=1992, quarter=4)
assert False
except InvalidInputException:
assert True
current_year = datetime.now().year
try:
get_filing_info(year=current_year + 1, quarter=4)
assert False
except InvalidInputException:
assert True