def test_04_perfect_overlap_5mm_margin(self): _, tumor_np = niftireader.load_image( _get_file_name('T01', '04_perfect_overlap_5mm_margin', 'Tumor')) _, ablation_np = niftireader.load_image( _get_file_name('T01', '04_perfect_overlap_5mm_margin', 'Ablation')) distances = margin.compute_distances(tumor_np, ablation_np, None, 1) df = margin.summarize_surface_dists('T01', 'L01', distances) print(df) record = df.loc[(df['Patient'] == "T01") & (df['Lesion'] == 'L01')].iloc[0] self.assertEqual(record["nr_voxels"], 222.0) self.assertAlmostEqual(record["min_distance"], 4.12, delta=0.01) self.assertAlmostEqual(record["max_distance"], 5.0, delta=0.01)
def test_03_n5mm_margin_subcapsular(self): _, tumor_np = niftireader.load_image( _get_file_name('T02', '03_-5mm_margin_subcapsular', 'Tumor')) _, ablation_np = niftireader.load_image( _get_file_name('T02', '03_-5mm_margin_subcapsular', 'Ablation')) _, liver_np = niftireader.load_image( _get_file_name('T02', '03_-5mm_margin_subcapsular', 'Liver')) distances = margin.compute_distances(tumor_np, ablation_np, liver_np, 1) df = margin.summarize_surface_dists('T02', 'L03', distances) print(df) record = df.loc[(df['Patient'] == "T02") & (df['Lesion'] == 'L03')].iloc[0] self.assertEqual(record["nr_voxels"], 317.0) self.assertAlmostEqual(record["min_distance"], -5.09, delta=0.01) self.assertAlmostEqual(record["max_distance"], -2.23, delta=0.01)