Update test_stat_test.py

mitre · Dec 20, 2023 · a05d053 · a05d053
1 parent 06f7185
commit a05d053
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/menelaus/data_drift/test_stat_test.py b/tests/menelaus/data_drift/test_stat_test.py
@@ -52,6 +52,17 @@ def test_update_with_drift(self):
         detector.set_reference(reference_data)
         detector.update(new_data)
         assert detector.drift_state is not None  #drift expected
+
+    def test_update_reset():
+        """Check GenericDetector.update behavior after drift alarm"""
+        detector = GenericDetector(representation=MockRepresentation(), 
+                      divergence=MockDivergence, 
+                      crit_function=MockCritFunction)
+        detector.set_reference([1,2,3])
+        detector.drift_state = 'drift'
+        detector.reset()
+        detector.update(X=[1,2,3])
+        assert detector.drift_state is None 
 
 class TestCHIDetector:
     def test_initialization(self):