def test_daily_query_data():
daily_report = parseDataDailyReport.DailyReportData()
daily_report.readData(
os.path.join(os.getcwd(), "tests", "test_csv", "test_daily"))
daily_report.queryData(["US"], ["South Carolina"],
["Abbeville, South Carolina, US"], "01/01/2020",
"09/01/2020", "confirmed")
check_equal_original = pd.DataFrame({
"Province_State": ["South Carolina"],
"Country_Region": ["US"],
"Last_Update": [datetime(2020, 6, 25, 0, 0, 0)],
"Confirmed": [91],
"Deaths": [0],
"Recovered": [0],
"Active": [91],
"Combined_Key": ["Abbeville, South Carolina, US"]
})
check_equal_parsed = pd.DataFrame({
"Province_State": ["South Carolina"],
"Country_Region": ["US"],
"Last_Update": [datetime(2020, 6, 25, 0, 0, 0)],
"Confirmed": [91],
"Combined_Key": ["Abbeville, South Carolina, US"]
})
assert daily_report.original_data.equals(check_equal_original)
assert daily_report.parsed_data.equals(check_equal_parsed)
def test_daily_export_txt():
daily_report = parseDataDailyReport.DailyReportData()
daily_report.readData(
os.path.join(os.getcwd(), "tests", "test_csv", "test_daily"))
daily_report.queryData(["US"], ["South Carolina"],
["Abbeville, South Carolina, US"], "01/01/2020",
"09/01/2020", "confirmed")
daily_report.exportTxt(os.path.join(os.getcwd(), "tests", "test_result"),
'')
assert os.path.isfile(
os.path.join(os.getcwd(), "tests", "test_result",
"txt_export_daily_report.html"))
def test_daily_initial_data():
daily_report = parseDataDailyReport.DailyReportData()
assert daily_report.original_data.empty
assert daily_report.parsed_data.empty