diff --git a/tests/test_image.py b/tests/test_image.py index e771f963679e04f74f7a874a1779e1f31b52f1f6..8f1459c8e5aeb2b9e2c8634770012fb379f59a1e 100644 --- a/tests/test_image.py +++ b/tests/test_image.py @@ -13,7 +13,7 @@ import glob import numpy as np import nibabel as nib -from nibabel.spatialimages import HeaderDataError +from nibabel.spatialimages import ImageFileError import fsl.data.constants as constants import fsl.data.image as fslimage @@ -49,8 +49,8 @@ def test_load(testdir): shouldRaise = [('notexist', fslpath.PathError), ('notexist.nii.gz', fslpath.PathError), ('ambiguous', fslpath.PathError), - ('notnifti', HeaderDataError), - ('notnifti.nii.gz', HeaderDataError)] + ('notnifti', ImageFileError), + ('notnifti.nii.gz', ImageFileError)] # Not raising an error means the test passes for fname in shouldPass: diff --git a/tests/test_immv_imcp.py b/tests/test_immv_imcp.py index 414f58c52294c8bdf56d052bf2723b5abe750e34..8fcb62751c967128affe04489c02f8b96142f7b2 100644 --- a/tests/test_immv_imcp.py +++ b/tests/test_immv_imcp.py @@ -16,6 +16,8 @@ import tempfile import numpy as np import nibabel as nib +from nibabel.spatialimages import ImageFileError + import pytest import fsl.utils.path as fslpath @@ -454,7 +456,7 @@ def test_imcp_script_shouldFail(move=False): if move: immv_script.main(imcp_args) else: imcp_script.main(imcp_args) assert False - except (RuntimeError, IOError, fslpath.PathError): + except (RuntimeError, IOError, fslpath.PathError, ImageFileError): pass sp.call('chmod u+rwx {}'.format(indir) .split())