@Test
public void testInducedTreeClassifiesDataSetCorrectly() throws Exception {
DataSet ds = DataSetFactory.getRestaurantDataSet();
DecisionTreeLearner learner = new DecisionTreeLearner();
learner.train(ds);
int[] result = learner.test(ds);
Assert.assertEquals(12, result[0]);
Assert.assertEquals(0, result[1]);
}
@Test