Add tests

Author: odow

Project.toml

@@ -8,6 +8,7 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MathOptInterface = "b8f27783-ece8-5eb3-8dc8-9495eed66fee"
 MutableArithmetics = "d8a4904e-b15c-11e9-3269-09a3773c0cb0"
 OrderedCollections = "bac558e1-5e72-5ebc-8fee-abe8a469f55d"
+Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 SnoopPrecompile = "66db9d55-30c0-4569-8b51-7e840670fc0c"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"

test/JuMPExtension.jl

@@ -35,7 +35,10 @@ mutable struct MyModel <: JuMP.AbstractModel
     con_to_name::Dict{ConstraintIndex,String}
     name_to_con::Union{Dict{String,ConstraintIndex},Nothing}
     objective_sense::JuMP.MOI.OptimizationSense
-    objective_function::JuMP.AbstractJuMPScalar
+    objective_function::Union{
+        JuMP.AbstractJuMPScalar,
+        Vector{<:JuMP.AbstractJuMPScalar},
+    }
     obj_dict::Dict{Symbol,Any}
     function MyModel()
         return new(
@@ -447,7 +450,10 @@ function JuMP.num_constraints(
     end
 end
 
-function JuMP.set_objective_function(m::MyModel, f::JuMP.AbstractJuMPScalar)
+function JuMP.set_objective_function(
+    m::MyModel,
+    f::Union{JuMP.AbstractJuMPScalar,Vector{<:JuMP.AbstractJuMPScalar}},
+)
     m.objective_function = f
     return
 end

test/runtests.jl

@@ -8,6 +8,9 @@
 # See https://github.com/jump-dev/JuMP.jl
 #############################################################################
 
+import Pkg
+Pkg.pkg"add MathOptInterface#od/vector-optimization"
+
 import JuMP
 import Test
 

test/test_generate_and_solve.jl

@@ -552,4 +552,42 @@ c2: x + y <= 1.0
     return
 end
 
+function test_generate_solve_vector_objective()
+    model = Model() do
+        return MOI.Utilities.MockOptimizer(MOI.Utilities.Model{Float64}())
+    end
+    @variable(model, x >= 0.0)
+    @variable(model, y >= 1.0)
+    @objective(model, Min, [x, y])
+    optimize!(model)
+    mock = unsafe_backend(model)
+    MOI.set(mock, MOI.TerminationStatus(), MOI.OPTIMAL)
+    MOI.set(mock, MOI.RawStatusString(), "Optimal")
+    MOI.set(mock, MOI.ResultCount(), 1)
+    MOI.set(mock, MOI.PrimalStatus(), MOI.FEASIBLE_POINT)
+    MOI.set(mock, MOI.DualStatus(), MOI.NO_SOLUTION)
+    # MOI.set(mock, MOI.ObjectiveValue(), [0.0, 1.0])
+    MOI.set(mock, MOI.ObjectiveBound(), [0.0, 1.0])
+    MOI.set(mock, MOI.VariablePrimal(), optimizer_index(x), 0.0)
+    MOI.set(mock, MOI.VariablePrimal(), optimizer_index(y), 1.0)
+    @test sprint(print, solution_summary(model)) == """
+* Solver : Mock
+
+* Status
+  Result count       : 1
+  Termination status : OPTIMAL
+  Message from the solver:
+  "Optimal"
+
+* Candidate solution (result #1)
+  Primal status      : FEASIBLE_POINT
+  Dual status        : NO_SOLUTION
+  Objective value    : [0.00000e+00,1.00000e+00]
+  Objective bound    : [0.00000e+00,1.00000e+00]
+
+* Work counters
+"""
+    return
+end
+
 end  # module

test/test_objective.jl

@@ -189,4 +189,42 @@ function test_extension_objective_constant(
     return
 end
 
+function test_extension_objective_vector_of_variables(
+    ModelType = Model,
+    VariableType = VariableRef,
+)
+    model = ModelType()
+    @variable(model, x[1:2])
+    @objective(model, Min, x)
+    @test isequal_canonical(objective_function(model), x)
+    @test isequal_canonical(objective_function(model, typeof(x)), x)
+    return
+end
+
+function test_extension_objective_vector_affine_function(
+    ModelType = Model,
+    VariableType = VariableRef,
+)
+    model = ModelType()
+    @variable(model, x[1:2])
+    f = 1.0 .* x .+ [2.0, 3.0]
+    @objective(model, Min, f)
+    @test isequal_canonical(objective_function(model), f)
+    @test isequal_canonical(objective_function(model, typeof(f)), f)
+    return
+end
+
+function test_extension_objective_vector_quadratic_function(
+    ModelType = Model,
+    VariableType = VariableRef,
+)
+    model = ModelType()
+    @variable(model, x[1:2])
+    f = 1.0 .* x .* x .+ [2.0, 3.0] .* x + [4.0, 5.0]
+    @objective(model, Min, f)
+    @test isequal_canonical(objective_function(model), f)
+    @test isequal_canonical(objective_function(model, typeof(f)), f)
+    return
+end
+
 end  # module