@Test
public void testClassifyData() throws DescriptorException {

  // Training data
  Data[] datas = generateTrainingDataA();
  // Build Forest
  DecisionForest forest = buildForest(datas);
  // Test data
  Data testData = DataLoader.loadData(datas[0].getDataset(), TEST_DATA);
  double[][] predictions = new double[testData.size()][];
  forest.classify(testData, predictions);
  assertArrayEquals(new double[][]{{1.0, Double.NaN, Double.NaN},
      {1.0, 0.0, Double.NaN}, {1.0, 1.0, Double.NaN}}, predictions);
}