@@ -54,8 +54,6 @@ def get_data(self, _obj):
5454
5555TEST_CASE_1 = [{}, ["test_image.nii.gz" ], (128 , 128 , 128 )]
5656
57- TEST_CASE_2 = [{}, ["test_image.nii.gz" ], (128 , 128 , 128 )]
58-
5957TEST_CASE_3 = [{}, ["test_image.nii.gz" , "test_image2.nii.gz" , "test_image3.nii.gz" ], (3 , 128 , 128 , 128 )]
6058
6159TEST_CASE_3_1 = [ # .mgz format
@@ -64,8 +62,6 @@ def get_data(self, _obj):
6462 (3 , 128 , 128 , 128 ),
6563]
6664
67- TEST_CASE_4 = [{}, ["test_image.nii.gz" , "test_image2.nii.gz" , "test_image3.nii.gz" ], (3 , 128 , 128 , 128 )]
68-
6965TEST_CASE_4_1 = [ # additional parameter
7066 {"mmap" : False },
7167 ["test_image.nii.gz" , "test_image2.nii.gz" , "test_image3.nii.gz" ],
@@ -92,8 +88,6 @@ def get_data(self, _obj):
9288
9389TEST_CASE_6 = [{"reader" : ITKReader () if has_itk else "itkreader" }, ["test_image.nii.gz" ], (128 , 128 , 128 )]
9490
95- TEST_CASE_7 = [{"reader" : ITKReader () if has_itk else "itkreader" }, ["test_image.nii.gz" ], (128 , 128 , 128 )]
96-
9791TEST_CASE_8 = [
9892 {"reader" : ITKReader () if has_itk else "itkreader" },
9993 ["test_image.nii.gz" , "test_image2.nii.gz" , "test_image3.nii.gz" ],
@@ -106,12 +100,6 @@ def get_data(self, _obj):
106100 (384 , 128 , 128 ),
107101]
108102
109- TEST_CASE_9 = [
110- {"reader" : ITKReader () if has_itk else "itkreader" },
111- ["test_image.nii.gz" , "test_image2.nii.gz" , "test_image3.nii.gz" ],
112- (3 , 128 , 128 , 128 ),
113- ]
114-
115103TEST_CASE_10 = [
116104 {"reader" : ITKReader (pixel_type = itk_uc ) if has_itk else "itkreader" },
117105 "tests/testing_data/CT_DICOM" ,
@@ -205,9 +193,7 @@ def tearDownClass(cls):
205193 shutil .rmtree (cls .tmpdir )
206194 super ().tearDownClass ()
207195
208- @parameterized .expand (
209- [TEST_CASE_1 , TEST_CASE_2 , TEST_CASE_3 , TEST_CASE_3_1 , TEST_CASE_4 , TEST_CASE_4_1 , TEST_CASE_5 ]
210- )
196+ @parameterized .expand ([TEST_CASE_1 , TEST_CASE_3 , TEST_CASE_3_1 , TEST_CASE_4_1 , TEST_CASE_5 ])
211197 def test_nibabel_reader (self , input_param , filenames , expected_shape ):
212198 test_image = np .random .rand (128 , 128 , 128 )
213199 with tempfile .TemporaryDirectory () as tempdir :
@@ -249,7 +235,7 @@ def test_nibabel_reader_gpu(self, input_param, filenames, expected_shape):
249235 result_cpu = LoadImage (image_only = True , ** input_param_cpu )(filenames )
250236 assert_allclose (result_cpu , result .cpu (), atol = 1e-6 )
251237
252- @parameterized .expand ([TEST_CASE_6 , TEST_CASE_7 , TEST_CASE_8 , TEST_CASE_8_1 , TEST_CASE_9 ])
238+ @parameterized .expand ([TEST_CASE_6 , TEST_CASE_8 , TEST_CASE_8_1 ])
253239 def test_itk_reader (self , input_param , filenames , expected_shape ):
254240 test_image = torch .randint (0 , 256 , (128 , 128 , 128 ), dtype = torch .uint8 ).numpy ()
255241 print ("Test image value range:" , test_image .min (), test_image .max ())
0 commit comments