def test_get_daily_cases(self): country = Country("TEST") with open("./tests/data/australiadata.json", "r") as filehandler: country.data = json.loads(filehandler.readline()) country.get_daily_cases() self.assertEqual(country.daily_cases, [9, 13, 3, 5])
def test_get_fourteen_day_average(self): country = Country("TEST") with open("./tests/data/fourteen_day_average_australia_data.json", "r") as filehandler: country.data = json.loads(filehandler.readline()) country.get_daily_cases() country.get_fourteen_day_average_cases() self.assertEqual( country.fourteen_day_average_cases, [7, 9, 7, 6, 10, 11, 11, 12, 13, 14, 13, 16, 15, 16, 17, 19, 21])