Apply suggestions from code review

lanl-ansi · Sep 5, 2023 · 2b0185e · 2b0185e
1 parent ccf2183
commit 2b0185e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/test/docs.jl b/test/docs.jl
@@ -31,7 +31,7 @@
         #print(pm.model)
 
         @test JuMP.num_nonlinear_constraints(pm.model) == 0
-        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr{JuMP.VariableRef}, JuMP.MOI.EqualTo{Float64}) == 12
+        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr, JuMP.MOI.EqualTo{Float64}) == 12
         @test JuMP.num_variables(pm.model) == 28
 
         result = optimize_model!(pm, optimizer=JuMP.optimizer_with_attributes(Ipopt.Optimizer, "print_level"=>0))

diff --git a/test/model.jl b/test/model.jl
@@ -16,7 +16,7 @@
         pm = instantiate_model("../test/data/matpower/case5.m", ACPPowerModel, PowerModels.build_opf, jump_model=m)
 
         @test JuMP.num_nonlinear_constraints(pm.model) == 0
-        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr{JuMP.VariableRef}, JuMP.MOI.EqualTo{Float64}) == 28
+        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr, JuMP.MOI.EqualTo{Float64}) == 28
         @test JuMP.num_variables(pm.model) == 49
 
         @test pm.model[:my_var] == x

diff --git a/test/opf-obj.jl b/test/opf-obj.jl
@@ -69,7 +69,7 @@ end
 
     @testset "jump model objective type" begin
         pm = instantiate_model(data, ACPPowerModel, build_opf)
-        @test JuMP.objective_function(pm.model) isa JuMP.NonlinearExpr{JuMP.VariableRef}
+        @test JuMP.objective_function(pm.model) isa JuMP.NonlinearExpr
     end
 
     @testset "opf objective" begin