add tests for conform and _transform_range

Author: Jakub Kaczmarzyk

nibabel/tests/test_processing.py

@@ -19,10 +19,11 @@
 
 import nibabel as nib
 from nibabel.processing import (sigma2fwhm, fwhm2sigma, adapt_affine,
-                                resample_from_to, resample_to_output, smooth_image)
+                                resample_from_to, resample_to_output, smooth_image,
+                                _transform_range, conform)
 from nibabel.nifti1 import Nifti1Image
 from nibabel.nifti2 import Nifti2Image
-from nibabel.orientations import flip_axis, inv_ornt_aff
+from nibabel.orientations import aff2axcodes, flip_axis, inv_ornt_aff
 from nibabel.affines import (AffineError, from_matvec, to_matvec, apply_affine,
                              voxel_sizes)
 from nibabel.eulerangles import euler2mat
@@ -426,3 +427,32 @@ def test_against_spm_resample():
     moved2output = resample_to_output(moved_anat, 4, order=1, cval=np.nan)
     spm2output = nib.load(pjoin(DATA_DIR, 'reoriented_anat_moved.nii'))
     assert_spm_resampling_close(moved_anat, moved2output, spm2output);
+
+
+def test__transform_range():
+    assert_array_equal(_transform_range([2, 4, 6], -1, 1), [-1, 0, 1])
+    assert_array_equal(_transform_range([-1, 0, 1], 2, 6), [2, 4, 6])
+    assert_array_equal(_transform_range(np.arange(11), 0, 5),
+                       np.arange(0, 5.5, 0.5))
+    assert_array_equal(_transform_range(np.arange(-100, 101), 0, 200),
+                       np.arange(201))
+
+
+def test_conform():
+    anat = nib.load(pjoin(DATA_DIR, 'anatomical.nii'))
+
+    c = conform(anat)
+    assert c.shape == (256, 256, 256)
+    assert c.header.get_zooms() == (1, 1, 1)
+    assert c.dataobj.dtype == np.dtype(np.uint8)
+    assert aff2axcodes(c.affine) == ('R', 'A', 'S')
+
+    c = conform(anat, out_shape=(100, 100, 200), voxel_size=(2, 2, 1.5))
+    assert c.shape == (100, 100, 200)
+    assert c.header.get_zooms() == (2, 2, 1.5)
+    assert c.dataobj.dtype == np.dtype(np.uint8)
+    assert aff2axcodes(c.affine) == ('R', 'A', 'S')
+
+    # Error on non-3D images.
+    func = nib.load(pjoin(DATA_DIR, 'functional.nii'))
+    assert_raises(ValueError, conform, func)