def test_mock_with_fit_param_raises():
X, y = make_classification(random_state=0)
clf = MockClassifierWithFitParam()
with pytest.raises(ValueError):
clf.fit(X, y)
def test_gridsearch_with_arraylike_fit_param(cache_cv):
# https://github.com/dask/dask-ml/issues/319
X, y = make_classification(random_state=0)
param_grid = {"foo_param": [0.0001, 0.1]}
a = dcv.GridSearchCV(
MockClassifierWithFitParam(),
param_grid,
cv=3,
refit=False,
cache_cv=cache_cv,
)
b = GridSearchCV(MockClassifierWithFitParam(), param_grid, cv=3, refit=False)
b.fit(X, y, mock_fit_param=[0, 1])
a.fit(X, y, mock_fit_param=[0, 1])