We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c83dd0c commit 41a726dCopy full SHA for 41a726d
imblearn/utils/estimator_checks.py
@@ -75,6 +75,19 @@ def sample_dataset_generator_fixture():
75
return sample_dataset_generator()
76
77
78
+def test_dataset_generator():
79
+ X_generated, y_generated = sample_dataset_generator()
80
+ X, y = make_classification(
81
+ n_samples=1000,
82
+ n_classes=3,
83
+ n_informative=4,
84
+ weights=[0.2, 0.3, 0.5],
85
+ random_state=0,
86
+ )
87
+ assert X_generated.all() == X.all()
88
+ assert y_generated.all() == y.all()
89
+
90
91
def _set_checking_parameters(estimator):
92
params = estimator.get_params()
93
name = estimator.__class__.__name__
0 commit comments