Merge pull request #817 from GarkGarcia/sqrt-re-im-fix

Implemented proper sympolic expantion for the `Re` and `Im` builtins
mathics · Sep 3, 2020 · b877bfb · b877bfb
2 parents 3089801 + 7a29e41
commit b877bfb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/mathics/builtin/arithmetic.py b/mathics/builtin/arithmetic.py
@@ -1095,6 +1095,11 @@ def apply_number(self, number, evaluation):
 
         return number
 
+    def apply(self, number, evaluation):
+        'Re[number_]'
+
+        return from_sympy(sympy.re(number.to_sympy().expand(complex=True)))
+
 
 class Im(SympyFunction):
     """
@@ -1127,6 +1132,11 @@ def apply_number(self, number, evaluation):
 
         return Integer(0)
 
+    def apply(self, number, evaluation):
+        'Im[number_]'
+
+        return from_sympy(sympy.im(number.to_sympy().expand(complex=True)))
+
 
 class Conjugate(_MPMathFunction):
     """