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())