diff --git a/tests/test_image.py b/tests/test_image.py index 9686f8739d7aacaf94898f80a433affc9d094311..e12c6acb7dd2b829a588b13551949e83d4bb9e43 100644 --- a/tests/test_image.py +++ b/tests/test_image.py @@ -1098,14 +1098,16 @@ def _test_Image_save(imgtype): # Load the image back in img2 = fslimage.Image(img.dataSource) - assert img.saveState - assert img.dataSource == expDataSource + assert img2.saveState + assert img2.dataSource == expDataSource - if imgtype > 0: - assert np.all(np.isclose(img.voxToWorldMat, xform)) + for i in (img, img2): + if imgtype > 0: + assert np.all(np.isclose(i.voxToWorldMat, xform)) - for (x, y, z), v in zip(rvoxs, rvals): - assert np.isclose(img[x, y, z], v) + for (x, y, z), v in zip(rvoxs, rvals): + assert np.isclose(i[x, y, z], v) + img = None img2 = None