@Test
public void getSegments() {

    final List<Segment> segments = MINING_MODEL.getSegmentation().getSegments();
    final List<KiePMMLSegment> retrieved = KiePMMLSegmentFactory.getSegments(DATA_DICTIONARY,
                                                                             TRANSFORMATION_DICTIONARY,
                                                                             segments,
                                                                             KNOWLEDGE_BUILDER);
    assertNotNull(retrieved);
    assertEquals(segments.size(), retrieved.size());
    for (int i = 0; i < segments.size(); i++) {
        commonEvaluateSegment(retrieved.get(i), segments.get(i));
    }
}