Merge pull request #7489 from anntzer/faster-real-if-close

charris · charris · commit 2af06c804931 · 2016-03-31T16:58:19.000-06:00
Faster real_if_close.
diff --git a/numpy/lib/type_check.py b/numpy/lib/type_check.py
@@ -424,7 +424,7 @@ def real_if_close(a,tol=100):
         from numpy.core import getlimits
         f = getlimits.finfo(a.dtype.type)
         tol = f.eps * tol
-    if _nx.allclose(a.imag, 0, atol=tol):
+    if _nx.all(_nx.absolute(a.imag) < tol):
         a = a.real
     return a
 
