Merge pull request scikit-learn#6393 from yenchenlin1994/add-test-for-local-linear-embedding

Add test for local linear embedding

Author: jakevdp

sklearn/manifold/tests/test_locally_linear.py

@@ -9,6 +9,7 @@
 from sklearn.manifold.locally_linear import barycenter_kneighbors_graph
 from sklearn.utils.testing import assert_less
 from sklearn.utils.testing import ignore_warnings
+from sklearn.utils.testing import assert_raise_message
 
 eigen_solvers = ['dense', 'arpack']
 
@@ -100,6 +101,19 @@ def test_lle_manifold():
                         tol * reconstruction_error, msg=details)
 
 
+# Test the error raised when parameter passed to lle is invalid
+def test_lle_init_parameters():
+    X = np.random.rand(5, 3)
+
+    clf = manifold.LocallyLinearEmbedding(eigen_solver="error")
+    msg = "unrecognized eigen_solver 'error'"
+    assert_raise_message(ValueError, msg, clf.fit, X)
+
+    clf = manifold.LocallyLinearEmbedding(method="error")
+    msg = "unrecognized method 'error'"
+    assert_raise_message(ValueError, msg, clf.fit, X)
+
+
 def test_pipeline():
     # check that LocallyLinearEmbedding works fine as a Pipeline
     # only checks that no error is raised.