def test_solution_valid_with_required_cities(self):
required_cities = [self.possible_cities["Cordoba"]]
max_trip_time = 14
route = [
self.possible_cities["Cordoba"], self.possible_cities["Rosario"],
self.possible_cities["Ushuaia"]
]
solution = Solution(origin_city=self.origin_city,
cities=route,
required_cities=required_cities,
max_trip_time=max_trip_time)
expected_total_stay_time = 6
self.assertEqual(expected_total_stay_time,
solution.get_total_stay_time())
expected_total_travel_time = 1 + 1 + 3 + 3
self.assertEqual(expected_total_travel_time,
solution.get_total_travel_time())
expected_total_trip_time = expected_total_stay_time + expected_total_travel_time
self.assertEqual(expected_total_trip_time,
solution.get_total_trip_time())
self.assertTrue(solution.is_valid_required_city())
self.assertTrue(solution.is_valid_knapsack_time())
self.assertTrue(solution.is_valid_total_trip_time())
self.assertTrue(solution.is_valid())
def test_solution_valid_with_just_origin_city(self):
required_cities = []
max_trip_time = 999999
route = []
solution = Solution(origin_city=self.origin_city,
cities=route,
required_cities=required_cities,
max_trip_time=max_trip_time)
expected_total_stay_time = 0
self.assertEqual(expected_total_stay_time,
solution.get_total_stay_time())
expected_total_travel_time = 0
self.assertEqual(expected_total_travel_time,
solution.get_total_travel_time())
expected_total_trip_time = expected_total_stay_time + expected_total_travel_time
self.assertEqual(expected_total_trip_time,
solution.get_total_trip_time())
self.assertTrue(solution.is_valid_required_city())
self.assertTrue(solution.is_valid_knapsack_time())
self.assertTrue(solution.is_valid_total_trip_time())
self.assertTrue(solution.is_valid())
