Rewrite compute and remove reference to computing in OPERATORS

mapytex/calculus/core/compute/__init__.py

@@ -12,6 +12,15 @@ Computing with MO
 
 from ..operator import OPERATORS
 from .exceptions import ComputeError
+from .add import add
+from .minus import minus
+from .multiply import multiply
+
+OPERATIONS = {
+        "+": add, 
+        "-": minus, 
+        "*": multiply,
+        }
 
 def compute(node, left_v, right_v):
     """ 
@@ -19,28 +28,24 @@ def compute(node, left_v, right_v):
 
     :example:
 
-    >>> compute("+", 1, 2)
+    >>> from ..MO.mo import MOnumber
-    3
+    >>> compute("+", MOnumber(1), MOnumber(2))
-    >>> compute("-", None, 2)
+    <MOnumber 3>
-    -2
+    >>> compute("-", None, MOnumber(2))
-    >>> compute("-", 1, 2)
+    <MOnumber - 2>
+    >>> compute("*", MOnumber(1), MOnumber(2))
+    <MOnumber 2>
+    >>> compute("~", MOnumber(1), MOnumber(2))
     Traceback (most recent call last):
         ...
-    mapytex.calculus.core.compute.exceptions.ComputeError: left_v need to be None for operator with arity 1
+    mapytex.calculus.core.compute.exceptions.ComputeError: Unknown operation (~)
     """
-    op = OPERATORS[node]
+    try:
-    lv = left_v
+        operation = OPERATIONS[node]
-    rv = right_v
+    except KeyError:
-    if op['arity'] == 1:
+        raise ComputeError(f"Unknown operation ({node})")
-        if lv is None:
+
-            return op["operate"](rv)()
+    return operation(left_v, right_v)
-        else:
-            raise ComputeError("left_v need to be None for operator with arity 1")
-    else:
-        try:
-            return op["operate"](lv)(rv)
-        except NotImplemented:
-            return op["roperate"](rv)(lv)
 
 
 # -----------------------------

mapytex/calculus/core/operator.py

@@ -15,31 +15,22 @@ OPERATORS = {
         "+": {'repr': "+",
             'arity': 2,
             'precedence': 0,
-            'operate': lambda x: x.__getattribute__("__add__"),
-            'roperate': lambda x: x.__getattribute__("__radd__"),
             },
         "-": {'repr': "-",
             'arity': 1,
             'precedence': 1,
-            'operate': lambda x: x.__getattribute__("__neg__"),
             },
         "*": {'repr': "*",
             'arity': 2,
             'precedence': 2,
-            'operate': lambda x: x.__getattribute__("__mul__"),
-            'roperate': lambda x: x.__getattribute__("__rmul__"),
             },
         "/": {'repr': "/",
             'arity': 2,
             'precedence': 3,
-            'operate': lambda x: x.__getattribute__("__div__"),
-            'roperate': lambda x: x.__getattribute__("__rdiv__"),
             },
         "^": {'repr': "^",
             'arity': 2,
             'precedence': 4,
-            'operate': lambda x: x.__getattribute__("__pow__"),
-            'roperate': lambda x: x.__getattribute__("__rpow__"),
             },
 }