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