def test_start_record_range(self, jhu_data, population_data, country):
# Setting
dhl = DataHandler(jhu_data, population_data, country)
dhl.init_records()
# Test
dhl.first_date = "01Apr2020"
assert dhl.first_date == "01Apr2020"
dhl.last_date = "01May2020"
assert dhl.last_date == "01May2020"
with pytest.raises(ValueError):
dhl.first_date = "01Jan2019"
with pytest.raises(ValueError):
tomorrow = Term.tomorrow(datetime.now().strftime(Term.DATE_FORMAT))
dhl.last_date = tomorrow
def test_show_complement(self, jhu_data, population_data, country):
# Setting
dhl = DataHandler(jhu_data, population_data, country)
dhl.init_records()
dhl.first_date = "01Apr2020"
dhl.last_date = "01Aug2020"
# Test
dhl_df = dhl.show_complement()
data_df = jhu_data.show_complement(
country=country, start_date="01Apr2020", end_date="01Aug2020")
assert dhl_df.equals(data_df)
def test_records(self, jhu_data, population_data, country):
warnings.simplefilter("ignore", category=UserWarning)
# Setting
dhl = DataHandler(jhu_data, population_data, country)
dhl.init_records()
dhl.first_date = "01Apr2020"
dhl.last_date = "01Aug2020"
# Test
df = dhl.records(variables=None, show_figure=False)
assert isinstance(df, pd.DataFrame)
assert set(df.columns) == set([Term.DATE, Term.CI, Term.F, Term.R])
dates = df[Term.DATE]
assert dates.min() == Term.date_obj(dhl.first_date)
assert dates.max() == Term.date_obj(dhl.last_date)
df2 = dhl.records(variables=["Susceptible"], show_figure=True)
assert set(df2.columns) == set([Term.DATE, Term.S])