Skip to content

Commit 1e5e7e1

Browse files
committed
failing test for wrong solver exception
1 parent b3affdb commit 1e5e7e1

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

sklearn/linear_model/tests/test_ridge.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -743,3 +743,19 @@ def test_deprecation_warning_dense_cholesky():
743743
for func in [func1, func2, func3]:
744744
assert_warns_message(warning_class, warning_message, func)
745745

746+
747+
def test_raises_value_error_if_solver_not_supported():
748+
"""Tests whether a ValueError is raised if a non-identified solver
749+
is passed to ridge_regression"""
750+
751+
wrong_solver = "This is not a solver (MagritteSolveCV QuantumBitcoin)"
752+
753+
exception = ValueError
754+
message = "Solver %s not understood" % wrong_solver
755+
756+
def func():
757+
X = np.eye(3)
758+
y = np.ones(3)
759+
ridge_regression(X, y, alpha=1., solver=wrong_solver)
760+
761+
assert_raise_message(exception, message, func)

0 commit comments

Comments
 (0)