add recursive test case and div/mul cases

CPMpy · Nov 9, 2023 · 3033c18 · 3033c18
1 parent b01370d
commit 3033c18
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/test_trans_linearize.py b/tests/test_trans_linearize.py
@@ -197,6 +197,9 @@ def test_sum(self):
         cons = canonical_comparison([cp.sum([a,b,c,10]) <= rhs])[0]
         self.assertEqual("sum([a, b, c]) <= -5", str(cons))
 
+        cons = canonical_comparison([(cp.sum([a, b, c, 10]) <= rhs) == (cp.sum([a]) <= b)])[0]
+        self.assertEqual('(sum([a, b, c]) <= -5) == (sum([1, -1] * [a, b]) <= 0)', str(cons))
+
         rhs = cp.sum([b,c])
         cons = canonical_comparison([cp.sum([a, b]) <= rhs])[0]
         self.assertEqual("sum([1, 1, -1, -1] * [a, b, b, c]) <= 0", str(cons))
@@ -209,8 +212,11 @@ def test_div(self):
         self.assertEqual("(a) // (b) <= 5", str(cons))
 
         #when adding division
-        #cons = canonical_comparison([a / b <= c / rhs])[0]
-        #cons = canonical_comparison([a + b <= c/rhs])[0]
+        cons = canonical_comparison([a / b <= c / rhs])[0]
+        self.assertEqual('?', str(cons))
+
+        cons = canonical_comparison([a + b <= c * a])[0]
+        self.assertEqual('(sum([a, b]) - c * a <= 0)', str(cons))
 
 
     def test_wsum(self):