Added tests to catch dice_metrics asserts

dice_score_3d/metrics.py

@@ -41,7 +41,7 @@ def dice_metrics(ground_truths: str, predictions: str, output_path: str, indices
     dtype = np.uint8 if dtype == 'uint8' else np.uint16
     assert os.path.isfile(ground_truths) and os.path.isfile(predictions) or \
            os.path.isdir(ground_truths) and os.path.isdir(predictions), ('Prediction path and GT path must both be a '
-                                                                         'a single file or a folder.')
+                                                                         'single file or a folder.')
 
     if os.path.isdir(ground_truths):
         gt_files = sorted([x for x in os.listdir(ground_truths) if x.startswith(prefix) and x.endswith(suffix)])

tests/test_metrics.py

@@ -3,8 +3,8 @@
 import unittest
 
 import numpy as np
-from dice_score_3d.reader import read_mask
 
+from dice_score_3d import dice_metrics
 from dice_score_3d.metrics import dice, multi_class_dice, evaluate_prediction
 from tests.utils import create_and_write_volume
 
@@ -43,6 +43,14 @@ def test_evaluate_prediction(self):
             tmp.close()
             os.unlink(tmp.name)
 
+    def test_dice_metrics(self):
+        self.assertRaisesRegex(AssertionError, 'Prediction path and GT path must both be a single file or a folder',
+                               dice_metrics, './', './random_string?.!@3$not_a_path', 'results.csv', {'Lung': 1})
+        self.assertRaisesRegex(AssertionError, 'Output path must be either .csv or .json, is results.txt',
+                               dice_metrics, './', './', 'results.txt', {'Lung': 1})
+        self.assertRaisesRegex(AssertionError, 'Indices must be integers, found .*',
+                               dice_metrics, './', './', 'results.csv', {'Lung': 'text'})
+
 
 if __name__ == '__main__':
     unittest.main()