coverage

gpkit/tests/t_constraints.py

@@ -11,8 +11,10 @@
 from gpkit.tests.helpers import run_tests
 from gpkit.exceptions import (InvalidGPConstraint, PrimalInfeasible,
                               DimensionalityError)
-from gpkit.constraints.relax import ConstraintsRelaxed
+from gpkit.constraints.relax import (ConstraintsRelaxed, ConstantsRelaxed,
+                                     ConstraintsRelaxedEqually)
 from gpkit.constraints.bounded import Bounded
+from gpkit.globals import NamedVariables
 import gpkit
 
 
@@ -55,6 +57,27 @@ def test_evalfn(self):
         sol = m.solve(verbosity=0)
         self.assertAlmostEqual(sol(x2), sol(x)**2)
 
+    def test_relax_list(self):
+        x = Variable("x")
+        x_max = Variable("x_max", 1)
+        x_min = Variable("x_min", 2)
+        constraints = [x_min <= x, x <= x_max]
+        ConstraintsRelaxed(constraints)
+        ConstantsRelaxed(constraints)
+        ConstraintsRelaxedEqually(constraints)
+
+    def test_relax_linked(self):
+        x = Variable("x")
+        x_max = Variable("x_max", 1)
+        x_min = Variable("x_min", lambda c: 2*c[x_max])
+        constraints = ConstraintSet([x_min <= x, x <= x_max])
+        _ = ConstantsRelaxed(constraints)
+        NamedVariables.reset_modelnumbers()
+        include_min = ConstantsRelaxed(constraints, include_only=["x_min"])
+        NamedVariables.reset_modelnumbers()
+        exclude_max = ConstantsRelaxed(constraints, exclude=["x_max"])
+        self.assertEqual(str(include_min), str(exclude_max))
+
     def test_equality_relaxation(self):
         x = Variable("x")
         m = Model(x, [x == 3, x == 4])