def test_rotate(self): # image rotated 9 degrees in first axis input_img = nbload(os.path.join(test_data_directory, 'img_rot.nii')) base_img = nbload(os.path.join(test_data_directory, 'img.nii')) registered_img, xfm_pred = transform(input_img, base_img, output_transform=True) xfm_true = np.linalg.inv(load_afni_xfm(os.path.join(test_data_directory, 'rotmat.aff12.1D'))) self.assertTrue(np.allclose(xfm_true, xfm_pred, rtol=0.01, atol=0.01))
def test_motion_correct_shift(self): input_img = nbload(os.path.join(test_data_directory, 'img.nii')) # base image is identical base_img = nbload(os.path.join(test_data_directory, 'img.nii')) base_affine = base_img.affine[:] base_affine[0,3] += 10 output_img = transform(input_img, Nifti1Image(base_img.get_data(), base_affine)) self.assertTrue(np.allclose(input_img.get_data(), output_img.get_data()))
def test_transform_identity(self): # load test image input_img = nbload(os.path.join(test_data_directory, 'img.nii')) # base image is identical base_path = os.path.join(test_data_directory, 'img.nii') output_img = transform(input_img, base_path) self.assertTrue(np.allclose(input_img.get_data(), output_img.get_data())) self.assertTrue(isinstance(output_img, Nifti1Image))
