Add rule for ArcTan[ComplexInfinity] (#1249)

Fixes #510
Mathics3 · Dec 28, 2024 · 4229363 · 4229363
1 parent 7925403
commit 4229363
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/mathics/builtin/numbers/trig.py b/mathics/builtin/numbers/trig.py
@@ -557,6 +557,7 @@ class ArcTan(MPMathFunction):
     rules = {
         "ArcTan[0]": "0",
         "ArcTan[1]": "Pi/4",
+        "ArcTan[DirectedInfinity[]]": "Indeterminate",
         "ArcTan[Undefined]": "Undefined",
         "ArcTan[Undefined, x_]": "Undefined",
         "ArcTan[y_, Undefined]": "Undefined",

diff --git a/test/builtin/numbers/test_trig.py b/test/builtin/numbers/test_trig.py
@@ -8,6 +8,7 @@
 from test.helper import check_evaluation
 
 import pytest
+from sympy.core.numbers import ComplexInfinity
 
 
 def test_ArcCos():
@@ -29,6 +30,12 @@ def test_ArcCos():
 @pytest.mark.parametrize(
     ("str_expr", "msgs", "str_expected", "fail_msg"),
     [
+        (
+            "ArcTan[ComplexInfinity]",
+            None,
+            "Indeterminate",
+            "Rule added for Arctan[ComplexInfinity]",
+        ),
         ("ArcTan[-1, 1]", None, "3 Pi / 4", None),
         ("ArcTan[1, -1]", None, "-Pi / 4", None),
         ("ArcTan[-1, -1]", None, "-3 Pi / 4", None),