FEAT: better repr for X.axis_name expressions

gdementen · gdementen · commit b9143cdfa3aa · 2023-10-23T16:02:41.000+02:00
(unary and binary ops)
diff --git a/larray/core/expr.py b/larray/core/expr.py
@@ -87,6 +87,9 @@ def evaluate(self, context):
         expr2 = expr_eval(self.expr2, context)
         return getattr(expr1, self.opname)(expr2)
 
+    def __repr__(self):
+        return f"BinaryOp({self.opname[2:-2]!r}, {self.expr1!r}, {self.expr2!r})"
+
 
 class UnaryOp(ExprNode):
     def __init__(self, op, expr):
@@ -97,3 +100,6 @@ def evaluate(self, context):
         # TODO: implement eval via numexpr
         expr = expr_eval(self.expr, context)
         return getattr(expr, self.opname)()
+
+    def __repr__(self):
+        return f"UnaryOp({self.opname[2:-2]!r}, {self.expr!r})"
