From fd1ea39f123f997418b1a9bb3f494eec5822bdb5 Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Wed, 28 Nov 2018 12:13:48 +0000
Subject: [PATCH] TEST: Expand Image.resample test
---
tests/test_image.py | 42 ++++++++++++++++++++++++++++++++++++++++--
1 file changed, 40 insertions(+), 2 deletions(-)
diff --git a/tests/test_image.py b/tests/test_image.py
index 2786cc501..3290ed361 100644
--- a/tests/test_image.py
+++ b/tests/test_image.py
@@ -1078,6 +1078,12 @@ def test_image_resample(seed):
make_random_image(fname, shape)
img = fslimage.Image(fname, mmap=False)
+ # bad shape
+ with pytest.raises(ValueError):
+ img.resample((10, 10))
+ with pytest.raises(ValueError):
+ img.resample((10, 10, 10, 10))
+
# resampling to the same shape should be a no-op
samei, samex = img.resample(shape)
assert np.all(samei == img[:])
@@ -1134,18 +1140,50 @@ def test_image_resample(seed):
assert np.all(np.isclose(resvals, origvals))
- # Test a 4D image
+ del img
+ img = None
+
+
+def test_image_resample_4d(seed):
+
+ fname = 'test.nii.gz'
+
+ with tempdir():
+
make_random_image(fname, (10, 10, 10, 10))
+
+ # resample one volume
img = fslimage.Image(fname)
slc = (slice(None), slice(None), slice(None), 3)
-
resampled = img.resample(img.shape[:3], slc)[0]
assert np.all(resampled == img[..., 3])
+ # resample up
resampled = img.resample((15, 15, 15), slc)[0]
assert tuple(resampled.shape) == (15, 15, 15)
+
+ # resample down
+ resampled = img.resample((5, 5, 5), slc)[0]
+ assert tuple(resampled.shape) == (5, 5, 5)
+
+ # resample the entire image
+ resampled = img.resample((15, 15, 15, 10), None)[0]
+ assert tuple(resampled.shape) == (15, 15, 15, 10)
+
+ resampled = img.resample((5, 5, 5, 10), None)[0]
+ assert tuple(resampled.shape) == (5, 5, 5, 10)
+
+ # resample along the fourth dim
+ resampled = img.resample((15, 15, 15, 15), None)[0]
+ assert tuple(resampled.shape) == (15, 15, 15, 15)
+
+ resampled = img.resample((5, 5, 5, 15), None)[0]
+ assert tuple(resampled.shape) == (5, 5, 5, 15)
+
del img
+ del resampled
img = None
+ resampled = None
def test_Image_init_xform_nifti1(): _test_Image_init_xform(1)
--
GitLab