PY36: Backport nullcontext

Author: effigies

nibabel/testing/__init__.py

@@ -24,6 +24,14 @@
 
 from itertools import zip_longest
 
+try:
+    from contextlib import nullcontext
+except ImportError:  # PY36
+    from contextlib import contextmanager
+    @contextmanager
+    def nullcontext():
+        yield
+
 
 def test_data(subdir=None, fname=None):
     if subdir is None:

nibabel/tests/test_image_api.py

@@ -26,7 +26,6 @@
 import warnings
 from functools import partial
 from itertools import product
-from contextlib import nullcontext
 import pathlib
 
 import numpy as np
@@ -46,8 +45,8 @@
 import pytest
 
 from numpy.testing import assert_almost_equal, assert_array_equal, assert_warns, assert_allclose
-from ..testing import (bytesio_round_trip, bytesio_filemap,
-                       assert_data_similar, clear_and_catch_warnings)
+from nibabel.testing import (bytesio_round_trip, bytesio_filemap, assert_data_similar,
+                             clear_and_catch_warnings, nullcontext)
 from ..tmpdirs import InTemporaryDirectory
 from ..deprecator import ExpiredDeprecationError
 

nibabel/tests/test_volumeutils.py

@@ -20,7 +20,6 @@
 import bz2
 import threading
 import time
-from contextlib import nullcontext
 
 import numpy as np
 
@@ -58,7 +57,7 @@
                            assert_array_equal)
 import pytest
 
-from ..testing import assert_dt_equal, assert_allclose_safely, suppress_warnings
+from nibabel.testing import nullcontext, assert_dt_equal, assert_allclose_safely, suppress_warnings
 
 #: convenience variables for numpy types
 FLOAT_TYPES = np.sctypes['float']