def test_build_descriptions():
param = copy(PARAM)
admits_file = 'tests/by_doubling_time/2020-03-28_projected_admits.csv'
census_file = 'tests/by_doubling_time/2020-03-28_projected_census.csv'
admits_df = pd.read_csv(admits_file, parse_dates=['date'])
chart = build_admits_chart(alt=alt, admits_df=admits_df)
description = build_descriptions(chart=chart, labels=param.labels)
hosp, icu, vent = description.split(
"\n\n") # break out the description into lines
max_hosp = chart.data['hospitalized'].max()
assert str(ceil(max_hosp)) in hosp
# TODO add test for asterisk
# test no asterisk
param.n_days = 600
admits_df = pd.read_csv(admits_file, parse_dates=['date'])
chart = build_admits_chart(alt=alt, admits_df=admits_df)
description = build_descriptions(chart=chart, labels=param.labels)
assert "*" not in description
# census chart
census_df = pd.read_csv(census_file, parse_dates=['date'])
chart = build_census_chart(alt=alt, census_df=census_df)
description = build_descriptions(chart=chart, labels=param.labels)
assert str(ceil(chart.data['ventilated'].max())) in description
assert str(chart.data['icu'].idxmax()) not in description
assert datetime.strftime(chart.data.iloc[chart.data['icu'].idxmax()].date,
'%b %d') in description
def test_admits_chart(admits_floor_df):
chart = build_admits_chart(alt=alt, admits_floor_df=admits_floor_df)
assert isinstance(chart, (alt.Chart, alt.LayerChart))
assert round(chart.data.iloc[40].icu, 0) == 38
# test fx call with no params
with pytest.raises(TypeError):
build_admits_chart()
def test_admits_chart():
admits_df = pd.read_csv(
"tests/by_doubling_time/2020-03-28_projected_admits.csv")
chart = build_admits_chart(alt=alt, admits_df=admits_df)
assert isinstance(chart, (alt.Chart, alt.LayerChart))
assert round(chart.data.iloc[40].icu, 0) == 38
# test fx call with no params
with pytest.raises(TypeError):
build_admits_chart()
def test_build_descriptions(admits_floor_df, param):
chart = build_admits_chart(alt=alt, admits_floor_df=admits_floor_df)
description = build_descriptions(chart=chart, labels=param.labels)
hosp, icu, vent = description.split("\n\n") # break out the description into lines
max_hosp = chart.data["hospitalized"].max()
assert str(ceil(max_hosp)) in hosp
def test_no_asterisk(admits_floor_df, param):
param.n_days = 600
chart = build_admits_chart(alt=alt, admits_floor_df=admits_floor_df)
description = build_descriptions(chart=chart, labels=param.labels)
assert "*" not in description
