def test_multi_data_2(self): api_caller = ApiCaller() country_list = ["Puerto Rico", "Sweden"] expected_response = { 'Puerto Rico': { 1960: None }, 'Sweden': { 1960: 6.58 } } self.assertEqual( expected_response, api_caller.get_multiple_data_range(country_list, 'emissions_per_capita', 1940, 1960))
def test_multi_data_1(self): api_caller = ApiCaller() country_list = ["Aruba", "Finland"] expected_response = { 'Aruba': { 1985: None, 1986: 179.683, 1987: 447.374 }, 'Finland': { 1985: 49665.848, 1986: 53329.181, 1987: 57656.241 } } self.assertEqual( expected_response, api_caller.get_multiple_data_range(country_list, 'emissions', 1985, 1987))