diff --git a/tests/test_globalconstraints.py b/tests/test_globalconstraints.py index 2b8c8e20e..8bc3aa65e 100644 --- a/tests/test_globalconstraints.py +++ b/tests/test_globalconstraints.py @@ -388,13 +388,16 @@ def test_lex_chain(self): from cpmpy.expressions.utils import argval self.assertTrue(sum(argval(X)) == 0) - Z = cp.intvar(0, 1, shape=(3,2)) + Z = cp.intvar(0, 1, shape=(4,2)) c = cp.LexChainLess(Z) m = cp.Model(c) self.assertTrue(m.solve()) self.assertTrue(sum(argval(Z[0])) == 0) self.assertTrue(sum(argval(Z[1])) == 1) - self.assertTrue(sum(argval(Z[2])) >= 1) + self.assertTrue(argval(Z[1,0]) == 0) + self.assertTrue(sum(argval(Z[2])) == 1) + self.assertTrue(argval(Z[2,1]) == 0) + self.assertTrue(sum(argval(Z[3])) >= 1) def test_indomain_onearg(self):