Consistency

Author: Joao-Dionisio

src/pyscipopt/recipes/getLocalConss.py

@@ -30,8 +30,21 @@ def getLocalConss(model: Model, node = None) -> list[Constraint]:
 
     return [model.getConss(), added_conss]
 
-def getNAddedConss(model: Model) -> int:
+def getNLocalConss(model: Model, node = None) -> list[int]:
     """
     Returns the number of local constraints of a node.
+
+    Parameters
+    ----------
+    model : Model
+        The model from which to retrieve the number of local constraints.
+    node : Node, optional
+        The node from which to retrieve the number of local constraints. If not provided, the current node is used.
+
+    Returns
+    -------
+    list[int]
+        A list of the number of local constraints. First entry is the number of global constraints, second entry is the number of all the added constraints.
     """
-    return len(getLocalConss(model)[1])
+    local_conss = getLocalConss(model, node)
+    return [len(local_conss[0]), len(local_conss[1])]

tests/test_recipe_getLocalConss.py

@@ -4,22 +4,24 @@
 
 def localconss(model, event):
     local_conss = getLocalConss(model)
-    assert len(local_conss[1]) == getNAddedConss(model)
+    assert len(local_conss[1]) == getNLocalConss(model)[1]
+    assert len(local_conss[0]) == len(model.getConss())
+    assert local_conss[0] == model.getConss()
 
     vars = model.getVars()
     if model.getCurrentNode().getNumber() == 1:
         pass
 
     elif model.getCurrentNode().getNumber() == 2:
         model.data["local_cons1"] = model.addCons(vars[0] + vars[1] <= 1, name="c1", local=True)
-        assert getNAddedConss(model) == 1
+        assert getNLocalConss(model)[1] == 1
         assert getLocalConss(model)[1][0] == model.data["local_cons1"]
 
     elif model.getCurrentNode().getNumber() == 4:
         local_conss = getLocalConss(model)
         model.data["local_cons2"] = model.addCons(vars[1] + vars[2] <= 1, name="c2", local=True)
         model.data["local_cons3"] = model.addCons(vars[2] + vars[3] <= 1, name="c3", local=True)
-        assert getNAddedConss(model) == 3
+        assert getNLocalConss(model)[1] == 3
         assert getLocalConss(model)[1][0] == model.data["local_cons1"]
         assert getLocalConss(model)[1][1] == model.data["local_cons2"]
         assert getLocalConss(model)[1][2] == model.data["local_cons3"]