Improve coverage

jump-dev · Dec 14, 2020 · 5f991d8 · 5f991d8
1 parent 9e4729a
commit 5f991d8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/model.jl b/test/model.jl
@@ -579,15 +579,17 @@ function test_copy_expr_aff()
     model = Model()
     @variable(model, x)
     @expression(model, ex, 2 * x + 1)
-    new_model, _ = copy_model(model)
+    new_model, ref_map = copy_model(model)
+    @test ref_map[ex] == new_model[:ex]
     @test new_model[:ex] == 2 * new_model[:x] + 1
 end
 
 function test_copy_expr_quad()
     model = Model()
     @variable(model, x)
     @expression(model, ex, 2 * x^2 + x + 1)
-    new_model, _ = copy_model(model)
+    new_model, ref_map = copy_model(model)
+    @test ref_map[ex] == new_model[:ex]
     @test new_model[:ex] == 2 * new_model[:x]^2 + new_model[:x] + 1
 end