Improved coverage for DiceCoefficient

pytorch · Jan 21, 2020 · a32edb9 · a32edb9
1 parent f3b6734
commit a32edb9
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/ignite/metrics/test_confusion_matrix.py b/tests/ignite/metrics/test_confusion_matrix.py
@@ -475,6 +475,25 @@ def test_cm_with_average():
     np.testing.assert_almost_equal(true_pr, res)
 
 
+def test_dice_coefficient_wrong_input():
+
+    with pytest.raises(TypeError, match="Argument cm should be instance of ConfusionMatrix"):
+        DiceCoefficient(None)
+
+    cm = ConfusionMatrix(num_classes=10)
+    with pytest.raises(ValueError, match="ignore_index should be non-negative integer"):
+        DiceCoefficient(cm, ignore_index=-1)
+
+    with pytest.raises(ValueError, match="ignore_index should be non-negative integer"):
+        DiceCoefficient(cm, ignore_index="a")
+
+    with pytest.raises(ValueError, match="ignore_index should be non-negative integer"):
+        DiceCoefficient(cm, ignore_index=10)
+
+    with pytest.raises(ValueError, match="ignore_index should be non-negative integer"):
+        DiceCoefficient(cm, ignore_index=11)
+
+
 def test_dice_coefficient():
 
     y_true, y_pred = get_y_true_y_pred()