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")
