@Test
public void getRegressionTablesMap() throws IOException {

    String targetFieldName = "targetFieldName";
    Map<String, KiePMMLTableSourceCategory> retrieved = KiePMMLRegressionModelFactory
            .getRegressionTablesMap(dataDictionary,
                                    regressionModel,
                                    targetFieldName,
                                    Collections.emptyList(),
                                    PACKAGE_NAME);
    int expectedSize = regressionTables.size() + 1; // One for classification
    assertEquals(expectedSize, retrieved.size());
    final Collection<KiePMMLTableSourceCategory> values = retrieved.values();
    regressionTables.forEach(regressionTable ->
                                     assertTrue(values.stream().anyMatch(kiePMMLTableSourceCategory -> kiePMMLTableSourceCategory.getCategory().equals(regressionTable.getTargetCategory()))));
}