@@ -81,16 +81,16 @@ def _assert_metrics(self,
8181 places = places )
8282
8383 @parameterized .parameters ((
84- np .zeros ([2 , 2 , 2 ], dtype = np . bool ),
85- np .zeros ([2 , 2 ], dtype = np . bool ),
84+ np .zeros ([2 , 2 , 2 ], dtype = bool ),
85+ np .zeros ([2 , 2 ], dtype = bool ),
8686 [1 , 1 ],
8787 ), (
88- np .zeros ([2 , 2 ], dtype = np . bool ),
89- np .zeros ([2 , 2 , 2 ], dtype = np . bool ),
88+ np .zeros ([2 , 2 ], dtype = bool ),
89+ np .zeros ([2 , 2 , 2 ], dtype = bool ),
9090 [1 , 1 ],
9191 ), (
92- np .zeros ([2 , 2 ], dtype = np . bool ),
93- np .zeros ([2 , 2 ], dtype = np . bool ),
92+ np .zeros ([2 , 2 ], dtype = bool ),
93+ np .zeros ([2 , 2 ], dtype = bool ),
9494 [1 , 1 , 1 ],
9595 ))
9696 def test_compute_surface_distances_raises_on_incompatible_shapes (
@@ -100,12 +100,12 @@ def test_compute_surface_distances_raises_on_incompatible_shapes(
100100 surface_distance .compute_surface_distances (mask_gt , mask_pred , spacing_mm )
101101
102102 @parameterized .parameters ((
103- np .zeros ([2 ], dtype = np . bool ),
104- np .zeros ([2 ], dtype = np . bool ),
103+ np .zeros ([2 ], dtype = bool ),
104+ np .zeros ([2 ], dtype = bool ),
105105 [1 ],
106106 ), (
107- np .zeros ([2 , 2 , 2 , 2 ], dtype = np . bool ),
108- np .zeros ([2 , 2 , 2 , 2 ], dtype = np . bool ),
107+ np .zeros ([2 , 2 , 2 , 2 ], dtype = bool ),
108+ np .zeros ([2 , 2 , 2 , 2 ], dtype = bool ),
109109 [1 , 1 , 1 , 1 ],
110110 ))
111111 def test_compute_surface_distances_raises_on_invalid_shapes (
@@ -118,8 +118,8 @@ def test_compute_surface_distances_raises_on_invalid_shapes(
118118class SurfaceDistance2DTest (SurfaceDistanceTest , parameterized .TestCase ):
119119
120120 def test_on_2_pixels_2mm_away (self ):
121- mask_gt = np .zeros ((128 , 128 ), np . bool )
122- mask_pred = np .zeros ((128 , 128 ), np . bool )
121+ mask_gt = np .zeros ((128 , 128 ), bool )
122+ mask_pred = np .zeros ((128 , 128 ), bool )
123123 mask_gt [50 , 70 ] = 1
124124 mask_pred [50 , 72 ] = 1
125125 surface_distances = surface_distance .compute_surface_distances (
@@ -160,7 +160,7 @@ def test_two_squares_shifted_by_one_pixel(self):
160160 [0 , 0 , 0 , 0 , 0 , 0 ],
161161 [0 , 0 , 0 , 0 , 0 , 0 ],
162162 ],
163- dtype = np . bool )
163+ dtype = bool )
164164
165165 mask_pred = np .asarray (
166166 [
@@ -171,7 +171,7 @@ def test_two_squares_shifted_by_one_pixel(self):
171171 [0 , 0 , 0 , 0 , 0 , 0 ],
172172 [0 , 0 , 0 , 0 , 0 , 0 ],
173173 ],
174- dtype = np . bool )
174+ dtype = bool )
175175
176176 vertical = 2
177177 horizontal = 1
@@ -236,8 +236,8 @@ def test_two_squares_shifted_by_one_pixel(self):
236236 mask_gt , mask_pred )))
237237
238238 def test_empty_prediction_mask (self ):
239- mask_gt = np .zeros ((128 , 128 ), np . bool )
240- mask_pred = np .zeros ((128 , 128 ), np . bool )
239+ mask_gt = np .zeros ((128 , 128 ), bool )
240+ mask_pred = np .zeros ((128 , 128 ), bool )
241241 mask_gt [50 , 60 ] = 1
242242 surface_distances = surface_distance .compute_surface_distances (
243243 mask_gt , mask_pred , spacing_mm = (3 , 2 ))
@@ -253,8 +253,8 @@ def test_empty_prediction_mask(self):
253253 expected_volumetric_dice = 0.0 )
254254
255255 def test_empty_ground_truth_mask (self ):
256- mask_gt = np .zeros ((128 , 128 ), np . bool )
257- mask_pred = np .zeros ((128 , 128 ), np . bool )
256+ mask_gt = np .zeros ((128 , 128 ), bool )
257+ mask_pred = np .zeros ((128 , 128 ), bool )
258258 mask_pred [50 , 60 ] = 1
259259 surface_distances = surface_distance .compute_surface_distances (
260260 mask_gt , mask_pred , spacing_mm = (3 , 2 ))
@@ -270,8 +270,8 @@ def test_empty_ground_truth_mask(self):
270270 expected_volumetric_dice = 0.0 )
271271
272272 def test_both_empty_masks (self ):
273- mask_gt = np .zeros ((128 , 128 ), np . bool )
274- mask_pred = np .zeros ((128 , 128 ), np . bool )
273+ mask_gt = np .zeros ((128 , 128 ), bool )
274+ mask_pred = np .zeros ((128 , 128 ), bool )
275275 surface_distances = surface_distance .compute_surface_distances (
276276 mask_gt , mask_pred , spacing_mm = (3 , 2 ))
277277 self ._assert_metrics (
@@ -289,8 +289,8 @@ def test_both_empty_masks(self):
289289class SurfaceDistance3DTest (SurfaceDistanceTest ):
290290
291291 def test_on_2_pixels_2mm_away (self ):
292- mask_gt = np .zeros ((128 , 128 , 128 ), np . bool )
293- mask_pred = np .zeros ((128 , 128 , 128 ), np . bool )
292+ mask_gt = np .zeros ((128 , 128 , 128 ), bool )
293+ mask_pred = np .zeros ((128 , 128 , 128 ), bool )
294294 mask_gt [50 , 60 , 70 ] = 1
295295 mask_pred [50 , 60 , 72 ] = 1
296296 surface_distances = surface_distance .compute_surface_distances (
@@ -304,8 +304,8 @@ def test_on_2_pixels_2mm_away(self):
304304 expected_volumetric_dice = 0.0 )
305305
306306 def test_two_cubes_shifted_by_one_pixel (self ):
307- mask_gt = np .zeros ((100 , 100 , 100 ), np . bool )
308- mask_pred = np .zeros ((100 , 100 , 100 ), np . bool )
307+ mask_gt = np .zeros ((100 , 100 , 100 ), bool )
308+ mask_pred = np .zeros ((100 , 100 , 100 ), bool )
309309 mask_gt [0 :50 , :, :] = 1
310310 mask_pred [0 :51 , :, :] = 1
311311 surface_distances = surface_distance .compute_surface_distances (
@@ -320,8 +320,8 @@ def test_two_cubes_shifted_by_one_pixel(self):
320320 expected_volumetric_dice = 0.990 )
321321
322322 def test_empty_prediction_mask (self ):
323- mask_gt = np .zeros ((128 , 128 , 128 ), np . bool )
324- mask_pred = np .zeros ((128 , 128 , 128 ), np . bool )
323+ mask_gt = np .zeros ((128 , 128 , 128 ), bool )
324+ mask_pred = np .zeros ((128 , 128 , 128 ), bool )
325325 mask_gt [50 , 60 , 70 ] = 1
326326 surface_distances = surface_distance .compute_surface_distances (
327327 mask_gt , mask_pred , spacing_mm = (3 , 2 , 1 ))
@@ -335,8 +335,8 @@ def test_empty_prediction_mask(self):
335335 expected_volumetric_dice = 0.0 )
336336
337337 def test_empty_ground_truth_mask (self ):
338- mask_gt = np .zeros ((128 , 128 , 128 ), np . bool )
339- mask_pred = np .zeros ((128 , 128 , 128 ), np . bool )
338+ mask_gt = np .zeros ((128 , 128 , 128 ), bool )
339+ mask_pred = np .zeros ((128 , 128 , 128 ), bool )
340340 mask_pred [50 , 60 , 72 ] = 1
341341 surface_distances = surface_distance .compute_surface_distances (
342342 mask_gt , mask_pred , spacing_mm = (3 , 2 , 1 ))
@@ -350,8 +350,8 @@ def test_empty_ground_truth_mask(self):
350350 expected_volumetric_dice = 0.0 )
351351
352352 def test_both_empty_masks (self ):
353- mask_gt = np .zeros ((128 , 128 , 128 ), np . bool )
354- mask_pred = np .zeros ((128 , 128 , 128 ), np . bool )
353+ mask_gt = np .zeros ((128 , 128 , 128 ), bool )
354+ mask_pred = np .zeros ((128 , 128 , 128 ), bool )
355355 surface_distances = surface_distance .compute_surface_distances (
356356 mask_gt , mask_pred , spacing_mm = (3 , 2 , 1 ))
357357 self ._assert_metrics (
0 commit comments