svn commit: r905232 - /lucene/mahout/trunk/core/src/test/java/org/apache/mahout/clustering/kmeans/TestKmeansClustering.java

[robinanil]

Author: robinanil
Date: Mon Feb  1 10:08:29 2010
New Revision: 905232

URL: http://svn.apache.org/viewvc?rev=905232&view=rev
Log:
TestKMeansWithCanopyClusterInput updated to use the lastest code. It was never 
updated since the first checkin

Modified:
    
lucene/mahout/trunk/core/src/test/java/org/apache/mahout/clustering/kmeans/TestKmeansClustering.java

Modified: 
lucene/mahout/trunk/core/src/test/java/org/apache/mahout/clustering/kmeans/TestKmeansClustering.java
URL: 
http://svn.apache.org/viewvc/lucene/mahout/trunk/core/src/test/java/org/apache/mahout/clustering/kmeans/TestKmeansClustering.java?rev=905232&r1=905231&r2=905232&view=diff
==============================================================================
--- 
lucene/mahout/trunk/core/src/test/java/org/apache/mahout/clustering/kmeans/TestKmeansClustering.java
 (original)
+++ 
lucene/mahout/trunk/core/src/test/java/org/apache/mahout/clustering/kmeans/TestKmeansClustering.java
 Mon Feb  1 10:08:29 2010
@@ -478,15 +478,21 @@
     String[] outFiles = outDir.list();
     assertEquals("output dir files?", 4, outFiles.length);
     DummyOutputCollector<Text, Text> collector = new 
DummyOutputCollector<Text, Text>();
-    for (String line : new FileLineIterable(new 
File("output/points/part-00000"))) {
-      String[] lineParts = line.split("\t");
-      assertEquals("line parts", 2, lineParts.length);
-      String cl = line.substring(0, line.indexOf(':'));
-      collector.collect(new Text(cl), new Text(lineParts[1]));
+    SequenceFile.Reader reader = new SequenceFile.Reader(fs, new 
Path("output/points/part-00000"), conf);
+   
+    //The key is the name of the vector, or the vector itself
+    Text key = new Text();
+    //The value is the cluster id
+    Text value = new Text();
+    while (reader.next(key, value)) {
+      collector.collect(value, key);
+      key = new Text();
+      value = new Text();
+
     }
-    assertEquals("num points[V0]", 4, collector.getValue("V0").size());
-    assertEquals("num points[V1]", 5, collector.getValue("V1").size());
+    reader.close();
+    
+    assertEquals("num points[0]", 4, collector.getValue("0").size());
+    assertEquals("num points[1]", 5, collector.getValue("1").size());
   }
-
-
 }