def test_salesman_sort(self):
long_locations = []
long_locations.append(Location("A", 10, 10))
long_locations.append(Location("B", 10, 40))
short_locations = []
short_locations.append(Location("A", 10, 10))
short_locations.append(Location("B", 10, 20))
salesmen = []
salesmen.append(Salesman(long_locations))
salesmen.append(Salesman(short_locations))
# Assert they are in reverse order
self.assertEqual(salesmen[0].calculate_fitness(), 30)
self.assertEqual(salesmen[1].calculate_fitness(), 10)
solution = Solution()
solution.sort_salesmen(salesmen)
# Assert they have been sorted
self.assertEqual(salesmen[0].calculate_fitness(), 10)
self.assertEqual(salesmen[1].calculate_fitness(), 30)
