TST: Add a runif_extra_has decorator.

Author: Ben Cipollini

nibabel/testing/__init__.py

@@ -9,12 +9,14 @@
 ''' Utilities for testing '''
 from __future__ import division, print_function
 
+import os
 import sys
 import warnings
 from os.path import dirname, abspath, join as pjoin
 
 import numpy as np
 
+from numpy.testing.decorators import skipif
 # Allow failed import of nose if not now running tests
 try:
     from nose.tools import (assert_equal, assert_not_equal,
@@ -164,3 +166,11 @@ def __init__(self, *args, **kwargs):
         warnings.warn('catch_warn_reset is deprecated and will be removed in '
                       'nibabel v3.0; use nibabel.testing.clear_and_catch_warnings.',
                       FutureWarning)
+
+
+def runif_extra_has(test_str):
+    """Decorator checks to see if NIPY_EXTRA_TESTS env var contains test_str"""
+    def decorator(func):
+        skip_set = os.environ.get('NIPY_EXTRA_TESTS', '').split(',')
+        return skipif(test_str not in skip_set, "Skip slow tests.")(func)
+    return decorator

nibabel/tests/test_nifti1.py

@@ -13,17 +13,18 @@
 
 import numpy as np
 
-from ..externals.six import BytesIO
-from ..casting import type_info, have_binary128
-from ..tmpdirs import InTemporaryDirectory
-from ..spatialimages import HeaderDataError
-from ..eulerangles import euler2mat
-from ..affines import from_matvec
-from .. import nifti1 as nifti1
-from ..nifti1 import (load, Nifti1Header, Nifti1PairHeader, Nifti1Image,
-                      Nifti1Pair, Nifti1Extension, Nifti1Extensions,
-                      data_type_codes, extension_codes, slice_order_codes)
-
+from nibabel import nifti1 as nifti1
+from nibabel.affines import from_matvec
+from nibabel.casting import type_info, have_binary128
+from nibabel.eulerangles import euler2mat
+from nibabel.externals.six import BytesIO
+from nibabel.nifti1 import (load, Nifti1Header, Nifti1PairHeader, Nifti1Image,
+                            Nifti1Pair, Nifti1Extension, Nifti1Extensions,
+                            data_type_codes, extension_codes,
+                            slice_order_codes)
+from nibabel.openers import ImageOpener
+from nibabel.spatialimages import HeaderDataError
+from nibabel.tmpdirs import InTemporaryDirectory
 from ..freesurfer import load as mghload
 
 from .test_arraywriters import rt_err_estimate, IUINT_TYPES
@@ -35,7 +36,7 @@
 from nose.tools import (assert_true, assert_false, assert_equal,
                         assert_raises)
 
-from ..testing import data_path, suppress_warnings
+from ..testing import data_path, suppress_warnings, runif_extra_has
 
 from . import test_analyze as tana
 from . import test_spm99analyze as tspm
@@ -1244,6 +1245,7 @@ def test_rt_bias(self):
                 assert_true(np.abs(bias) < bias_thresh)
 
 
+@runif_extra_has('slow')
 def test_large_nifti1():
     image_shape = (91, 109, 91, 1200)
     img = Nifti1Image(np.zeros(image_shape, dtype=np.float32),
@@ -1252,5 +1254,5 @@ def test_large_nifti1():
         img.to_filename('test.nii.gz')
         del img
         data = load('test.nii.gz').get_data()
-    assert_equal(np.asarray(image_shape), data.shape)
+    assert_array_equal(np.asarray(image_shape), data.shape)
     assert_true(np.all(data == 0.))