Fix tests.

evidentlyai · Jul 28, 2022 · 9b8e565 · 9b8e565
1 parent 4346cf8
commit 9b8e565
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/evidently/tests/classification_performance_tests.py b/src/evidently/tests/classification_performance_tests.py
@@ -362,7 +362,7 @@ def get_value(self, result: DatasetClassificationPerformanceMetrics):
 
     def get_description(self, value: Numeric) -> str:
         if value is None:
-            return "ot enough data to calculate Logarithmic Loss. Consider providing probabilities instead of labels."
+            return "Not enough data to calculate Logarithmic Loss. Consider providing probabilities instead of labels."
 
         else:
             return f" Logarithmic Loss is {value:.3g}. Test Threshold is {self.get_condition()}"

diff --git a/tests/tests/test_classification_performance_tests.py b/tests/tests/test_classification_performance_tests.py
@@ -222,7 +222,8 @@ def test_log_loss_test_cannot_calculate_log_loss() -> None:
     suite.run(current_data=test_dataset, reference_data=None, column_mapping=column_mapping)
     assert not suite
     test_info = suite.as_dict()["tests"][0]
-    assert test_info["description"] == "No log loss value for the data"
+    assert test_info["description"] == "Not enough data to calculate Logarithmic Loss." \
+                                       " Consider providing probabilities instead of labels."
     assert test_info["status"] == "ERROR"
 
 
@@ -282,7 +283,8 @@ def test_roc_auc_test_cannot_calculate_roc_auc() -> None:
     suite.run(current_data=test_dataset, reference_data=None, column_mapping=column_mapping)
     assert not suite
     test_info = suite.as_dict()["tests"][0]
-    assert test_info["description"] == "No ROC AUC Score value for the data"
+    assert test_info["description"] == "Not enough data to calculate ROC AUC." \
+                                       " Consider providing probabilities instead of labels."
     assert test_info["status"] == "ERROR"