update test

Hendrik-code · Aug 29, 2024 · f1d98a6 · f1d98a6
1 parent 65c2249
commit f1d98a6
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/unit_tests/test_semantic.py b/unit_tests/test_semantic.py
@@ -33,7 +33,7 @@ def __init__(
             acquisition="sag",
             log_name="DummySegModel",
             modeltype="unet",
-            model_expected_orientation=["P", "I", "R"],
+            model_expected_orientation=("P", "I", "R"),
             available_folds=1,
             inference_augmentation=False,
             resolution_range=[0.75, 0.75, 1.65],
@@ -65,10 +65,7 @@ def test_phase_preprocess(self):
             preprossed_input, errcode = preprocess_input(mri, debug_data={}, pad_size=pad_size, verbose=True)
             print(mri)
             print(preprossed_input)
-
-            # backchanged_origin = tuple(preprossed_input.origin[idx] + origin_diff[idx] for idx in range(3))
-            self.assertTrue(preprossed_input.assert_affine(origin=mri.origin, error_tolerance=origin_diff))
-            # affine=mri.affine,
+            self.assertTrue(preprossed_input.assert_affine(origin=mri.origin, origin_tolerance=origin_diff))
             self.assertTrue(preprossed_input.assert_affine(rotation=mri.rotation, orientation=mri.orientation, zoom=mri.zoom))
             self.assertEqual(errcode, ErrCode.OK)
             for idx, s in enumerate(mri.shape):