use of org.drools.core.impl.InternalRuleUnitExecutor in project drools by kiegroup.
the class MiningmodelTest method testSelectSecondSegmentFirst.
@Test
public void testSelectSecondSegmentFirst() {
RuleUnitExecutor executor = createExecutor(source1);
// KieRuntimeLogger console = ((InternalRuleUnitExecutor)executor).addConsoleLogger();
PMMLRequestData request = new PMMLRequestData("1234", "SampleMine");
request.addRequestParam(new ParameterInfo<>("1234", "fld1", Double.class, 45.0));
request.addRequestParam(new ParameterInfo<>("1234", "fld2", Double.class, 60.0));
request.addRequestParam(new ParameterInfo<>("1234", "fld6", String.class, "optA"));
PMML4Result resultHolder = new PMML4Result();
resultHolder.setCorrelationId(request.getCorrelationId());
DataSource<PMMLRequestData> childModelRequest = executor.newDataSource("childModelRequest");
DataSource<PMML4Result> childModelResults = executor.newDataSource("childModelResults");
DataSource<SegmentExecution> childModelSegments = executor.newDataSource("childModelSegments");
DataSource<? extends AbstractPMMLData> miningModelPojo = executor.newDataSource("miningModelPojo");
List<String> possiblePackages = this.calculatePossiblePackageNames("SampleMine");
Class<? extends RuleUnit> ruleUnitClass = this.getStartingRuleUnit("Start Mining - SampleMine", (InternalKnowledgeBase) kbase, possiblePackages);
assertNotNull(ruleUnitClass);
data.insert(request);
resultData.insert(resultHolder);
executor.run(ruleUnitClass);
// console.close();
resultData.forEach(rd -> {
assertEquals(request.getCorrelationId(), rd.getCorrelationId());
assertEquals("OK", rd.getResultCode());
if (rd.getSegmentationId() == null) {
assertNotNull(rd.getResultValue("Fld5", null));
String value = rd.getResultValue("Fld5", "value", String.class).orElse(null);
assertEquals("tgtZ", value);
AbstractTreeToken token = rd.getResultValue("MissingTreeToken", null, AbstractTreeToken.class).orElse(null);
assertNotNull(token);
assertEquals(0.6, token.getConfidence().doubleValue(), 0.0);
assertEquals("null", token.getCurrent());
}
});
int segmentsExecuted = 0;
for (Iterator<SegmentExecution> iter = childModelSegments.iterator(); iter.hasNext(); ) {
SegmentExecution cms = iter.next();
assertEquals(request.getCorrelationId(), cms.getCorrelationId());
if (cms.getState() == SegmentExecutionState.COMPLETE)
segmentsExecuted++;
}
assertEquals(1, segmentsExecuted);
}
use of org.drools.core.impl.InternalRuleUnitExecutor in project drools by kiegroup.
the class ScorecardTest method testScorecard.
@Test
public void testScorecard() throws Exception {
RuleUnitExecutor executor = createExecutor(source1);
PMMLRequestData requestData = createRequest("123", "Sample Score", 33.0, "SKYDIVER", "KN", true);
PMML4Result resultHolder = new PMML4Result();
List<String> possiblePackages = calculatePossiblePackageNames("Sample Score", "org.drools.scorecards.example");
Class<? extends RuleUnit> unitClass = getStartingRuleUnit("RuleUnitIndicator", (InternalKnowledgeBase) kbase, possiblePackages);
assertNotNull(unitClass);
executor.run(unitClass);
Collection<? extends EntryPoint> eps = ((InternalRuleUnitExecutor) executor).getKieSession().getEntryPoints();
eps.forEach(ep -> {
System.out.println(ep);
});
data.insert(requestData);
resultData.insert(resultHolder);
executor.run(unitClass);
assertEquals(3, resultHolder.getResultVariables().size());
Object scorecard = resultHolder.getResultValue("ScoreCard", null);
assertNotNull(scorecard);
Double score = resultHolder.getResultValue("ScoreCard", "score", Double.class).orElse(null);
assertEquals(41.345, score, 0.000);
Object ranking = resultHolder.getResultValue("ScoreCard", "ranking");
assertNotNull(ranking);
assertTrue(ranking instanceof LinkedHashMap);
LinkedHashMap map = (LinkedHashMap) ranking;
assertTrue(map.containsKey("LX00"));
assertTrue(map.containsKey("RES"));
assertTrue(map.containsKey("CX2"));
assertEquals(-1.0, map.get("LX00"));
assertEquals(-10.0, map.get("RES"));
assertEquals(-30.0, map.get("CX2"));
Iterator iter = map.keySet().iterator();
assertEquals("LX00", iter.next());
assertEquals("RES", iter.next());
assertEquals("CX2", iter.next());
}
Aggregations