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