TEST: add tests to check that warnings are raised

Author: Michiel Cottaar

nibabel/cifti2/tests/test_cifti2.py

@@ -7,7 +7,7 @@
 
 from nibabel import cifti2 as ci
 from nibabel.nifti2 import Nifti2Header
-from nibabel.cifti2.cifti2 import _float_01, _value_if_klass, Cifti2HeaderError, Cifti2NamedMap, Cifti2MatrixIndicesMap
+from nibabel.cifti2.cifti2 import _float_01, _value_if_klass, Cifti2HeaderError
 
 from nose.tools import assert_true, assert_equal, assert_raises, assert_is_none
 
@@ -359,7 +359,7 @@ class TestCifti2ImageAPI(_TDA):
 
     def make_imaker(self, arr, header=None, ni_header=None):
         for idx, sz in enumerate(arr.shape):
-            maps = [Cifti2NamedMap(str(value)) for value in range(sz)]
+            maps = [ci.Cifti2NamedMap(str(value)) for value in range(sz)]
             mim = ci.Cifti2MatrixIndicesMap(
                 (idx, ), 'CIFTI_INDEX_TYPE_SCALARS', maps=maps
             )

nibabel/cifti2/tests/test_new_cifti2.py

@@ -13,6 +13,7 @@
 from nibabel.tmpdirs import InTemporaryDirectory
 
 from nose.tools import assert_true, assert_equal, assert_raises
+from nibabel.testing import clear_and_catch_warnings, error_warnings, suppress_warnings
 
 affine = [[-1.5, 0, 0, 90],
           [0, 1.5, 0, -85],
@@ -516,6 +517,10 @@ def test_wrong_shape():
         np.random.randn(3, 10),
         np.random.randn(2, 9),
     ):
-        img = ci.Cifti2Image(data, hdr)
+        with clear_and_catch_warnings():
+            with error_warnings():
+                assert_raises(UserWarning, ci.Cifti2Image, data, hdr)
+        with suppress_warnings():
+            img = ci.Cifti2Image(data, hdr)
         assert_raises(ValueError, img.to_file_map)