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