def test_save_load(): shape = (2, 4, 6) npt = np.float32 data = np.arange(np.prod(shape), dtype=npt).reshape(shape) affine = np.diag([1, 2, 3, 1]) affine[:3,3] = [3,2,1] img = ni1.Nifti1Image(data, affine) img.set_data_dtype(npt) with InTemporaryDirectory() as pth: pth = mkdtemp() nifn = pjoin(pth, 'an_image.nii') sifn = pjoin(pth, 'another_image.img') ni1.save(img, nifn) re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) # These and subsequent del statements are to prevent confusing # windows errors when trying to open files or delete the # temporary directory. del re_img try: import scipy.io except ImportError: # ignore if there is no matfile reader, and restart pass else: spm2.save(img, sifn) re_img2 = nils.load(sifn) yield assert_true(isinstance(re_img2, spm2.Spm2AnalyzeImage)) yield assert_array_equal(re_img2.get_data(), data) yield assert_array_equal(re_img2.get_affine(), affine) del re_img2 spm99.save(img, sifn) re_img3 = nils.load(sifn) yield assert_true(isinstance(re_img3, spm99.Spm99AnalyzeImage)) yield assert_array_equal(re_img3.get_data(), data) yield assert_array_equal(re_img3.get_affine(), affine) ni1.save(re_img3, nifn) del re_img3 re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) del re_img
def test_save_load(): shape = (2, 4, 6) npt = np.float32 data = np.arange(np.prod(shape), dtype=npt).reshape(shape) affine = np.diag([1, 2, 3, 1]) affine[:3, 3] = [3, 2, 1] img = ni1.Nifti1Image(data, affine) img.set_data_dtype(npt) with InTemporaryDirectory() as pth: pth = mkdtemp() nifn = pjoin(pth, 'an_image.nii') sifn = pjoin(pth, 'another_image.img') ni1.save(img, nifn) re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) # These and subsequent del statements are to prevent confusing # windows errors when trying to open files or delete the # temporary directory. del re_img try: import scipy.io except ImportError: # ignore if there is no matfile reader, and restart pass else: spm2.save(img, sifn) re_img2 = nils.load(sifn) yield assert_true(isinstance(re_img2, spm2.Spm2AnalyzeImage)) yield assert_array_equal(re_img2.get_data(), data) yield assert_array_equal(re_img2.get_affine(), affine) del re_img2 spm99.save(img, sifn) re_img3 = nils.load(sifn) yield assert_true(isinstance(re_img3, spm99.Spm99AnalyzeImage)) yield assert_array_equal(re_img3.get_data(), data) yield assert_array_equal(re_img3.get_affine(), affine) ni1.save(re_img3, nifn) del re_img3 re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) del re_img
def test_save_load(): shape = (2, 4, 6) npt = np.float32 data = np.arange(np.prod(shape), dtype=npt).reshape(shape) affine = np.diag([1, 2, 3, 1]) affine[:3,3] = [3,2,1] img = ni1.Nifti1Image(data, affine) img.set_data_dtype(npt) with InTemporaryDirectory() as pth: nifn = 'an_image.nii' sifn = 'another_image.img' ni1.save(img, nifn) re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) # These and subsequent del statements are to prevent confusing # windows errors when trying to open files or delete the # temporary directory. del re_img if have_scipy: # skip we we cannot read .mat files spm2.save(img, sifn) re_img2 = nils.load(sifn) yield assert_true(isinstance(re_img2, spm2.Spm2AnalyzeImage)) yield assert_array_equal(re_img2.get_data(), data) yield assert_array_equal(re_img2.get_affine(), affine) del re_img2 spm99.save(img, sifn) re_img3 = nils.load(sifn) yield assert_true(isinstance(re_img3, spm99.Spm99AnalyzeImage)) yield assert_array_equal(re_img3.get_data(), data) yield assert_array_equal(re_img3.get_affine(), affine) ni1.save(re_img3, nifn) del re_img3 re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) del re_img
def test_save_load(): shape = (2, 4, 6) npt = np.float32 data = np.arange(np.prod(shape), dtype=npt).reshape(shape) affine = np.diag([1, 2, 3, 1]) affine[:3, 3] = [3, 2, 1] img = ni1.Nifti1Image(data, affine) img.set_data_dtype(npt) with InTemporaryDirectory() as pth: nifn = 'an_image.nii' sifn = 'another_image.img' ni1.save(img, nifn) re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) # These and subsequent del statements are to prevent confusing # windows errors when trying to open files or delete the # temporary directory. del re_img if have_scipy: # skip we we cannot read .mat files spm2.save(img, sifn) re_img2 = nils.load(sifn) yield assert_true(isinstance(re_img2, spm2.Spm2AnalyzeImage)) yield assert_array_equal(re_img2.get_data(), data) yield assert_array_equal(re_img2.get_affine(), affine) del re_img2 spm99.save(img, sifn) re_img3 = nils.load(sifn) yield assert_true(isinstance(re_img3, spm99.Spm99AnalyzeImage)) yield assert_array_equal(re_img3.get_data(), data) yield assert_array_equal(re_img3.get_affine(), affine) ni1.save(re_img3, nifn) del re_img3 re_img = nils.load(nifn) yield assert_true(isinstance(re_img, ni1.Nifti1Image)) yield assert_array_equal(re_img.get_data(), data) yield assert_array_equal(re_img.get_affine(), affine) del re_img