Add test for precomputed distance.

mblondel · mblondel · commit cf811d226db0 · 2012-03-31T21:45:22.000+09:00
diff --git a/sklearn/neighbors/tests/test_nearest_centroid.py b/sklearn/neighbors/tests/test_nearest_centroid.py
@@ -9,6 +9,7 @@
 
 from sklearn.neighbors import NearestCentroid
 from sklearn import datasets
+from sklearn.metrics.pairwise import pairwise_distances
 
 # toy sample
 X = [[-2, -1], [-1, -1], [-1, -2], [1, 1], [1, 2], [2, 1]]
@@ -46,6 +47,13 @@ def test_classification_toy():
     assert_array_equal(clf.predict(T_csr), true_result)
 
 
+def test_precomputed():
+    clf = NearestCentroid(metric="precomputed")
+    clf.fit(X, y)
+    S = pairwise_distances(T, clf.centroids_)
+    assert_array_equal(clf.predict(S), true_result)
+
+
 def test_iris():
     """Check consistency on dataset iris."""
     for metric in ('euclidean', 'cosine'):
