From 8ef1b18b4fbdd30d6190049c6b02a5e9347f483a Mon Sep 17 00:00:00 2001 From: Tim Chen <115333718+951378644@users.noreply.github.com> Date: Wed, 27 Dec 2023 16:19:50 -0500 Subject: [PATCH] Update test_dl_detector.py --- tests/menelaus/data_drift/test_dl_detector.py | 15 ++------------- 1 file changed, 2 insertions(+), 13 deletions(-) diff --git a/tests/menelaus/data_drift/test_dl_detector.py b/tests/menelaus/data_drift/test_dl_detector.py index 711ecff..efc62ba 100644 --- a/tests/menelaus/data_drift/test_dl_detector.py +++ b/tests/menelaus/data_drift/test_dl_detector.py @@ -24,9 +24,8 @@ def test_dl_detector_set_reference(): np.ones((rows, columns)), columns=[f"Column_{i}" for i in range(columns)] ) det.set_reference(ref) - assert det.reference == pd.DataFrame( - scaler.fit_transform(ref), columns=ref.columns, index=ref.index - ) + scale_ref = pd.DataFrame(scaler.fit_transform(ref), columns=ref.columns, index=ref.index) + assert det.reference == scale_ref def test_dl_detector_update_1(): @@ -80,13 +79,3 @@ def test_dl_detector_update_4(): det._drift_state = "drift" det.update(ref.replace(1, 2)) assert det.drift_state is None - - -def test_dl_detector_reset(): - """Check dl_detector.reset works as intended""" - det = DL_Detector() - det.batches_since_reset = 1 - det.drift_state = "drift" - det.reset() - assert det.batches_since_reset == 0 - assert det.drift_state is None