def test_interpret_nv3(self):
"""Test interpret for num_vehicles=3"""
vehicle_routing = VehicleRouting(self.graph, num_vehicles=3)
self.assertEqual(
vehicle_routing.interpret(self.result_nv3),
[[[0, 1], [1, 0]], [[0, 2], [2, 0]], [[0, 3], [3, 0]]],
)
def test_interpret_d2(self):
"""Test interpret for depot=2"""
vehicle_routing = VehicleRouting(self.graph, depot=2)
self.assertEqual(
vehicle_routing.interpret(self.result_d2),
[[[2, 0], [0, 1], [1, 2]], [[2, 3], [3, 2]]],
)
def test_edgelist(self):
"""Test _edgelist"""
vehicle_routing = VehicleRouting(self.graph)
self.assertEqual(
vehicle_routing._edgelist(vehicle_routing.interpret(self.result)),
[[0, 1], [1, 0], [0, 2], [2, 3], [3, 0]],
)
def test_edge_color_nv3(self):
"""Test _edge_color for num_vehicles=3"""
vehicle_routing = VehicleRouting(self.graph, num_vehicles=3)
self.assertEqual(
vehicle_routing._edge_color(
vehicle_routing.interpret(self.result_nv3)),
[0.0, 0.0, 1 / 3, 1 / 3, 2 / 3, 2 / 3])
def test_edge_color_d2(self):
"""Test _edge_color for depot=2"""
vehicle_routing = VehicleRouting(self.graph, depot=2)
self.assertEqual(
vehicle_routing._edge_color(
vehicle_routing.interpret(self.result_d2)),
[0.0, 0.0, 0.0, 0.5, 0.5])
def test_edge_color(self):
"""Test _edge_color"""
vehicle_routing = VehicleRouting(self.graph)
self.assertEqual(
vehicle_routing._edge_color(vehicle_routing.interpret(
self.result)),
[0.0, 0.0, 0.5, 0.5, 0.5],
)