Skip to content

Commit d963b2f

Browse files
committed
Add test for local linear embedding
1 parent a7038a4 commit d963b2f

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

sklearn/manifold/tests/test_locally_linear.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from sklearn.manifold.locally_linear import barycenter_kneighbors_graph
1010
from sklearn.utils.testing import assert_less
1111
from sklearn.utils.testing import ignore_warnings
12+
from sklearn.utils.testing import assert_raise_message
1213

1314
eigen_solvers = ['dense', 'arpack']
1415

@@ -100,6 +101,19 @@ def test_lle_manifold():
100101
tol * reconstruction_error, msg=details)
101102

102103

104+
# Test the error raised when parameter passed to lle is invalid
105+
def test_lle_init_parameters():
106+
X = np.random.rand(5, 3)
107+
108+
clf = manifold.LocallyLinearEmbedding(eigen_solver="error")
109+
msg = "unrecognized eigen_solver 'error'"
110+
assert_raise_message(ValueError, msg, clf.fit, X)
111+
112+
clf = manifold.LocallyLinearEmbedding(method="error")
113+
msg = "unrecognized method 'error'"
114+
assert_raise_message(ValueError, msg, clf.fit, X)
115+
116+
103117
def test_pipeline():
104118
# check that LocallyLinearEmbedding works fine as a Pipeline
105119
# only checks that no error is raised.

0 commit comments

Comments
 (0)