don't use .ae method in library code

mpmath · Jan 10, 2023 · 9d7884b · 9d7884b
1 parent 1258e33
commit 9d7884b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mpmath/calculus/extrapolation.py b/mpmath/calculus/extrapolation.py
@@ -1222,7 +1222,7 @@ def init_levin(m):
                     error = lerror
                     best = est
             if TRY_EULER_MACLAURIN:
-                if ctx.mpc(ctx.sign(partial[-1]) / ctx.sign(partial[-2])).ae(-1):
+                if ctx.almosteq(ctx.mpc(ctx.sign(partial[-1]) / ctx.sign(partial[-2])), -1):
                     if verbose:
                         print ("NOT using Euler-Maclaurin: the series appears"
                             " to be alternating, so numerical\n quadrature"

diff --git a/mpmath/tests/test_summation.py b/mpmath/tests/test_summation.py
@@ -15,6 +15,8 @@ def test_nsum():
     assert nsum(lambda k: 1/k**2, [1, inf]).ae(pi**2 / 6)
     assert nsum(lambda k: 2**k/fac(k), [0, inf]).ae(exp(2))
     assert nsum(lambda k: 1/k**2, [4, inf], method='e').ae(0.2838229557371153)
+    assert abs(fp.nsum(lambda k: 1/k**4, [1, fp.inf]) - 1.082323233711138) < 1e-5
+    assert abs(fp.nsum(lambda k: 1/k**4, [1, fp.inf], method='e') - 1.082323233711138) < 1e-4
 
 def test_nprod():
     mp.dps = 15