RF: Move setup_test to nibabel/__init__.py

Author: effigies

nibabel/__init__.py

@@ -34,6 +34,18 @@
 
 For more detailed information see the :ref:`manual`.
 """
+
+
+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(legacy="1.13")
+
+
 # module imports
 from . import analyze as ana
 from . import spm99analyze as spm99
@@ -67,7 +79,6 @@
 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,7 +6,7 @@
 import numpy as np
 
 from six.moves import reduce
-from .testing import setup_test  # noqa
+from . import setup_test  # noqa
 
 
 class AffineError(ValueError):

nibabel/casting.py

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

nibabel/nicom/dwiparams.py

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

nibabel/nifti1.py

@@ -27,7 +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  # noqa
+from . import setup_test  # noqa
 
 # 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,7 +27,7 @@
 
 import math
 import numpy as np
-from .testing import setup_test  # noqa
+from . import setup_test  # noqa
 
 MAX_FLOAT = np.maximum_sctype(np.float)
 FLOAT_EPS = np.finfo(np.float).eps

nibabel/testing/__init__.py

@@ -213,13 +213,3 @@ 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(legacy="1.13")