Skip to content

Commit

Permalink
Add 'ref_count', 'pred_count' to unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@valosekj
valosekj committed Jun 7, 2024
1 parent 4dd0790 commit 3dc77ca
Showing 1 changed file with 23 additions and 8 deletions.
31 changes: 23 additions & 8 deletions test/test_metrics/test_pairwise_measures_neuropoly.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,8 @@
from compute_metrics_reloaded import compute_metrics_single_subject
import tempfile

METRICS = ['dsc', 'fbeta', 'nsd', 'vol_diff', 'rel_vol_error', 'lesion_ppv', 'lesion_sensitivity', 'lesion_f1_score']
METRICS = ['dsc', 'fbeta', 'nsd', 'vol_diff', 'rel_vol_error', 'lesion_ppv', 'lesion_sensitivity', 'lesion_f1_score',
'ref_count', 'pred_count']


class TestComputeMetricsReloaded(unittest.TestCase):
Expand Down Expand Up @@ -65,7 +66,9 @@ def test_empty_ref_and_pred(self):
'vol_diff': np.nan,
'lesion_ppv': 1.0,
'lesion_sensitivity': 1.0,
'lesion_f1_score': 1.0}}
'lesion_f1_score': 1.0,
'ref_count': 0,
'pred_count': 0}}

# Create empty reference
self.create_dummy_nii(self.ref_file, np.zeros((10, 10, 10)))
Expand All @@ -90,7 +93,9 @@ def test_empty_ref(self):
'vol_diff': np.inf,
'lesion_ppv': 0.0,
'lesion_sensitivity': 1.0,
'lesion_f1_score': 0.0}}
'lesion_f1_score': 0.0,
'ref_count': 0,
'pred_count': 1}}

# Create empty reference
self.create_dummy_nii(self.ref_file, np.zeros((10, 10, 10)))
Expand All @@ -117,7 +122,9 @@ def test_empty_pred(self):
'vol_diff': 1.0,
'lesion_ppv': 0.0,
'lesion_sensitivity': 0.0,
'lesion_f1_score': 0.0}}
'lesion_f1_score': 0.0,
'ref_count': 1,
'pred_count': 0}}

# Create non-empty reference
ref = np.zeros((10, 10, 10))
Expand All @@ -144,7 +151,9 @@ def test_non_empty_ref_and_pred(self):
'vol_diff': 3.0,
'lesion_ppv': 1.0,
'lesion_sensitivity': 1.0,
'lesion_f1_score': 1.0}}
'lesion_f1_score': 1.0,
'ref_count': 1,
'pred_count': 1}}

# Create non-empty reference
ref = np.zeros((10, 10, 10))
Expand Down Expand Up @@ -174,7 +183,9 @@ def test_non_empty_ref_and_pred_multi_class(self):
'EmptyPred': False,
'lesion_ppv': 1.0,
'lesion_sensitivity': 1.0,
'lesion_f1_score': 1.0},
'lesion_f1_score': 1.0,
'ref_count': 1,
'pred_count': 1},
2.0: {'dsc': 0.26666666666666666,
'fbeta': 0.26666667461395266,
'nsd': 0.5373134328358209,
Expand All @@ -184,7 +195,9 @@ def test_non_empty_ref_and_pred_multi_class(self):
'EmptyPred': False,
'lesion_ppv': 1.0,
'lesion_sensitivity': 1.0,
'lesion_f1_score': 1.0}}
'lesion_f1_score': 1.0,
'ref_count': 1,
'pred_count': 1}}

# Create non-empty reference
ref = np.zeros((10, 10, 10))
Expand Down Expand Up @@ -215,7 +228,9 @@ def test_non_empty_ref_and_pred_with_full_overlap(self):
'vol_diff': 0.0,
'lesion_ppv': 1.0,
'lesion_sensitivity': 1.0,
'lesion_f1_score': 1.0}}
'lesion_f1_score': 1.0,
'ref_count': 1,
'pred_count': 1}}

# Create non-empty reference
ref = np.zeros((10, 10, 10))
Expand Down

0 comments on commit 3dc77ca

Please sign in to comment.