// the tree should receive the partition's index
assertEquals(partition, k.partition());
// make sure all the trees of the other partitions are handled in the
// correct order
assertEquals(index, k.treeId());
int[] predictions = output.getValues()[current].getPredictions();
// all the instances of the partition should be classified
assertEquals(split.length, predictions.length);