def test_daily_change_calculation(self): dummy_data = testutils.get_dummy_data() d = DataHandler() deaths = d.get_total_deaths_per_country_and_day(dummy_data) expected_df = testutils.get_dummy_change_data() actual = d.get_daily_change_of_deaths(deaths) actual.reset_index(drop=True, inplace=True) self.assertTrue(expected_df.equals(actual), "Function generated daily change in deaths must be indentical to the expected") deaths.loc[0, 'deaths'] = 1 actual = d.get_daily_change_of_deaths(deaths) self.assertEqual(actual.loc[0, 'deaths_change'], 1, "The first day's change in deaths should be 1") deaths.loc[0, 'deaths'] = 4 actual = d.get_daily_change_of_deaths(deaths) self.assertEqual(actual.loc[0, 'deaths_change'], 4, "The first day's change in deaths should be 4")