Merge pull request #50 from CompML/features/#40/unit_test_for_precision

Features/#40/unit test for precision
CompML · Dec 31, 2020 · 98a48ee · 98a48ee
2 parents d9ab381 + bd642d8
commit 98a48ee
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_precision.py b/tests/test_precision.py
@@ -49,6 +49,18 @@ def test_PrecisionClass_score(self):
         score = obj.score(real, pred)
         self.assertEqual(score, 1.0)
 
+    def test_PrecisionClass_update_precision(self):
+        """Test of _update_precision function.
+        """
+
+        real = np.array([1, 1, 0, 0, 0])
+        pred = np.array([0, 1, 0, 0, 0])
+
+        obj = TimeSeriesPrecision()
+        real_anomalies, predicted_anomalies = obj._prepare_data(real, pred)
+
+        score = obj._update_precision(real_anomalies, predicted_anomalies)
+        self.assertEqual(score, 1.0)
 
     def test_precision_function(self):
         """Teest of ts_precision function.