PyProblem: add constraints getter

Author: Benjamin Chrétien

src/roboptim/core/__init__.py

@@ -287,6 +287,7 @@ class PyProblem(object):
     def __init__(self, cost):
         self.cost = cost
         self._problem = Problem (cost._function)
+        self._constraints = list()
 
     def __str__ (self):
         return strProblem (self._problem)
@@ -317,6 +318,11 @@ def argumentScaling(self, value):
 
     def addConstraint (self, constraint, bounds):
         addConstraint (self._problem, constraint._function, bounds)
+        self._constraints.append (constraint)
+
+    @property
+    def constraints(self):
+        return self._constraints
 
 
 class PySolver(object):

tests/function.py

@@ -147,6 +147,7 @@ def test_problem(self):
         self.assertEqual ("square function (differentiable function)", "%s" % cost)
         problem = roboptim.core.PyProblem (cost)
         print (problem)
+        self.assertEqual (problem.constraints, [])
 
     def test_solver(self):
         cost = Square()
@@ -167,6 +168,8 @@ def test_solver(self):
         g2 = DoubleSquare ()
         problem.addConstraint (g2, numpy.array ([[-1., 10.],[2., 3.]]))
 
+        self.assertEqual (problem.constraints, [g1, g2])
+
         solver = roboptim.core.PySolver ("ipopt", problem)
         print (solver)