@Test
public void populateGetProbabilityMapMethodSupported() {

    SUPPORTED_NORMALIZATION_METHODS.forEach(normalizationMethod -> {
        KiePMMLRegressionTableClassificationFactory.populateGetProbabilityMapMethod(normalizationMethod,
                                                                                    modelTemplate);
        MethodDeclaration methodDeclaration =
                modelTemplate.getMethodsByName("getProbabilityMap").get(0);
        BlockStmt body = methodDeclaration.getBody().get();
        assertNotNull(body.getStatements());
        assertFalse(body.getStatements().isEmpty());
    });
}