math exponentiation now works as expected for inf

rjdbcm · Nov 22, 2021 · f5db962 · f5db962
1 parent 99fc114
commit f5db962
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/Aspidites/math.py b/Aspidites/math.py
@@ -152,10 +152,6 @@ def SafeMod(a: Numeric, b: Numeric) -> Union[Numeric, Undefined]:
 
 # noinspection PyPep8Naming, PyProtectedMember,PyUnresolvedReferences
 def SafeExp(a: Numeric, b: Numeric) -> Union[Numeric, Undefined]:
-    if (
-        (a == 0 and b == 0) or (isinf(a) and b == 0) or (isinf(b) and a == 0)
-    ):  # pragma: no cover
-        return Undefined(SafeExp, a, b)
     try:
         return a ** b
     except OverflowError:

diff --git a/Aspidites/tests/test_math.py b/Aspidites/tests/test_math.py
@@ -48,9 +48,9 @@ def test_safe_div(x, y):
 @hypothesis.settings(deadline=None)
 @pt.mark.filterwarnings("ignore::RuntimeWarning")
 def test_safe_exp(x, y):
-    assert SafeExp(0, 0) == Undefined()
-    assert SafeExp(0, inf) == Undefined()
-    assert SafeExp(inf, 0) == Undefined()
+    assert SafeExp(0, 0) == 1
+    assert SafeExp(0, inf) == 0.0
+    assert SafeExp(inf, 0) == 1.0
     assume(x != 0 and y != 0)
     try:
         x ** y