Merge pull request #556 from effigies/fix/numpy_print_doctests

MRG: Use legacy float array printing for now

Using `.. testsetup::` should avoid visual changes to the docs.

Introduced in numpy/numpy@710e032

Author: matthew-brett

nibabel/__init__.py

@@ -67,6 +67,7 @@
 from . import mriutils
 from . import streamlines
 from . import viewers
+from .testing import setup_test
 
 # Note test requirement for "mock".  Requirement for "nose" tested by numpy.
 try:

nibabel/affines.py

@@ -6,6 +6,7 @@
 import numpy as np
 
 from six.moves import reduce
+from .testing import setup_test  # flake8: noqa F401
 
 
 class AffineError(ValueError):

nibabel/casting.py

@@ -8,6 +8,7 @@
 from platform import processor, machine
 
 import numpy as np
+from .testing import setup_test  # flake8: noqa F401
 
 
 class CastingError(Exception):

nibabel/nicom/dwiparams.py

@@ -21,6 +21,7 @@
 '''
 import numpy as np
 import numpy.linalg as npl
+from ..testing import setup_test  # flake8: noqa F401
 
 
 def B2q(B, tol=None):

nibabel/nifti1.py

@@ -27,6 +27,7 @@
 from .spm99analyze import SpmAnalyzeHeader
 from .casting import have_binary128
 from .pydicom_compat import have_dicom, pydicom as pdcm
+from .testing import setup_test  # flake8: noqa F401
 
 # nifti1 flat header definition for Analyze-like first 348 bytes
 # first number in comments indicates offset in file header in bytes

nibabel/quaternions.py

@@ -27,6 +27,7 @@
 
 import math
 import numpy as np
+from .testing import setup_test  # flake8: noqa F401
 
 MAX_FLOAT = np.maximum_sctype(np.float)
 FLOAT_EPS = np.finfo(np.float).eps

nibabel/testing/__init__.py

@@ -209,3 +209,13 @@ def assert_arr_dict_equal(dict1, dict2):
     for key, value1 in dict1.items():
         value2 = dict2[key]
         assert_array_equal(value1, value2)
+
+
+def setup_test():
+    """ Set numpy print options to "legacy" for new versions of numpy
+
+    If imported into a file, nosetest will run this before any doctests.
+    """
+    from distutils.version import LooseVersion
+    if LooseVersion(np.__version__) >= LooseVersion('1.14'):
+        np.set_printoptions(sign='legacy')