Implement some constraint methods in Pyomo

miplearn/solvers/internal.py

@@ -18,10 +18,6 @@ from miplearn.types import (
 logger = logging.getLogger(__name__)
 
 
-class ExtractedConstraint(ABC):
-    pass
-
-
 class Constraint:
     pass
 

miplearn/solvers/pyomo/base.py

@@ -270,12 +270,6 @@ class BasePyomoSolver(InternalSolver):
     def _get_node_count_regexp(self) -> Optional[str]:
         return None
 
-    def extract_constraint(self, cid):
-        raise Exception("Not implemented")
-
-    def is_constraint_satisfied(self, cobj):
-        raise Exception("Not implemented")
-
     def relax(self) -> None:
         for var in self._bin_vars:
             lb, ub = var.bounds
@@ -285,15 +279,35 @@ class BasePyomoSolver(InternalSolver):
             self._pyomo_solver.update_var(var)
 
     def get_inequality_slacks(self) -> Dict[str, float]:
-        raise Exception("not implemented")
+        result: Dict[str, float] = {}
+        for (cname, cobj) in self._cname_to_constr.items():
+            if cobj.equality:
+                continue
+            result[cname] = cobj.slack()
+        return result
+
+    def get_constraint_sense(self, cid: str) -> str:
+        cobj = self._cname_to_constr[cid]
+        has_ub = cobj.has_ub()
+        has_lb = cobj.has_lb()
+        assert (not has_lb) or (not has_ub), "range constraints not supported"
+        if has_lb:
+            return ">"
+        elif has_ub:
+            return "<"
+        else:
+            return "="
+
+    def set_constraint_sense(self, cid: str, sense: str) -> None:
+        raise Exception("Not implemented")
 
-    def set_constraint_sense(self, cid, sense):
+    def extract_constraint(self, cid: str) -> Constraint:
         raise Exception("Not implemented")
 
-    def get_constraint_sense(self, cid):
+    def is_constraint_satisfied(self, cobj: Constraint) -> bool:
         raise Exception("Not implemented")
 
-    def set_constraint_rhs(self, cid, rhs):
+    def set_constraint_rhs(self, cid: str, rhs: float) -> None:
         raise Exception("Not implemented")
 
     def is_infeasible(self) -> bool:
@@ -302,5 +316,5 @@ class BasePyomoSolver(InternalSolver):
     def get_dual(self, cid):
         raise Exception("Not implemented")
 
-    def get_sense(self):
+    def get_sense(self) -> str:
-        raise Exception("Not implemented")
+        return self._obj_sense

miplearn/solvers/tests/test_internal_solver.py

@@ -133,18 +133,17 @@ def test_internal_solver():
         stats = solver.solve()
         assert stats["Lower bound"] == 1030.0
 
-        if isinstance(solver, GurobiSolver):
+        assert solver.get_sense() == "max"
-
+        assert solver.get_constraint_sense("cut") == "<"
-            assert solver.get_sense() == "max"
+        assert solver.get_constraint_sense("eq_capacity") == "<"
-            assert solver.get_constraint_sense("cut") == "<"
-            assert solver.get_constraint_sense("eq_capacity") == "<"
 
-            # Verify slacks
+        # Verify slacks
-            assert solver.get_inequality_slacks() == {
+        assert solver.get_inequality_slacks() == {
-                "cut": 0.0,
+            "cut": 0.0,
-                "eq_capacity": 3.0,
+            "eq_capacity": 3.0,
-            }
+        }
 
+        if isinstance(solver, GurobiSolver):
             # Extract the new constraint
             cobj = solver.extract_constraint("cut")