Update testing

orausch · orausch · commit 01af41359ec5 · 2018-07-28T17:01:44.000+02:00
diff --git a/sklearn/tests/test_metaestimators.py b/sklearn/tests/test_metaestimators.py
@@ -15,6 +15,7 @@
 from sklearn.feature_selection import RFE, RFECV
 from sklearn.ensemble import BaggingClassifier
 from sklearn.exceptions import NotFittedError
+from sklearn.semi_supervised.self_training import SelfTraining
 
 
 class DelegatorData(object):
@@ -43,7 +44,10 @@ def __init__(self, name, construct, skip_methods=(),
     DelegatorData('BaggingClassifier', BaggingClassifier,
                   skip_methods=['transform', 'inverse_transform', 'score',
                                 'predict_proba', 'predict_log_proba',
-                                'predict'])
+                                'predict']),
+    DelegatorData('Self-Training', lambda est: SelfTraining(est),
+                  skip_methods=['transform', 'inverse_transform', 'score',
+                                'predict_log_proba', 'decision_funtion'])
 ]
 
 
diff --git a/sklearn/utils/testing.py b/sklearn/utils/testing.py
@@ -539,7 +539,7 @@ def uninstall_mldata_mock():
                    "MultiOutputRegressor", "MultiOutputClassifier",
                    "OutputCodeClassifier", "OneVsRestClassifier",
                    "RFE", "RFECV", "BaseEnsemble", "ClassifierChain",
-                   "RegressorChain"]
+                   "RegressorChain", "SelfTraining"]
 # estimators that there is no way to default-construct sensibly
 OTHER = ["Pipeline", "FeatureUnion", "GridSearchCV", "RandomizedSearchCV",
          "SelectFromModel", "ColumnTransformer"]
