test list of lists

Author: Zeroto521

tests/test_cons.py

@@ -180,18 +180,27 @@ def test_cons_indicator():
     assert c1.getConshdlrName() == "indicator"
 
 def test_cons_indicator_with_matrix_binvar():
+    # to fix #1043
     m = Model()
 
     x = m.addVar(vtype="B")
-    binvar = m.addMatrixVar(1, vtype="B")
-    # binvar is a matrix variable to fix #1043
-    m.addConsIndicator(x >= 1, binvar, activeone=True)
-    m.addConsIndicator(x <= 0, binvar, activeone=False)
-
-    m.setObjective(binvar.sum(), "maximize")
+    # binvar is a matrix variable
+    binvar1 = m.addMatrixVar(1, vtype="B")
+    m.addConsIndicator(x >= 1, binvar1, activeone=True)
+    m.addConsIndicator(x <= 0, binvar1, activeone=False)
+
+    y = m.addMatrixVar(2, vtype="B")
+    # binvar2 is a list of lists
+    binvar2 = [[m.addVar(vtype="B")], [m.addVar(vtype="B")]]
+    m.addConsIndicator(y >= 1, binvar2, activeone=True)
+    m.addConsIndicator(y <= 0, binvar2, activeone=False)
+
+    m.setObjective(binvar1[0] + binvar2[0][0] + binvar2[0][1], "maximize")
     m.optimize()
 
     assert m.getVal(x) == 1
+    assert m.getVal(y[0]) == 1
+    assert m.getVal(y[1]) == 1
 
 @pytest.mark.xfail(
     reason="addConsIndicator doesn't behave as expected when binary variable is False. See Issue #717."