def test_too_many_coords(self):
'''
Test the distance function with nonsensical inputs
'''
with self.assertRaises(ValueError):
great_circle_distance(
# Not a real coordinate
(1, 2, 3),
(4, 5, 6),
)
def test_not_floats(self):
'''
Test the distance function with nonsensical inputs
'''
with self.assertRaises(ValueError):
great_circle_distance(
# Almost looks like a coordinate...
(42.424242, '42.424242'),
('7.77778', 6.66),
)
def test_distance_between_two_points(self):
'''
Test the distance between the Spire in Dublin and Empire State Building
in New York City is 5111.02 km (Google maps says it is...)
'''
d = great_circle_distance(
(53.349995, -6.260223), # Dublin Spire
(40.748652, -73.985640), # Empire state Building
)
d = float("{0:.2f}".format(d)) # round to 2 places for asserting
assert d == 5111.02
def filter_for_nearby_customers(destination, customers, max_dist):
nearby_customers = []
for customer in customers:
dist = great_circle_distance(
customer.coordinates,
destination.coordinates
)
if dist <= max_dist:
nearby_customers.append(customer)
sorted_customers = sorted(nearby_customers, key=lambda c: c.uid)
return sorted_customers