def test_get_daily_listing():
ed = edgar_index.edgar_index()
document = ed.get_daily_listing(datetime.date(2018, 1, 2))
rows = document.split("\n")
# get the middle row
row = rows[int(len(rows) / 2)].split("|")
assert len(row) == 5
def get_13f_listings(date: datetime.datetime, populate: bool) -> pd.DataFrame:
ed_i = edgar_index.edgar_index()
df = ed_i.get_full_listing_as_pd(date)
df = df[df["Form Type"].isin(["13F-HR", "13F-HR/A", "13F-NT",
"13F-NT/A"])].reset_index(drop=True)
df["File"] = np.nan
if populate:
for index, row in df.iterrows():
df["File"] = ed_i.get_filing(row["File Name"])
return df
def test_get_daily_index():
ed = edgar_index.edgar_index()
index_dict = ed._get_daily_index(datetime.date(2018, 1, 1))
assert len(index_dict) == 1
assert list(index_dict.keys()) == ["directory"]
assert list(
index_dict["directory"].keys()) == ["item", "name", "parent-dir"]
assert len(index_dict["directory"]["item"]) == 310
assert index_dict["directory"]["name"] == "daily-index/2018/QTR1/"
assert list(index_dict["directory"]["item"][0].keys()) == [
"last-modified", "name", "type", "href", "size"
]
def test_get_quarter():
ed = edgar_index.edgar_index()
assert ed._get_quarter(1) == 1
assert ed._get_quarter(2) == 1
assert ed._get_quarter(3) == 1
assert ed._get_quarter(4) == 2
assert ed._get_quarter(5) == 2
assert ed._get_quarter(6) == 2
assert ed._get_quarter(7) == 3
assert ed._get_quarter(8) == 3
assert ed._get_quarter(9) == 3
assert ed._get_quarter(10) == 4
assert ed._get_quarter(11) == 4
assert ed._get_quarter(12) == 4
assert ed._get_quarter(0) == 1
assert ed._get_quarter(13) == 4
def test_get_daily_index_url():
ed = edgar_index.edgar_index()
assert ed._get_daily_index_url(
datetime.date(2018, 1, 1)
) == "https://www.sec.gov/Archives/edgar/daily-index/2018/QTR1/index.json"
assert ed._get_daily_index_url(
datetime.date(2018, 8, 1)
) == "https://www.sec.gov/Archives/edgar/daily-index/2018/QTR3/index.json"
assert ed._get_daily_index_url(
datetime.date(2019, 2, 1)
) == "https://www.sec.gov/Archives/edgar/daily-index/2019/QTR1/index.json"
assert ed._get_daily_index_url(
datetime.date(2019, 4, 1)
) == "https://www.sec.gov/Archives/edgar/daily-index/2019/QTR2/index.json"
assert ed._get_daily_index_url(
datetime.date(2019, 7, 1)
) == "https://www.sec.gov/Archives/edgar/daily-index/2019/QTR3/index.json"
assert ed._get_daily_index_url(
datetime.date(2019, 12, 1)
) == "https://www.sec.gov/Archives/edgar/daily-index/2019/QTR4/index.json"
def test_get_filing():
ed = edgar_index.edgar_index()
raw_filing = ed.get_filing("edgar/data/1000097/0000919574-18-000008.txt")
assert len(raw_filing) > 0
def test_get_filing_list():
ed = edgar_index.edgar_index()
df = ed.get_filing_list([1000097], datetime.date(2018, 1, 1),
datetime.date(2018, 2, 1))
assert df.shape == (1, 6)
assert list(df["CIK"]) == [1000097]
def test_get_daily_listing_as_pd():
ed = edgar_index.edgar_index()
df = ed.get_daily_listing_as_pd(datetime.date(2018, 1, 2))
assert df.shape[1] == 6
def test_get_daily_listing_url():
ed = edgar_index.edgar_index()
assert ed._get_daily_listing_url(datetime.date(2018, 1, 2)) == (
"https://www.sec.gov/Archives/edgar/daily-index/2018/QTR1/master.20180102.idx",
True)
def test_get_full_listing_url():
ed = edgar_index.edgar_index()
assert ed._get_full_listing_url(datetime.date(2018, 1, 1)) == (
"https://www.sec.gov/Archives/edgar/full-index/2018/QTR1/master.idx",
True)