@@ -206,9 +206,6 @@ def test_datamodule_description():
206
206
207
207
@patch ("matplotlib.pyplot.show" )
208
208
def test_hemor_contrast (mock_show , rtk_data , tmp_path_factory ):
209
- # target_dir = "./data/rtk/infer"
210
- # if os.path.exists(target_dir):
211
- # shutil.rmtree(target_dir)
212
209
out_ = str (tmp_path_factory .mktemp ("out" ))
213
210
hemorrhage_contrast (input_path = str (rtk_data ), output_folder = out_ )
214
211
content = os .listdir (out_ )
@@ -226,20 +223,17 @@ def test_hemor_contrast(mock_show, rtk_data, tmp_path_factory):
226
223
def test_segm_detection_pipeline_metrics (
227
224
mock_show , tmp_path_factory , task , rtk_data , rtk_downloader
228
225
):
229
-
230
226
path = {"source" : rtk_segm , "target" : rtk_data }
231
227
dm = DicomCocoDataModuleRTK (infer = path , transform = resize_transform )
232
228
dm .setup_infer ()
233
229
ds = dm .predict_dataloader ()
234
230
ds .transform = resize_transform
235
231
236
232
samples_number = len (ds )
237
-
238
233
out_dir = tmp_path_factory .mktemp ("out" )
239
234
for i in range (samples_number ):
240
235
random_numpy = np .random .randint (0 , 1 , [256 , 256 , 1 ])
241
236
np .save (os .path .join (out_dir , f"{ i } .npy" ), random_numpy )
242
-
243
237
process_metrics (input_path = rtk_data , output_folder = out_dir )
244
238
assert mock_show .call_count > 0
245
239
assert rtk_downloader .call_count > 0
0 commit comments