Revert changes in test_devnet.py

yzhao062 · Dec 24, 2024 · 433815b · 433815b
1 parent 7b1cc4e
commit 433815b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git 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)