Merge pull request #989 from henriquebecker91/hb/batch_delete_test

Tests for vector specialized version of MOI.delete

Author: blegat

src/Test/UnitTests/modifications.jl

@@ -353,4 +353,41 @@ function delete_variable_with_single_variable_obj(model::MOI.ModelLike,
 end
 modificationtests["delete_variable_with_single_variable_obj"] = delete_variable_with_single_variable_obj
 
+"""
+    delete_variables_in_a_batch(model::MOI.ModelLike, config::TestConfig)
+
+Test deleting many variables in a batch (i.e. using the delete method which
+takes a vector of variable references). If `config.solve=true` confirm that it
+solves correctly.
+"""
+function delete_variables_in_a_batch(model::MOI.ModelLike,
+                                     config::TestConfig)
+    MOI.empty!(model)
+    @test MOI.is_empty(model)
+    MOIU.loadfromstring!(model,"""
+        variables: x, y, z
+        minobjective: 1.0 * x + 2.0 * y + 3.0 * z
+        c1: x >= 1.0
+        c2: y >= 1.0
+        c3: z >= 1.0
+    """)
+    x, y, z = MOI.get(model, MOI.ListOfVariableIndices())
+    @test MOI.is_valid(model, x)
+    @test MOI.is_valid(model, y)
+    @test MOI.is_valid(model, z)
+    if config.solve
+        MOI.optimize!(model)
+        @test MOI.get(model, MOI.ObjectiveValue()) == 6.0
+    end
+    MOI.delete(model, [x, z])
+    @test !MOI.is_valid(model, x)
+    @test MOI.is_valid(model, y)
+    @test !MOI.is_valid(model, z)
+    if config.solve
+        MOI.optimize!(model)
+        @test MOI.get(model, MOI.ObjectiveValue()) == 2.0
+    end
+end
+modificationtests["delete_variables_in_a_batch"] = delete_variables_in_a_batch
+
 @moitestset modification

test/Test/unit.jl

@@ -398,6 +398,19 @@ end
     # `UniversalFallback` needed for `MOI.Silent`
     mock   = MOIU.MockOptimizer(MOIU.UniversalFallback(MOIU.Model{Float64}()))
     config = MOIT.TestConfig()
+    @testset "delete_variables_in_a_batch" begin
+        MOIU.set_mock_optimize!(mock,
+            (mock::MOIU.MockOptimizer) -> MOIU.mock_optimize!(mock,
+                MOI.OPTIMAL, (MOI.FEASIBLE_POINT, [1.0, 1.0, 1.0]),
+                MOI.FEASIBLE_POINT
+            ),
+            (mock::MOIU.MockOptimizer) -> MOIU.mock_optimize!(mock,
+                MOI.OPTIMAL, (MOI.FEASIBLE_POINT, [1.0]),
+                MOI.FEASIBLE_POINT
+            )
+        )
+        MOIT.delete_variables_in_a_batch(mock, config)
+    end
     @testset "solve_set_singlevariable_lessthan" begin
         MOIU.set_mock_optimize!(mock,
             (mock::MOIU.MockOptimizer) -> MOIU.mock_optimize!(mock,