From 433815b0686519a3608e97ac7517626cb31df879 Mon Sep 17 00:00:00 2001 From: Michael Vasilev Date: Tue, 24 Dec 2024 09:19:48 +0300 Subject: [PATCH] Revert changes in test_devnet.py --- pyod/test/test_devnet.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/pyod/test/test_devnet.py b/pyod/test/test_devnet.py index b7499b8c..9f33b251 100644 --- a/pyod/test/test_devnet.py +++ b/pyod/test/test_devnet.py @@ -120,6 +120,20 @@ def test_prediction_with_rejection_stats(self): assert (ub_rejrate <= 1) assert (ub_cost >= 0) + def test_fit_predict(self): + pred_labels = self.clf.fit_predict(self.X_train, self.y_train) + assert_equal(pred_labels.shape, self.y_train.shape) + + def test_fit_predict_score(self): + self.clf.fit_predict_score(self.X_test, self.y_test) + self.clf.fit_predict_score(self.X_test, self.y_test, + scoring='roc_auc_score') + self.clf.fit_predict_score(self.X_test, self.y_test, + scoring='prc_n_score') + with assert_raises(NotImplementedError): + self.clf.fit_predict_score(self.X_test, self.y_test, + scoring='something') + def test_model_clone(self): pass # clone_clf = clone(self.clf)