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)