mahout-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From robina...@apache.org
Subject svn commit: r905232 - /lucene/mahout/trunk/core/src/test/java/org/apache/mahout/clustering/kmeans/TestKmeansClustering.java
Date Mon, 01 Feb 2010 10:08:30 GMT
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());
   }
-
-
 }



Mime
View raw message