def test_getters_return_expected_data(self):
instance = Location(self.__test_name, self.__test_lon, self.__test_lat,
self.__test_ID, self.__test_country)
self.assertEqual(instance.get_name(), self.__test_name)
self.assertEqual(instance.get_lon(), self.__test_lon)
self.assertEqual(instance.get_lat(), self.__test_lat)
self.assertEqual(instance.get_ID(), self.__test_ID)
self.assertEqual(instance.get_country(), self.__test_country)
def test_getters_return_expected_data(self):
instance = Location(self.__test_name, self.__test_lon, self.__test_lat,
self.__test_ID, self.__test_country)
self.assertEqual(instance.get_name(), self.__test_name)
self.assertEqual(instance.get_lon(), self.__test_lon)
self.assertEqual(instance.get_lat(), self.__test_lat)
self.assertEqual(instance.get_ID(), self.__test_ID)
self.assertEqual(instance.get_country(), self.__test_country)
def test_location_for(self):
expected = Location('Dongen', 4.938890, 51.626671, 2756723, 'NL')
result = self._instance.location_for('dongen')
self.assertEqual(result.get_name(), expected.get_name())
self.assertEqual(result.get_country(), expected.get_country())
self.assertEqual(result.get_ID(), expected.get_ID())
self.assertEqual(result.get_lat(), expected.get_lat())
self.assertEqual(result.get_lon(), expected.get_lon())
self.assertTrue(self._instance.location_for('aaaaaaaaaa') is None)
def test_location_for(self):
ref_to_original = CityIDRegistry._get_lines
CityIDRegistry._get_lines = self._mock_get_lines
expected = Location('dongdu', 117.699997, 35.849998, 1812597, 'CN')
result_1 = self._instance.location_for('dongdu')
result_2 = self._instance.location_for('aaaaaaaaaa')
CityIDRegistry._get_lines = ref_to_original
self.assertEqual(result_1.get_name(), expected.get_name())
self.assertEqual(result_1.get_country(), expected.get_country())
self.assertEqual(result_1.get_ID(), expected.get_ID())
self.assertEqual(result_1.get_lat(), expected.get_lat())
self.assertEqual(result_1.get_lon(), expected.get_lon())
self.assertTrue(result_2 is None)
def test_location_for(self):
ref_to_original = CityIDRegistry._get_lines
CityIDRegistry._get_lines = self._mock_get_lines
expected = Location('dongdu', 117.699997, 35.849998, 1812597, 'CN')
result_1 = self._instance.location_for('dongdu')
result_2 = self._instance.location_for('aaaaaaaaaa')
CityIDRegistry._get_lines = ref_to_original
self.assertEqual(result_1.get_name(), expected.get_name())
self.assertEqual(result_1.get_country(), expected.get_country())
self.assertEqual(result_1.get_ID(), expected.get_ID())
self.assertEqual(result_1.get_lat(), expected.get_lat())
self.assertEqual(result_1.get_lon(), expected.get_lon())
self.assertTrue(result_2 is None)