@Test
public void populateOutputFieldsMap() {

    final List<KiePMMLOutputField> outputFields = new ArrayList<>();
    KiePMMLOutputField predictedOutputField = getOutputField("KOF-TARGET", RESULT_FEATURE.PREDICTED_VALUE,
                                                             "TARGET");
    outputFields.add(predictedOutputField);
    final List<KiePMMLOutputField> probabilityOutputFields = IntStream.range(0, 2)
            .mapToObj(index -> getOutputField("KOF-PROB-" + index, RESULT_FEATURE.PROBABILITY, "PROB-" + index))
            .collect(Collectors.toList());
    outputFields.addAll(probabilityOutputFields);
    KiePMMLRegressionTableRegressionFactory.populateOutputFieldsMap(modelTemplate, outputFields);
    MethodDeclaration methodDeclaration =
            modelTemplate.getMethodsByName("populateOutputFieldsMapWithResult").get(0);
    BlockStmt body = methodDeclaration.getBody().get();
    NodeList<Statement> retrieved = body.getStatements();
    assertEquals(1, retrieved.size());
}