Add tests for using ExprGraphs from NLP evaluators (#1043)

Author: joehuchette

src/Test/nlp.jl

@@ -1,4 +1,4 @@
-
+const VI = MOI.VariableIndex
 
 struct HS071 <: MOI.AbstractNLPEvaluator
     enable_hessian::Bool
@@ -24,9 +24,23 @@ end
 
 function MOI.features_available(d::HS071)
     if d.enable_hessian
-        return [:Grad, :Jac, :Hess]
+        return [:Grad, :Jac, :Hess, :ExprGraph]
+    else
+        return [:Grad, :Jac, :ExprGraph]
+    end
+end
+
+MOI.objective_expr(d::HS071) = :(x[$(VI(1))] * x[$(VI(4))] * (x[$(VI(1))] +
+                                 x[$(VI(2))] + x[$(VI(3))]) + x[$(VI(3))])
+
+function MOI.constraint_expr(d::HS071, i::Int)
+    if i == 1
+        return :(x[$(VI(1))] * x[$(VI(2))] * x[$(VI(3))] * x[$(VI(4))] >= 25.0)
+    elseif i == 2
+        return :(x[$(VI(1))]^2 + x[$(VI(2))]^2 +
+                 x[$(VI(3))]^2 + x[$(VI(4))]^2 == 40.0)
     else
-        return [:Grad, :Jac]
+        error("Out of bounds constraint.")
     end
 end
 
@@ -178,9 +192,19 @@ end
 
 function MOI.features_available(d::FeasibilitySenseEvaluator)
     if d.enable_hessian
-        return [:Grad, :Jac, :Hess]
+        return [:Grad, :Jac, :Hess, :ExprGraph]
+    else
+        return [:Grad, :Jac, :ExprGraph]
+    end
+end
+
+MOI.objective_expr(d::FeasibilitySenseEvaluator) = :()
+
+function MOI.constraint_expr(d::FeasibilitySenseEvaluator, i::Int)
+    if i == 1
+        return :(x[$(VI(1))]^2 == 1)
     else
-        return [:Grad, :Jac]
+        error("Out of bounds constraint.")
     end
 end
 
@@ -253,7 +277,7 @@ function feasibility_sense_test_template(model::MOI.ModelLike,
 
         @test MOI.get(model, MOI.ObjectiveValue()) ≈ 0.0 atol=atol rtol=rtol
 
-        @test MOI.get(model, MOI.VariablePrimal(), x) ≈ 1.0 atol=atol rtol=rtol
+        @test abs(MOI.get(model, MOI.VariablePrimal(), x)) ≈ 1.0 atol=atol rtol=rtol
     end
 end
 

test/Test/nlp.jl

@@ -3,6 +3,36 @@ using Test
 import MathOptInterface
 const MOI = MathOptInterface
 
+@testset "hs071" begin
+    mock = MOI.Utilities.MockOptimizer(
+        MOI.Utilities.UniversalFallback(MOI.Utilities.Model{Float64}()),
+        eval_objective_value=false
+    )
+    config = MOI.Test.TestConfig(optimal_status = MOI.LOCALLY_SOLVED)
+    MOI.Utilities.set_mock_optimize!(
+        mock,
+        (mock) -> begin
+            MOI.Utilities.mock_optimize!(
+                mock, config.optimal_status,
+                [1.0, 4.7429996418092970, 3.8211499817883077, 1.379408289755698]
+            )
+            MOI.set(mock, MOI.ObjectiveValue(), 17.014017145179164)
+        end
+    )
+    MOI.Test.hs071_test(mock, config)
+    MOI.Test.hs071_no_hessian_test(mock, config)
+
+    d = MOI.Test.HS071(false)
+    VI = MOI.VariableIndex
+    @test MOI.objective_expr(d) == :(x[$(VI(1))] * x[$(VI(4))] * (x[$(VI(1))] +
+                                     x[$(VI(2))] + x[$(VI(3))]) + x[$(VI(3))])
+    @test MOI.constraint_expr(d, 1) ==
+        :(x[$(VI(1))] * x[$(VI(2))] * x[$(VI(3))] * x[$(VI(4))] >= 25.0)
+    @test MOI.constraint_expr(d, 2) ==
+        :(x[$(VI(1))]^2 + x[$(VI(2))]^2 + x[$(VI(3))]^2 + x[$(VI(4))]^2 == 40.0)
+    @test_throws ErrorException MOI.constraint_expr(d, 3)
+end
+
 @testset "mixed_complementarity" begin
     mock = MOI.Utilities.MockOptimizer(
         MOI.Utilities.UniversalFallback(MOI.Utilities.Model{Float64}())