Defined repr for all formula objects

Author: Cubix1729

src/logicalpy/base.py

@@ -21,6 +21,9 @@ def __eq__(self, other):
             return self.name == other.name
         return False
 
+    def __repr__(self) -> str:
+        return f"Proposition('{self.name}')"
+
     def as_latex(self) -> str:
         return self.name
 
@@ -77,6 +80,9 @@ def __eq__(self, other) -> bool:
             return self.a == other.a
         return False
 
+    def __repr__(self) -> str:
+        return f"Not({repr(self.a)})"
+
     def as_latex(self) -> str:
         return r"\neg " + self.a.as_latex()
 
@@ -136,6 +142,9 @@ def __eq__(self, other) -> bool:
             return self.a == other.a and self.b == other.b
         return False
 
+    def __repr__(self) -> str:
+        return type(self).__name__ + f"({repr(self.a)}, {repr(self.b)})"
+
     def as_latex(self) -> str:
         return f"({self.a.as_latex()} {self.LATEX_SYMBOL} {self.b.as_latex()})"
 
@@ -331,6 +340,9 @@ def __eq__(self, other) -> bool:
             return self.formula == other.formula
         return False
 
+    def __repr__(self) -> str:
+        return f"Formula({repr(self.formula)})"
+
     def __hash__(self) -> int:
         return hash(str(self))
 

tests/test_base.py

@@ -23,6 +23,10 @@ def test_str(self):
         Q = Proposition("Q")
         self.assertEqual(str(Formula(Implies(And(Or(P, Q), Not(Q)), P))), "((P ∨ Q) ∧ ¬Q) → P")
 
+    def test_repr(self):
+        P = Proposition("P")
+        self.assertEqual(repr(Formula(Or(And(P, P), P))), "Formula(Or(And(Proposition('P'), Proposition('P')), Proposition('P')))")
+
     def test_propositions(self):
         P = Proposition("P")
         Q = Proposition("Q")