def test_one_phase_with_few_records(self, jhu_data, population_data):
population = population_data.value("Italy")
sr_df = jhu_data.to_sr(country="Italy",
population=population,
start_date="25Mar2020",
end_date="26Mar2020")
with pytest.raises(ValueError):
trend = Trend(sr_df)
trend.run()
def test_one_phase(self, jhu_data, population_data):
population = population_data.value("Italy")
sr_df = jhu_data.to_sr(country="Italy",
population=population,
start_date="25Mar2020",
end_date="02Apr2020")
trend = Trend(sr_df)
trend.run()
assert isinstance(trend.rmsle(), float)
warnings.simplefilter("ignore", category=UserWarning)
trend.show(area=jhu_data.area_name(country="Italy"))