added test

whylabs · Apr 1, 2021 · a000273 · a000273
1 parent 3cdae3d
commit a000273
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/tests/unit/core/metrics/test_model_metrics.py b/tests/unit/core/metrics/test_model_metrics.py
@@ -51,9 +51,22 @@ def test_merge_metrics_with_none_confusion_matrix():
     metrics = ModelMetrics()
     other = ModelMetrics()
     other.confusion_matrix = None
-    metrics.merge(other)
+    new_metrics = metrics.merge(other)
 
 
+def test_merge_metrics_with_none_regression_matrix():
+    metrics = ModelMetrics()
+    other = ModelMetrics()
+    other.regression_metrics = None
+    new_metrics= metrics.merge(other)
+
+def test_merge_metrics_with_none_confusion_matrix():
+    metrics = ModelMetrics()
+    other = ModelMetrics()
+    other.confusion_matrix = None
+    other.regression_metrics = None
+
+    new_metrics = metrics.merge(other)
 
 def test_model_metrics_init():
     reg_met = RegressionMetrics()