@Test
public void getSegmentationSourcesMap() {

    final String segmentationName = "SEGMENTATION_NAME";
    final String packageName = "packagename";
    final List<KiePMMLModel> nestedModels = new ArrayList<>();
    final Map<String, String> retrieved = KiePMMLSegmentationFactory.getSegmentationSourcesMap(packageName,
                                                                                               DATA_DICTIONARY,
                                                                                               TRANSFORMATION_DICTIONARY,
                                                                                               MINING_MODEL.getSegmentation(),
                                                                                               segmentationName,
                                                                                               KNOWLEDGE_BUILDER,
                                                                                               nestedModels);
    assertNotNull(retrieved);
    int expectedNestedModels = MINING_MODEL.getSegmentation().getSegments().size();
    assertEquals(expectedNestedModels, nestedModels.size());
}