Example 1 with KIE_PMML_RULE_MAPPERS_CLASS_NAME
use of org.kie.pmml.evaluator.assembler.factories.PMMLRuleMappersFactory.KIE_PMML_RULE_MAPPERS_CLASS_NAME 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));
}
Also used :
IntStream(java.util.stream.IntStream)
HasRule(org.kie.pmml.commons.HasRule)
HashMap(java.util.HashMap)
FileSystemResource(org.drools.core.io.impl.FileSystemResource)
ArrayList(java.util.ArrayList)
KiePMMLModel(org.kie.pmml.commons.model.KiePMMLModel)
StringUtils.generateUUID(org.drools.core.util.StringUtils.generateUUID)
KIE_PMML_RULE_MAPPER_CLASS_NAME(org.kie.pmml.evaluator.assembler.factories.PMMLRuleMapperFactory.KIE_PMML_RULE_MAPPER_CLASS_NAME)
MINING_FUNCTION(org.kie.pmml.api.enums.MINING_FUNCTION)
KiePMMLTestingModelWithSources(org.kie.pmml.commons.testingutility.KiePMMLTestingModelWithSources)
HasNestedModels(org.kie.pmml.commons.model.HasNestedModels)
Assert.assertTrue(org.junit.Assert.assertTrue)
Test(org.junit.Test)
Collectors(java.util.stream.Collectors)
File(java.io.File)
Resource(org.kie.api.io.Resource)
List(java.util.List)
HasSourcesMap(org.kie.pmml.commons.model.HasSourcesMap)
KiePMMLTestingModel(org.kie.pmml.commons.testingutility.KiePMMLTestingModel)
Assert.assertFalse(org.junit.Assert.assertFalse)
KiePMMLExtension(org.kie.pmml.commons.model.KiePMMLExtension)
Collections(java.util.Collections)
KiePMMLException(org.kie.pmml.api.exceptions.KiePMMLException)
Assert.assertEquals(org.junit.Assert.assertEquals)
KIE_PMML_RULE_MAPPERS_CLASS_NAME(org.kie.pmml.evaluator.assembler.factories.PMMLRuleMappersFactory.KIE_PMML_RULE_MAPPERS_CLASS_NAME)
KiePMMLTestingModelWithSources(org.kie.pmml.commons.testingutility.KiePMMLTestingModelWithSources)
Test(org.junit.Test)
Aggregations
File (java.io.File)1
ArrayList (java.util.ArrayList)1
Collections (java.util.Collections)1
HashMap (java.util.HashMap)1
List (java.util.List)1
Collectors (java.util.stream.Collectors)1
IntStream (java.util.stream.IntStream)1
FileSystemResource (org.drools.core.io.impl.FileSystemResource)1
StringUtils.generateUUID (org.drools.core.util.StringUtils.generateUUID)1
Assert.assertEquals (org.junit.Assert.assertEquals)1
Assert.assertFalse (org.junit.Assert.assertFalse)1
Assert.assertTrue (org.junit.Assert.assertTrue)1
Test (org.junit.Test)1
Resource (org.kie.api.io.Resource)1
MINING_FUNCTION (org.kie.pmml.api.enums.MINING_FUNCTION)1
KiePMMLException (org.kie.pmml.api.exceptions.KiePMMLException)1
HasRule (org.kie.pmml.commons.HasRule)1
HasNestedModels (org.kie.pmml.commons.model.HasNestedModels)1
HasSourcesMap (org.kie.pmml.commons.model.HasSourcesMap)1
KiePMMLExtension (org.kie.pmml.commons.model.KiePMMLExtension)1
