routing: test AllowedVehicles in Java and .Net

Author: Mizux

ortools/javatests/com/google/ortools/routing/RoutingModelTest.java

@@ -15,6 +15,7 @@
 
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
@@ -325,6 +326,37 @@ public void testRoutingModel_addDisjunction() {
     assertEquals(8, solution.objectiveValue());
   }
 
+  @Test
+  public void testAllowedVehicles() {
+    final IndexManager manager = new IndexManager(2, 2, 0);
+    assertNotNull(manager);
+    final Model model = new Model(manager);
+    assertNotNull(model);
+
+    // out of range vehicle index is allowed.
+    model.setAllowedVehiclesForIndex(new int[] {13}, 1);
+    assertFalse(model.isVehicleAllowedForIndex(0, 1));
+    assertFalse(model.isVehicleAllowedForIndex(1, 1));
+    assertTrue(model.isVehicleAllowedForIndex(13, 1));
+
+    // empty array means any vehicles are allowed.
+    model.setAllowedVehiclesForIndex(new int[] {}, 1);
+    assertTrue(model.isVehicleAllowedForIndex(0, 1));
+    assertTrue(model.isVehicleAllowedForIndex(1, 1));
+    assertTrue(model.isVehicleAllowedForIndex(42, 1));
+
+    // only allow first vehicle for node 1.
+    model.setAllowedVehiclesForIndex(new int[] {0}, 1);
+    assertTrue(model.isVehicleAllowedForIndex(0, 1));
+    assertFalse(model.isVehicleAllowedForIndex(1, 1));
+    assertFalse(model.isVehicleAllowedForIndex(2, 1));
+
+    final Assignment solution = model.solve(null);
+    assertNotNull(solution);
+    assertEquals(1, solution.value(model.nextVar(model.start(0))));
+    assertEquals(2, model.getVehicleClassesCount());
+  }
+
   @Test
   public void testRoutingModel_addConstantDimension() {
     final IndexManager manager = new IndexManager(10, 1, 0);

ortools/routing/csharp/RoutingModelTests.cs

@@ -260,6 +260,38 @@ long SolveObjective(bool addRouteConstraint)
         Assert.Equal(8, baselineObjective);
         Assert.Equal(18, constrainedObjective);
     }
+
+    [Fact]
+    public void TestAllowedVehicles()
+    {
+        IndexManager manager = new IndexManager(2, 2, 0);
+        Assert.NotNull(manager);
+        Model model = new Model(manager);
+        Assert.NotNull(model);
+
+        // out of range vehicle index is allowed.
+        model.SetAllowedVehiclesForIndex(new int[] { 13 }, 1);
+        Assert.False(model.IsVehicleAllowedForIndex(0, 1));
+        Assert.False(model.IsVehicleAllowedForIndex(1, 1));
+        Assert.True(model.IsVehicleAllowedForIndex(13, 1));
+
+        // empty array means any vehicles are allowed.
+        model.SetAllowedVehiclesForIndex(new int[] { }, 1);
+        Assert.True(model.IsVehicleAllowedForIndex(0, 1));
+        Assert.True(model.IsVehicleAllowedForIndex(1, 1));
+        Assert.True(model.IsVehicleAllowedForIndex(42, 1));
+
+        // only allow first vehicle for node 1.
+        model.SetAllowedVehiclesForIndex(new int[] { 0 }, 1);
+        Assert.True(model.IsVehicleAllowedForIndex(0, 1));
+        Assert.False(model.IsVehicleAllowedForIndex(1, 1));
+        Assert.False(model.IsVehicleAllowedForIndex(2, 1));
+
+        Assignment solution = model.Solve(null);
+        Assert.NotNull(solution);
+        Assert.Equal(1, solution.Value(model.NextVar(model.Start(0))));
+        Assert.Equal(2, model.GetVehicleClassesCount());
+    }
 }
 
 // TODO(user): Add tests for Routing[Cost|Vehicle|Resource]ClassIndex

ortools/routing/csharp/routing.i

@@ -449,7 +449,6 @@ using Domain = Google.OrTools.Util.Domain;
 %unignore Model::RegisterUnaryTransitVector;
 %unignore Model::ResourceVar;
 %unignore Model::RestoreAssignment;
-%unignore Model::SetAllowedVehiclesForIndex;
 %unignore Model::SetAllowedVehiclesForIndex(const std::vector<int>&, int64_t);
 %unignore Model::SetAmortizedCostFactorsOfAllVehicles;
 %unignore Model::SetAmortizedCostFactorsOfVehicle;

ortools/routing/java/routing.swig

@@ -347,7 +347,13 @@ import java.util.function.LongUnaryOperator;
 
 %rename (restoreAssignment) Model::RestoreAssignment;
 %rename (routesToAssignment) Model::RoutesToAssignment;
-%rename (setAllowedVehiclesForIndex) Model::SetAllowedVehiclesForIndex;
+%rename (setAllowedVehiclesForIndex) Model::SetAllowedVehiclesForIndex(const std::vector<int>&, int64_t);
+%extend Model {
+  void SetAllowedVehiclesForIndex(const std::vector<int>& vehicles,
+                                  int64_t index) {
+    $self->SetAllowedVehiclesForIndex(absl::Span<const int>(vehicles), index);
+  }
+}
 %rename (setAmortizedCostFactorsOfAllVehicles) Model::SetAmortizedCostFactorsOfAllVehicles;
 %rename (setAmortizedCostFactorsOfVehicle) Model::SetAmortizedCostFactorsOfVehicle;
 %rename (setArcCostEvaluatorOfAllVehicles) Model::SetArcCostEvaluatorOfAllVehicles;