def test_one_phase(self, jhu_data, population_data): population = population_data.value("Italy") trend = Trend(jhu_data, population, "Italy", start_date="25Mar2020", end_date="02Apr2020") trend.analyse() assert isinstance(trend.rmsle(), float) trend.show()
def test_one_phase_with_dataframe(self, jhu_data, population_data): population = population_data.value("Italy") clean_df = jhu_data.cleaned() warnings.filterwarnings("ignore", category=DeprecationWarning) trend = Trend(clean_df, population, "Italy", start_date="25Mar2020", end_date="02Apr2020") trend.analyse() assert isinstance(trend.rmsle(), float) trend.show()