some tests for lists in n

CPMpy · May 27, 2024 · 9eadddc · 9eadddc
1 parent ee25234
commit 9eadddc
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_globalconstraints.py b/tests/test_globalconstraints.py
@@ -145,6 +145,11 @@ def test_alldifferent_except_n(self):
         bv = cp.boolvar()
         self.assertTrue(cp.Model([cp.AllDifferentExceptN([iv[0], bv],4)]).solve())
 
+        # test with list of n
+        iv = cp.intvar(0, 4, shape=7)
+        self.assertFalse(cp.Model([cp.AllDifferentExceptN([iv], [7,8])]).solve())
+        self.assertTrue(cp.Model([cp.AllDifferentExceptN([iv], [4, 1])]).solve())
+
     def test_not_alldifferentexcept0(self):
         iv = cp.intvar(-8, 8, shape=3)
         self.assertTrue(cp.Model([~cp.AllDifferentExcept0(iv)]).solve())
@@ -959,6 +964,11 @@ def test_allEqualExceptn(self):
         self.assertTrue(cp.Model([cp.AllEqualExceptN([a,b,False, a | b], 0)]).solve())
         self.assertTrue(cp.Model([cp.AllEqualExceptN([a,b,False, a | b, y], -1)]).solve())
 
+        # test with list of n
+        iv = cp.intvar(0, 4, shape=7)
+        self.assertFalse(cp.Model([cp.AllEqualExceptN([iv], [7,8]), iv[0] != iv[1]]).solve())
+        self.assertTrue(cp.Model([cp.AllEqualExceptN([iv], [4, 1]), iv[0] != iv[1]]).solve())
+
     def test_not_allEqualExceptn(self):
         x = cp.intvar(lb=0, ub=3, shape=3)
         n = 2