consistent affien pydicom/itk backend test_load_image.py

Project-MONAI · Sep 5, 2023 · b5038a3 · b5038a3
1 parent 26e8913
commit b5038a3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_load_image.py b/tests/test_load_image.py
@@ -276,8 +276,7 @@ def test_dicom_reader_consistency_single(self):
             itk_result = LoadImage(image_only=True, **itk_param)(self.data_dir)
             pydicom_result = LoadImage(image_only=True, **pydicom_param)(self.data_dir)
             np.testing.assert_allclose(pydicom_result, itk_result.squeeze())
-            np.testing.assert_allclose(pydicom_result.affine[:2, :2], itk_result.affine[:2, :2])
-            np.testing.assert_allclose(pydicom_result.affine[..., 3], itk_result.affine[..., 3])
+            np.testing.assert_allclose(pydicom_result.affine, itk_result.affine)
 
     def test_load_nifti_multichannel(self):
         test_image = np.random.randint(0, 256, size=(31, 64, 16, 2)).astype(np.float32)