add tests

CPMpy · Nov 6, 2023 · 6eb4abf · 6eb4abf
1 parent d2d7520
commit 6eb4abf
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tests/test_expressions.py b/tests/test_expressions.py
@@ -6,6 +6,7 @@
 from cpmpy.expressions import *
 from cpmpy.expressions.variables import NDVarArray
 from cpmpy.expressions.core import Operator, Expression
+from cpmpy.expressions.utils import get_bounds
 
 class TestComparison(unittest.TestCase):
     def test_comps(self):
@@ -444,6 +445,27 @@ def test_incomplete_func(self):
             self.assertTrue(m.solve(solver="z3"))
             self.assertTrue(cons.value())
 
+
+    def test_list(self):
+
+        # cpm_array
+        iv = cp.intvar(0,10,shape=3)
+        lbs, ubs = iv.get_bounds()
+        self.assertListEqual([0,0,0], lbs.tolist())
+        self.assertListEqual([10,10,10], ubs.tolist())
+        # list
+        iv = [cp.intvar(0,10) for _ in range(3)]
+        lbs, ubs = get_bounds(iv)
+        self.assertListEqual([0, 0, 0], lbs)
+        self.assertListEqual([10, 10, 10], ubs)
+        # nested list
+        exprs = [intvar(0,1), [intvar(2,3), intvar(4,5)], [intvar(5,6)]]
+        lbs, ubs = get_bounds(exprs)
+        self.assertListEqual([0,[2,4],[5]], lbs)
+        self.assertListEqual([1,[3,5],[6]], ubs)
+
+
+
     def test_not_operator(self):
         p = boolvar()
         q = boolvar()