def test_sec_file_generator():
data = sec_file_generator(download_data_from_sec(construct_url(2009, 1)), 'sub.txt')
for row in data:
print(row)
def test_sec_file_generator_raises():
with ShouldRaise(KeyError("Invalid file name blah.txt, not in ['pre.txt', 'sub.txt', 'readme.htm', 'num.txt', 'tag.txt']")):
data = sec_file_generator(download_data_from_sec(construct_url(2009, 1)), 'blah.txt')
for row in data:
print(row)
