Fix OpResult repr.

1 files changed, 6 insertions, 2 deletions

diff --git a/pyecsca/ec/formula/base.py b/pyecsca/ec/formula/base.py
index a9402f1..429f4c1 100644
--- a/pyecsca/ec/formula/base.py
+++ b/pyecsca/ec/formula/base.py
@@ -43,9 +43,13 @@ class OpResult:
         return self.name
 
     def __repr__(self):
-        # TODO: This repr is broken for square and neg and inv.
         char = self.op.op_str
-        parents = char.join(str(parent) for parent in self.parents)
+        if self.op == OpType.Inv:
+            parents = "1" + char + str(self.parents[0])
+        elif self.op == OpType.Neg:
+            parents = char + str(self.parents[0])
+        else:
+            parents = char.join(str(parent) for parent in self.parents)
         return f"{self.name} = {parents}"