Update test_psi_detector.py

mitre · Nov 22, 2023 · dab9abf · dab9abf
1 parent a46710c
commit dab9abf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/menelaus/data_drift/test_psi_detector.py b/tests/menelaus/data_drift/test_psi_detector.py
@@ -61,5 +61,5 @@ def test_psi_compute_PSI():
     det = PSI()
     det.set_reference(X = ref)
     det.update(X = test)
-    assertEqual(det.PSI_value == result)
-    assertTrue(det.PSI_value >= 0 and det.PSI_value <= 1)
+    assert det.PSI_value == result
+    assert det.PSI_value >= 0 and det.PSI_value <= 1