}
@Test
public void testGainCalculation() throws Exception {
DataSet ds = DataSetFactory.getRestaurantDataSet();
double gain = ds.calculateGainFor("patrons");
Assert.assertEquals(0.541, gain, 0.001);
gain = ds.calculateGainFor("type");
Assert.assertEquals(0.0, gain, 0.001);
}
}