@Test
public void getSegmentsSourcesMap() {

    final List<Segment> segments = MINING_MODEL.getSegmentation().getSegments();
    final String packageName = "packagename";
    final List<KiePMMLModel> nestedModels = new ArrayList<>();
    final Map<String, String> retrieved = KiePMMLSegmentFactory.getSegmentsSourcesMap(
            packageName,
            DATA_DICTIONARY,
            TRANSFORMATION_DICTIONARY,
            segments,
            KNOWLEDGE_BUILDER,
            nestedModels);
    assertNotNull(retrieved);
    commonEvaluateNestedModels(nestedModels);
    for (Segment segment : segments) {
        commonEvaluateMap(retrieved, segment);
    }
}