33
33
check_class_weight_balanced_linear_classifier ,
34
34
check_transformer_n_iter ,
35
35
check_non_transformer_estimators_n_iter ,
36
- check_get_params_invariance )
36
+ check_get_params_invariance ,
37
+ _set_test_name )
37
38
38
39
39
40
def test_all_estimator_no_base_class ():
@@ -55,7 +56,8 @@ def test_all_estimators():
55
56
56
57
for name , Estimator in estimators :
57
58
# some can just not be sensibly default constructed
58
- yield check_parameters_default_constructible , name , Estimator
59
+ yield (_set_test_name (check_parameters_default_constructible , name ),
60
+ name , Estimator )
59
61
60
62
61
63
def test_non_meta_estimators ():
@@ -70,9 +72,9 @@ def test_non_meta_estimators():
70
72
if issubclass (Estimator , ProjectedGradientNMF ):
71
73
# The ProjectedGradientNMF class is deprecated
72
74
with ignore_warnings ():
73
- yield check , name , Estimator
75
+ yield _set_test_name ( check , name ) , name , Estimator
74
76
else :
75
- yield check , name , Estimator
77
+ yield _set_test_name ( check , name ) , name , Estimator
76
78
77
79
78
80
def test_configure ():
@@ -114,7 +116,8 @@ def test_class_weight_balanced_linear_classifiers():
114
116
issubclass (clazz , LinearClassifierMixin ))]
115
117
116
118
for name , Classifier in linear_classifiers :
117
- yield check_class_weight_balanced_linear_classifier , name , Classifier
119
+ yield _set_test_name (check_class_weight_balanced_linear_classifier ,
120
+ name ), name , Classifier
118
121
119
122
120
123
@ignore_warnings
@@ -196,8 +199,9 @@ def test_non_transformer_estimators_n_iter():
196
199
else :
197
200
# Multitask models related to ENet cannot handle
198
201
# if y is mono-output.
199
- yield (check_non_transformer_estimators_n_iter ,
200
- name , estimator , 'Multi' in name )
202
+ yield (_set_test_name (
203
+ check_non_transformer_estimators_n_iter , name ),
204
+ name , estimator , 'Multi' in name )
201
205
202
206
203
207
def test_transformer_n_iter ():
@@ -218,9 +222,12 @@ def test_transformer_n_iter():
218
222
if isinstance (estimator , ProjectedGradientNMF ):
219
223
# The ProjectedGradientNMF class is deprecated
220
224
with ignore_warnings ():
221
- yield check_transformer_n_iter , name , estimator
225
+ yield _set_test_name (
226
+ check_transformer_n_iter , name ), name , estimator
222
227
else :
223
- yield check_transformer_n_iter , name , estimator
228
+ yield _set_test_name (
229
+ check_transformer_n_iter , name ), name , estimator
230
+
224
231
225
232
def test_get_params_invariance ():
226
233
# Test for estimators that support get_params, that
@@ -234,6 +241,8 @@ def test_get_params_invariance():
234
241
# If class is deprecated, ignore deprecated warnings
235
242
if hasattr (Estimator .__init__ , "deprecated_original" ):
236
243
with ignore_warnings ():
237
- yield check_get_params_invariance , name , Estimator
244
+ yield _set_test_name (
245
+ check_get_params_invariance , name ), name , Estimator
238
246
else :
239
- yield check_get_params_invariance , name , Estimator
247
+ yield _set_test_name (
248
+ check_get_params_invariance , name ), name , Estimator
0 commit comments