From f1d98a686af8916b7080742417929d9c81244cba Mon Sep 17 00:00:00 2001 From: robert Date: Thu, 29 Aug 2024 12:28:09 +0200 Subject: [PATCH] update test --- unit_tests/test_semantic.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/unit_tests/test_semantic.py b/unit_tests/test_semantic.py index e7b3176..206986b 100644 --- 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):