public static Map<String, String> getKiePMMLClusteringModelSourcesMap(final ClusteringCompilationDTO compilationDTO) {

        logger.trace("getKiePMMLClusteringModelSourcesMap {}", compilationDTO);

        String simpleClassName = compilationDTO.getSimpleClassName();

        CompilationUnit compilationUnit = JavaParserUtils.getKiePMMLModelCompilationUnit(simpleClassName,
                                                                                         compilationDTO.getPackageName(),
                                                                                         KIE_PMML_CLUSTERING_MODEL_TEMPLATE_JAVA,
                                                                                         KIE_PMML_CLUSTERING_MODEL_TEMPLATE);
        ClassOrInterfaceDeclaration modelTemplate = compilationUnit.getClassByName(simpleClassName)
                .orElseThrow(() -> new KiePMMLException(MAIN_CLASS_NOT_FOUND + ": " + simpleClassName));
        setStaticGetter(compilationDTO, modelTemplate);
        populateGetClustersMethod(modelTemplate, compilationDTO.getModel());
        populateGetClusteringFieldsMethod(modelTemplate, compilationDTO.getModel());

        Map<String, String> sourcesMap = new HashMap<>();
        sourcesMap.put(getFullClassName(compilationUnit), compilationUnit.toString());

        return sourcesMap;
    }
@Test
    void getKiePMMLClusteringModelSourcesMap() {
        final CommonCompilationDTO<ClusteringModel> compilationDTO =
                CommonCompilationDTO.fromGeneratedPackageNameAndFields(PACKAGE_NAME,
                                                                       pmml,
                                                                       clusteringModel,
                                                                       new PMMLCompilationContextMock(), "fileName");
        Map<String, String> retrieved =
                KiePMMLClusteringModelFactory.getKiePMMLClusteringModelSourcesMap(ClusteringCompilationDTO.fromCompilationDTO(compilationDTO));
        assertThat(retrieved).isNotNull();
        assertThat(retrieved).hasSize(1);
    }