assertion.py

"""
>>> v = {"a": 1, "b": 2}
>>> assert v == {"a": 1, "b": Any()}
>>> assert v == {"a": 1, "b": IsA(int)}
"""



class Any:
    def __eq__(self, o):
        return True


class NotNone:
    def __eq__(self, o):
        return o is not None


class IsTrue:
    def __eq__(self, o):
        return bool(o)


class IsA:
    def __init__(self, typ):
        self.typ = typ

    def __eq__(self, o):
        return isinstance(o, self.typ)