Fix @constraintref for JuMPArray (#1330)

jump-dev · Jun 8, 2018 · 02b12a1 · 02b12a1
1 parent d381446
commit 02b12a1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/macros.jl b/src/macros.jl
@@ -1139,7 +1139,7 @@ macro constraintref(var)
         idxpairs = IndexPair[]
 
         code = quote
-            $(esc(gendict(varname, :ConstraintRef, idxsets...)))
+            $(gendict(esc(varname), :ConstraintRef, esc.(idxsets)...))
             nothing
         end
         return code

diff --git a/test/variable.jl b/test/variable.jl
@@ -239,4 +239,11 @@ end
         @test typeof(Y) == SparseMatrixCSC{Float64, Int}
         @test Y == sparse([1, 3], [2, 3], [1, 2])
     end
+
+    @testset "@constraintref does not work with JuMPArray #1329" begin
+        m = Model()
+        cities = [:LA, :NY]
+        @constraintref cref[cities]
+        @test cref.indexsets[1] == cities
+    end
 end