Update test_psi_detector.py

mitre · Nov 22, 2023 · 9e91dcc · 9e91dcc
1 parent 5d9f06f
commit 9e91dcc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/menelaus/data_drift/test_psi_detector.py b/tests/menelaus/data_drift/test_psi_detector.py
@@ -57,4 +57,9 @@ def test_psi_compute_PSI():
     test = np.random.randint(0,100,100)
     det.set_reference(X = ref)
     det.update(X = test)
+    result = det.PSI_value
+    det = PSI()
+    det.set_reference(X = ref)
+    det.update(X = test)
+    assert det.PSI_value == result
     assert det.PSI_value >= 0 and det.PSI_value <= 1