Add tests

Author: blegat

src/MOI_wrapper.jl

@@ -440,14 +440,14 @@ function MOI.set(
     value,
 )
     primal_dual_map = optimizer.dual_problem.primal_dual_map
-    if ci in keys(primal_dual_map.constrained_var_dual)
+    if ci in keys(primal_dual_map.primal_constrained_variables)
         msg = "Setting starting value for variables constrained at creation is not supported yet"
         throw(MOI.SetAttributeNotAllowed(attr, msg))
     end
     MOI.set(
         optimizer.dual_problem.dual_model,
         _variable_dual_attribute(attr),
-        get_vi_dual_problem(optimizer, ci),
+        primal_dual_map.primal_constraint_data[ci].dual_variables[],
         value,
     )
     return
@@ -543,9 +543,9 @@ end
 function MOI.set(
     optimizer::DualOptimizer,
     attr::Union{MOI.ConstraintPrimal,MOI.ConstraintPrimalStart},
-    ci::MOI.ConstraintIndex{F},
+    ci::MOI.ConstraintIndex{<:MOI.AbstractScalarFunction},
     value,
-) where {F<:MOI.AbstractScalarFunction}
+)
     primal_dual_map = optimizer.dual_problem.primal_dual_map
     if ci in keys(primal_dual_map.constrained_var_dual)
         error(

test/Tests/test_attributes.jl

@@ -0,0 +1,76 @@
+# Copyright (c) 2017: Guilherme Bodin, and contributors
+#
+# Use of this source code is governed by an MIT-style license that can be found
+# in the LICENSE.md file or at https://opensource.org/licenses/MIT.
+
+module TestAttributes
+
+using Test
+
+import MathOptInterface as MOI
+import MathOptInterface.Utilities as MOIU
+import Dualization
+
+function runtests()
+    for name in names(@__MODULE__; all = true)
+        if startswith("$(name)", "test_")
+            @testset "$(name)" begin
+                getfield(@__MODULE__, name)()
+            end
+        end
+    end
+    return
+end
+
+###
+### Helper structs
+###
+
+struct DummyModelAttribute <: MOI.AbstractModelAttribute end
+
+struct DummyVariableAttribute <: MOI.AbstractVariableAttribute end
+
+struct DummyConstraintAttribute <: MOI.AbstractConstraintAttribute end
+
+###
+### The tests
+###
+
+function test_constraint_attribute()
+    T = Float64
+    mock = MOI.Utilities.MockOptimizer(
+        MOI.Utilities.UniversalFallback(MOI.Utilities.Model{T}());
+        eval_variable_constraint_dual = false,
+    )
+    dual = MOI.instantiate(() -> Dualization.DualOptimizer(mock), with_cache_type = T)
+    x = MOI.add_variable(dual)
+    set_constant = -4
+    set = MOI.GreaterThan(T(set_constant))
+    ci = MOI.add_constraint(dual, T(1) * x, set)
+    MOI.set(dual, MOI.ObjectiveSense(), MOI.MIN_SENSE)
+    obj = T(2) * x
+    MOI.set(dual, MOI.ObjectiveFunction{typeof(obj)}(), obj)
+    MOI.optimize!(dual)
+    for attr in [MOI.ConstraintDualStart(), MOI.ConstraintPrimalStart()]
+        attr = MOI.ConstraintDualStart()
+        @test MOI.supports(dual, attr, typeof(ci))
+        value = rand(T)
+        MOI.set(dual, attr, ci, value)
+        @test MOI.get(dual, attr, ci) == value
+    end
+
+    value = rand(T)
+    mock_vi = MOI.get(mock, MOI.ListOfVariableIndices())[]
+    MOI.set(mock, MOI.VariablePrimal(), mock_vi, value)
+    @test MOI.get(dual.optimizer, MOI.ConstraintDual(), ci) ≈ value
+
+    value = rand(T)
+    mock_ci = MOI.get(mock, MOI.ListOfConstraintIndices{MOI.VariableIndex,typeof(set)}())[]
+    MOI.set(mock, MOI.ConstraintDual(), mock_ci, value)
+    @test MOI.get(dual.optimizer, MOI.ConstraintPrimal(), ci) ≈ value + set_constant
+    return
+end
+
+end  # module
+
+TestAttributes.runtests()

test/runtests.jl

@@ -57,6 +57,7 @@ include("Problems/Power/power_cone_problems.jl")
 include("Problems/Feasibility/feasibility_problems.jl")
 
 # Run tests to travis ci
+include("Tests/test_attributes.jl")
 include("Tests/test_structures.jl")
 include("Tests/test_supported.jl")
 include("Tests/test_objective_coefficients.jl")