You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@hama.apache.org by ed...@apache.org on 2012/08/24 04:28:52 UTC
svn commit: r1376809 -
/hama/trunk/ml/src/test/java/org/apache/hama/ml/TestKMeansBSP.java
Author: edwardyoon
Date: Fri Aug 24 02:28:51 2012
New Revision: 1376809
URL: http://svn.apache.org/viewvc?rev=1376809&view=rev
Log:
Fix minor problem.
Modified:
hama/trunk/ml/src/test/java/org/apache/hama/ml/TestKMeansBSP.java
Modified: hama/trunk/ml/src/test/java/org/apache/hama/ml/TestKMeansBSP.java
URL: http://svn.apache.org/viewvc/hama/trunk/ml/src/test/java/org/apache/hama/ml/TestKMeansBSP.java?rev=1376809&r1=1376808&r2=1376809&view=diff
==============================================================================
--- hama/trunk/ml/src/test/java/org/apache/hama/ml/TestKMeansBSP.java (original)
+++ hama/trunk/ml/src/test/java/org/apache/hama/ml/TestKMeansBSP.java Fri Aug 24 02:28:51 2012
@@ -33,54 +33,55 @@ import org.apache.hama.ml.math.DoubleVec
public class TestKMeansBSP extends TestCase {
public void testRunJob() throws Exception {
-
Configuration conf = new Configuration();
Path in = new Path("/tmp/clustering/in/in.txt");
Path out = new Path("/tmp/clustering/out/");
FileSystem fs = FileSystem.get(conf);
Path center = null;
- if (fs.isFile(in))
- center = new Path(in.getParent(), "center/cen.seq");
- else
- center = new Path(in, "center/cen.seq");
- Path centerOut = new Path(out, "center/center_output.seq");
- conf.set(KMeansBSP.CENTER_IN_PATH, center.toString());
- conf.set(KMeansBSP.CENTER_OUT_PATH, centerOut.toString());
- int iterations = 10;
- conf.setInt(KMeansBSP.MAX_ITERATIONS_KEY, iterations);
- int k = 1;
-
- FSDataOutputStream create = fs.create(in);
- BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(create));
- StringBuilder sb = new StringBuilder();
-
- for (int i = 0; i < 100; i++) {
- sb.append(i);
- sb.append('\t');
- sb.append(i);
- sb.append('\n');
- }
-
- bw.write(sb.toString());
- bw.close();
-
- in = KMeansBSP.prepareInputText(k, conf, in, center, out, fs);
- BSPJob job = KMeansBSP.createJob(conf, in, out, true);
-
- // just submit the job
- boolean result = job.waitForCompletion(true);
-
- assertEquals(true, result);
-
- HashMap<Integer, DoubleVector> centerMap = KMeansBSP.readOutput(conf, out,
- centerOut, fs);
- System.out.println(centerMap);
- assertEquals(1, centerMap.size());
- DoubleVector doubleVector = centerMap.get(0);
- assertTrue(doubleVector.get(0) > 50 && doubleVector.get(0) < 51);
- assertTrue(doubleVector.get(1) > 50 && doubleVector.get(1) < 51);
+ try {
+ center = new Path(in.getParent(), "center/cen.seq");
+ Path centerOut = new Path(out, "center/center_output.seq");
+ conf.set(KMeansBSP.CENTER_IN_PATH, center.toString());
+ conf.set(KMeansBSP.CENTER_OUT_PATH, centerOut.toString());
+ int iterations = 10;
+ conf.setInt(KMeansBSP.MAX_ITERATIONS_KEY, iterations);
+ int k = 1;
+
+ FSDataOutputStream create = fs.create(in);
+ BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(create));
+ StringBuilder sb = new StringBuilder();
+
+ for (int i = 0; i < 100; i++) {
+ sb.append(i);
+ sb.append('\t');
+ sb.append(i);
+ sb.append('\n');
+ }
+
+ bw.write(sb.toString());
+ bw.close();
+
+ in = KMeansBSP.prepareInputText(k, conf, in, center, out, fs);
+
+ BSPJob job = KMeansBSP.createJob(conf, in, out, true);
+
+ // just submit the job
+ boolean result = job.waitForCompletion(true);
+
+ assertEquals(true, result);
+
+ HashMap<Integer, DoubleVector> centerMap = KMeansBSP.readOutput(conf,
+ out, centerOut, fs);
+ System.out.println(centerMap);
+ assertEquals(1, centerMap.size());
+ DoubleVector doubleVector = centerMap.get(0);
+ assertTrue(doubleVector.get(0) > 50 && doubleVector.get(0) < 51);
+ assertTrue(doubleVector.get(1) > 50 && doubleVector.get(1) < 51);
+ } finally {
+ fs.delete(new Path("/tmp/clustering"), true);
+ }
}
}