private static final double [][] MATRIX = { {1, 1}, {2, 3} };
private static final double [] VECTOR = {9, 16};
@Test
public void testVectorMatrixMultiplicationMapper() throws Exception {
VectorMatrixMultiplicationMapper mapper = new VectorMatrixMultiplicationMapper();
Configuration conf = new Configuration();
// set up all the parameters for the job
Vector toSave = new DenseVector(VECTOR);
DummyRecordWriter<IntWritable, VectorWritable> writer = new
DummyRecordWriter<IntWritable, VectorWritable>();
Mapper<IntWritable, VectorWritable, IntWritable, VectorWritable>.Context
context = DummyRecordWriter.build(mapper, conf, writer);
mapper.setup(toSave);
// run the job
for (int i = 0; i < MATRIX.length; i++) {
Vector v = new RandomAccessSparseVector(MATRIX[i].length);
v.assign(MATRIX[i]);
mapper.map(new IntWritable(i), new VectorWritable(v), context);
}
// check the results
assertEquals("Number of map results", MATRIX.length, writer.getData().size());
for (int i = 0; i < MATRIX.length; i++) {