@Test
public void testTaskContext() {

  TaskContext context = new TaskContext(null, null, null, null, null, null,
      null);
  
  context.setInputKeyClass(IntWritable.class);
  Assert.assertEquals(IntWritable.class.getName(), context.getInputKeyClass
      ().getName());
  context.setInputValueClass(Text.class);
  Assert.assertEquals(Text.class.getName(), context.getInputValueClass()
      .getName());
 
  context.setOutputKeyClass(LongWritable.class);
  Assert.assertEquals(LongWritable.class.getName(), context
      .getOutputKeyClass().getName());
  context.setOutputValueClass(FloatWritable.class);
  Assert.assertEquals(FloatWritable.class.getName(), context
      .getOutputValueClass().getName());
}