public void testCurrentBestLearnerOnRestaurantDataSet() throws Exception {
DataSet ds = DataSetFactory.getRestaurantDataSet();
CurrentBestLearner learner = new CurrentBestLearner("Yes");
learner.train(ds);
int[] result = learner.test(ds);
Assert.assertEquals(12, result[0]);
Assert.assertEquals(0, result[1]);
}
}