@Test
public void testClassifyData() throws DescriptorException {

  // Training data
  Data[] datas = generateTrainingDataA();
  // Build Forest
  DecisionForest forest = buildForest(datas);
  // Test data
  Dataset dataset = datas[0].getDataset();
  Data testData = DataLoader.loadData(dataset, TEST_DATA);
  double[][] predictions = new double[testData.size()][];
  forest.classify(testData, predictions);
  double noValue = dataset.valueOf(4, "no");
  double yesValue = dataset.valueOf(4, "yes");
  assertArrayEquals(new double[][]{{noValue, Double.NaN, Double.NaN},
      {noValue, yesValue, Double.NaN}, {noValue, noValue, Double.NaN}}, predictions);
}