constraint.py

class Constraint:

    def distance_squared(self, u):
        """
        Squared distance of a given point to the set

        :param u: given point
        :return: squared distance
        :rtype: float
        """
        raise NotImplementedError(
            "Method `distance_squared` is not implemented")

    def project(self, u):
        """
        Project a given point onto the set

        :param u: given point

        :return: projection of `u` onto this set
        """
        raise NotImplementedError("Method `project` is not implemented")

    def is_convex(self):
        """
        Whether the set is convex 
        """
        return False

    def is_compact(self):
        """
        Whether the set is compact
        """
        return False

    def dimension(self):
        """
        Constraint dimension

        Derived classes can override this method to return the dimension of the 
        constraint, where possible, or return `None` if the constraint does not 
        have a fixed dimension.
        """