From 28061d80c29d343fce9bbc1322017ee068447ffc Mon Sep 17 00:00:00 2001 From: Paul McCarthy <pauldmccarthy@gmail.com> Date: Thu, 22 Nov 2018 14:51:18 +0000 Subject: [PATCH] TEST: Test fixExt --- tests/test_image.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/test_image.py b/tests/test_image.py index d468b4b76..2786cc501 100644 --- a/tests/test_image.py +++ b/tests/test_image.py @@ -504,6 +504,31 @@ def test_defaultExt(): assert fslimage.defaultExt() == e +def test_fixExt(): + with tempdir(): + + # error if if file doesn't exist + with pytest.raises(fslpath.PathError): + fslimage.fixExt('file.nii.gz') + + with open('file.nii', 'w') as f: + f.write('1') + assert fslimage.fixExt('file.nii.gz') == 'file.nii' + assert fslimage.fixExt('file.nii') == 'file.nii' + + with open('file.nii.gz', 'w') as f: + f.write('1') + + assert fslimage.fixExt('file.nii.gz') == 'file.nii.gz' + assert fslimage.fixExt('file.nii') == 'file.nii' + + os.remove('file.nii') + os.remove('file.nii.gz') + with open('file.nii.gz', 'w') as f: + f.write('1') + assert fslimage.fixExt('file.nii') == 'file.nii.gz' + + def test_Image_orientation_analyze_neuro(): _test_Image_orientation(0, 'neuro') def test_Image_orientation_analyze_radio(): _test_Image_orientation(0, 'radio') def test_Image_orientation_nifti1_neuro(): _test_Image_orientation(1, 'neuro') -- GitLab