NF: add np.allclose doctest modifier

matthew-brett · matthew-brett · commit 30be1963fb96 · 2018-01-15T09:37:28.000Z
This is to work round the numpy printing changes.
diff --git a/nipy/testing/doctester.py b/nipy/testing/doctester.py
@@ -19,6 +19,7 @@
                                                NumpyOutputChecker)
 
 IGNORE_OUTPUT = register_optionflag('IGNORE_OUTPUT')
+NP_ALLCLOSE = register_optionflag('NP_ALLCLOSE')
 SYMPY_EQUAL = register_optionflag('SYMPY_EQUAL')
 STRUCTARR_EQUAL = register_optionflag('STRUCTARR_EQUAL')
 STRIP_ARRAY_REPR = register_optionflag('STRIP_ARRAY_REPR')
@@ -186,6 +187,10 @@ def check_output(self, want, got, optionflags):
                 dp = 6
             want = round_numbers(want, dp)
             got = round_numbers(got, dp)
+        # Are the arrays close when run through numpy?
+        if NP_ALLCLOSE & optionflags:
+            res = np.allclose(eval(want), eval(got))
+            return res == wanted_tf
         # Are the strings equal when run through sympy?
         if SYMPY_EQUAL & optionflags:
             from sympy import sympify
diff --git a/nipy/testing/tests/test_doctesting.py b/nipy/testing/tests/test_doctesting.py
@@ -75,6 +75,17 @@ def check_ignore_output():
     'The answer'
     """
 
+
+def check_np_allclose():
+    """ Check NP_ALLCLOSE option
+
+    >>> np.array([38.0890515 , -3.45429252])  #doctest: +NP_ALLCLOSE
+    array([ 38.0890515 ,  -3.45429252])
+    >>> np.array([38.0899 , -3.45429252])  #doctest: +NP_ALLCLOSE +NOT_EQUAL
+    array([ 38.0890515 ,  -3.45429252])
+    """
+
+
 def check_sympy_equal():
     """ Check SYMPY_EQUAL option
 
