test for ValueError

paulgb · paulgb · commit d7593fc47428 · 2013-09-29T09:52:39.000-04:00
diff --git a/sklearn/tests/test_pipeline.py b/sklearn/tests/test_pipeline.py
@@ -82,6 +82,9 @@ def test_pipeline_init():
     filter1 = SelectKBest(f_classif)
     pipe = Pipeline([('anova', filter1), ('svc', clf)])
 
+    # Check that we can't use the same stage name twice
+    assert_raises(ValueError, Pipeline, [('svc', SVC()), ('svc', SVC())])
+
     # Check that params are set
     pipe.set_params(svc__C=0.1)
     assert_equal(clf.C, 0.1)
