Author: tommaso
Date: Tue Oct 16 06:25:44 2012
New Revision: 1398665
URL: http://svn.apache.org/viewvc?rev=1398665&view=rev
Log:
[HAMA-651] - added unit test for VectorDoubleFileInputFormat
Added:
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/VectorDoubleFileInputFormatTest.java
(with props)
hama/trunk/ml/src/test/resources/vd_file_sample.txt (with props)
Added:
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/VectorDoubleFileInputFormatTest.java
URL:
http://svn.apache.org/viewvc/hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/VectorDoubleFileInputFormatTest.java?rev=1398665&view=auto
==============================================================================
---
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/VectorDoubleFileInputFormatTest.java
(added)
+++
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/VectorDoubleFileInputFormatTest.java
Tue Oct 16 06:25:44 2012
@@ -0,0 +1,56 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.hama.ml.regression;
+
+import org.apache.hadoop.fs.Path;
+import org.apache.hadoop.io.DoubleWritable;
+import org.apache.hama.bsp.BSPJob;
+import org.apache.hama.bsp.FileSplit;
+import org.apache.hama.bsp.InputSplit;
+import org.apache.hama.bsp.RecordReader;
+import org.apache.hama.ml.math.DenseDoubleVector;
+import org.apache.hama.ml.writable.VectorWritable;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * Testcase for {@link VectorDoubleFileInputFormat}
+ */
+public class VectorDoubleFileInputFormatTest {
+
+ @Test
+ public void testFileRead() throws Exception {
+ VectorDoubleFileInputFormat inputFormat = new
VectorDoubleFileInputFormat();
+ Path file = new Path("src/test/resources/vd_file_sample.txt");
+ InputSplit split = new FileSplit(file, 0, 1000, new String[]{"localhost"});
+ BSPJob job = new BSPJob();
+ RecordReader<VectorWritable,DoubleWritable> recordReader =
inputFormat.getRecordReader(split, job);
+ assertNotNull(recordReader);
+ VectorWritable key = recordReader.createKey();
+ assertNotNull(key);
+ DoubleWritable value = recordReader.createValue();
+ assertNotNull(value);
+ assertTrue(recordReader.next(key, value));
+ assertEquals(new DenseDoubleVector(new double[]{2d, 3d, 4d}),
key.getVector());
+ assertEquals(new DoubleWritable(1d), value);
+ }
+
+}
Propchange:
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/VectorDoubleFileInputFormatTest.java
------------------------------------------------------------------------------
svn:eol-style = native
Added: hama/trunk/ml/src/test/resources/vd_file_sample.txt
URL:
http://svn.apache.org/viewvc/hama/trunk/ml/src/test/resources/vd_file_sample.txt?rev=1398665&view=auto
==============================================================================
--- hama/trunk/ml/src/test/resources/vd_file_sample.txt (added)
+++ hama/trunk/ml/src/test/resources/vd_file_sample.txt Tue Oct 16 06:25:44 2012
@@ -0,0 +1,2 @@
+1>2 3 4
+5>6 7 8
\ No newline at end of file
Propchange: hama/trunk/ml/src/test/resources/vd_file_sample.txt
------------------------------------------------------------------------------
svn:eol-style = native
