def test_distance_to(self):
test_location = Location(50.0, 50.0, LocationType.Plant, 100000)
test_target_location1 = Location(50.0, 55.0, LocationType.Port)
test_target_location2 = Location(55.0, 50.0, LocationType.Port)
test_target_location3 = Location(49.0, 51.0, LocationType.Port)
self.assertEqual(test_location.distance_to(test_target_location1), 5)
self.assertEqual(test_location.distance_to(test_target_location2), 5)
self.assertEqual(test_location.distance_to(test_target_location3),
sqrt(2))
def test_distance_to_fail(self):
test_location = Location(50.0, 50.0, LocationType.Plant, 100000)
with self.assertRaises(TypeError):
test_location.distance_to(1)
