Skip to content

Commit 0585462

Browse files
committed
Tests that are checking that _grid_search method throws correct ValueError have been added.
1 parent 5e9e941 commit 0585462

File tree

3 files changed

+22
-4
lines changed

3 files changed

+22
-4
lines changed

sklearn/grid_search.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -332,11 +332,11 @@ def _check_param_grid(param_grid):
332332

333333
check = [isinstance(v, k) for k in (list, tuple, np.ndarray)]
334334
if True not in check:
335-
raise ValueError("Parameter values for parameter ({}) need to "
335+
raise ValueError("Parameter values for parameter ({0}) need to "
336336
"be a sequence.".format(name))
337337

338338
if len(v) == 0:
339-
raise ValueError("Parameter values for parameter ({}) need to "
339+
raise ValueError("Parameter values for parameter ({0}) need to "
340340
"be a non-empty sequence.".format(name))
341341

342342

sklearn/model_selection/_search.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -333,11 +333,11 @@ def _check_param_grid(param_grid):
333333

334334
check = [isinstance(v, k) for k in (list, tuple, np.ndarray)]
335335
if True not in check:
336-
raise ValueError("Parameter values for parameter ({}) need to "
336+
raise ValueError("Parameter values for parameter ({0}) need to "
337337
"be a sequence.".format(name))
338338

339339
if len(v) == 0:
340-
raise ValueError("Parameter values for parameter ({}) need to "
340+
raise ValueError("Parameter values for parameter ({0}) need to "
341341
"be a non-empty sequence.".format(name))
342342

343343

sklearn/model_selection/tests/test_search.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,24 @@ def test_grid_search():
165165
assert_raises(ValueError, grid_search.fit, X, y)
166166

167167

168+
def test_grid_search_incorrect_param_grid():
169+
assert_raise_message(
170+
ValueError,
171+
"Parameter values for parameter (C) need to be a sequence.",
172+
GridSearchCV,
173+
None,
174+
{'C': 1})
175+
176+
177+
def test_grid_search_incorrect_param_grid():
178+
assert_raise_message(
179+
ValueError,
180+
"Parameter values for parameter (C) need to be a non-empty sequence.",
181+
GridSearchCV,
182+
None,
183+
{'C': []})
184+
185+
168186
@ignore_warnings
169187
def test_grid_search_no_score():
170188
# Test grid-search on classifier that has no score function.

0 commit comments

Comments
 (0)