Examples with KiePMMLTestingModelWithSources org.kie.pmml.commons.testingutility.KiePMMLTestingModelWithSources used on opensource projects

Example 1 with KiePMMLTestingModelWithSources

use of org.kie.pmml.commons.testingutility.KiePMMLTestingModelWithSources in project drools by kiegroup.

the class PMMLCompilerServiceTest method populateWithPMMLRuleMappers.

@Test
public void populateWithPMMLRuleMappers() {
    final List<KiePMMLModel> toPopulate = new ArrayList<>();
    toPopulate.add(new KiePMMLTestingModelWithSources("TEST", "kmodulePackageName", Collections.emptyMap()));
    toPopulate.add(new KiePMMLModelHasRule("TEST", Collections.emptyList()));
    toPopulate.add(new KiePMMLModelHasNestedModelsHasRule("TEST", Collections.emptyList()));
    toPopulate.add(new KiePMMLModelHasNestedModelsHasSourceMap("TEST", Collections.emptyList()));
    toPopulate.forEach(kiePMMLModel -> assertTrue(((HasSourcesMap) kiePMMLModel).getSourcesMap().isEmpty()));
    final File file = new File("foo.pmml");
    final Resource resource = new FileSystemResource(file);
    PMMLCompilerService.populateWithPMMLRuleMappers(toPopulate, resource);
    toPopulate.forEach(kiePmmlModel -> {
        if (kiePmmlModel instanceof HasRule || kiePmmlModel instanceof KiePMMLModelHasNestedModelsHasRule) {
            assertFalse(((HasSourcesMap) kiePmmlModel).getSourcesMap().isEmpty());
            String expected = kiePmmlModel.getKModulePackageName() + "." + KIE_PMML_RULE_MAPPERS_CLASS_NAME;
            assertTrue(((HasSourcesMap) kiePmmlModel).getSourcesMap().containsKey(expected));
            if (kiePmmlModel instanceof HasRule) {
                expected = kiePmmlModel.getKModulePackageName() + "." + KIE_PMML_RULE_MAPPER_CLASS_NAME;
                assertTrue(((HasSourcesMap) kiePmmlModel).getSourcesMap().containsKey(expected));
            }
        } else {
            assertTrue(((HasSourcesMap) kiePmmlModel).getSourcesMap().isEmpty());
        }
    });
}

Example 2 with KiePMMLTestingModelWithSources

use of org.kie.pmml.commons.testingutility.KiePMMLTestingModelWithSources in project drools by kiegroup.

the class PMMLCompilerServiceTest method addPMMLRuleMappersHasSourceMap.

@Test
public void addPMMLRuleMappersHasSourceMap() {
    KiePMMLTestingModelWithSources kiePmmlModel = new KiePMMLTestingModelWithSources("TEST", "kmodulePackageName", new HashMap<>());
    assertTrue(kiePmmlModel.getSourcesMap().isEmpty());
    final List<String> generatedRuleMappers = IntStream.range(0, 3).mapToObj(i -> "apackage" + i + "." + KIE_PMML_RULE_MAPPER_CLASS_NAME).collect(Collectors.toList());
    PMMLCompilerService.addPMMLRuleMappers(kiePmmlModel, generatedRuleMappers, "source_path");
    assertFalse(kiePmmlModel.getSourcesMap().isEmpty());
    String expected = kiePmmlModel.getKModulePackageName() + "." + KIE_PMML_RULE_MAPPERS_CLASS_NAME;
    assertTrue(kiePmmlModel.getSourcesMap().containsKey(expected));
}

Example 3 with KiePMMLTestingModelWithSources

use of org.kie.pmml.commons.testingutility.KiePMMLTestingModelWithSources in project drools by kiegroup.

the class PMMLCompilerServiceTest method addPMMLRuleMapperHasSourcesMap.

@Test
public void addPMMLRuleMapperHasSourcesMap() {
    KiePMMLTestingModelWithSources kiePmmlModel = new KiePMMLTestingModelWithSources("TEST", "kmodulePackageName", Collections.emptyMap());
    assertTrue(kiePmmlModel.getSourcesMap().isEmpty());
    final List<String> generatedRuleMappers = IntStream.range(0, 3).mapToObj(i -> "apackage.Rule_" + i).collect(Collectors.toList());
    PMMLCompilerService.addPMMLRuleMapper(kiePmmlModel, generatedRuleMappers, "source_path");
    assertTrue(kiePmmlModel.getSourcesMap().isEmpty());
}