TEST: Create and import setup_test fixture into affected files

Author: effigies

nibabel/affines.py

@@ -1,17 +1,12 @@
 # emacs: -*- mode: python; py-indent-offset: 4; indent-tabs-mode: nil -*-
 # vi: set ft=python sts=4 ts=4 sw=4 et:
 """ Utility routines for working with points and affine transforms
-
-.. testsetup::
-
-    from distutils.version import LooseVersion
-    if LooseVersion(np.__version__) > LooseVersion('1.13.1'):
-        np.set_printoptions(sign='legacy')
 """
 
 import numpy as np
 
 from six.moves import reduce
+from .testing import setup_test
 
 
 class AffineError(ValueError):

nibabel/casting.py

@@ -2,18 +2,13 @@
 
 Most routines work round some numpy oddities in floating point precision and
 casting.  Others work round numpy casting to and from python ints
-
-.. testsetup::
-
-    from distutils.version import LooseVersion
-    if LooseVersion(np.__version__) > LooseVersion('1.13.1'):
-        np.set_printoptions(sign='legacy')
 """
 
 from numbers import Integral
 from platform import processor, machine
 
 import numpy as np
+from .testing import setup_test
 
 
 class CastingError(Exception):

nibabel/nicom/dwiparams.py

@@ -18,14 +18,10 @@
 
    B ~ (q_est . q_est.T) / norm(q_est)
 
-.. testsetup::
-
-    from distutils.version import LooseVersion
-    if LooseVersion(np.__version__) > LooseVersion('1.13.1'):
-        np.set_printoptions(sign='legacy')
 '''
 import numpy as np
 import numpy.linalg as npl
+from ..testing import setup_test
 
 
 def B2q(B, tol=None):

nibabel/nifti1.py

@@ -9,12 +9,6 @@
 ''' Read / write access to NIfTI1 image format
 
 NIfTI1 format defined at http://nifti.nimh.nih.gov/nifti-1/
-
-.. testsetup::
-
-    from distutils.version import LooseVersion
-    if LooseVersion(np.__version__) > LooseVersion('1.13.1'):
-        np.set_printoptions(sign='legacy')
 '''
 from __future__ import division, print_function
 import warnings
@@ -33,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
 
 # nifti1 flat header definition for Analyze-like first 348 bytes
 # first number in comments indicates offset in file header in bytes

nibabel/quaternions.py

@@ -23,16 +23,11 @@
 >>> M = quat2mat(q) # from this module
 >>> vec = np.array([1, 2, 3]).reshape((3,1)) # column vector
 >>> tvec = np.dot(M, vec)
-
-.. testsetup::
-
-    from distutils.version import LooseVersion
-    if LooseVersion(np.__version__) > LooseVersion('1.13.1'):
-        np.set_printoptions(sign='legacy')
 '''
 
 import math
 import numpy as np
+from .testing import setup_test
 
 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.13.1'):
+        np.set_printoptions(sign='legacy')