public static Map<String, String> getKiePMMLTreeModelSourcesMap(final DroolsCompilationDTO<TreeModel> compilationDTO) {
        logger.trace("getKiePMMLTreeModelSourcesMap {} {} {}", compilationDTO.getFields(),
                     compilationDTO.getModel(), compilationDTO.getPackageName());
        CompilationUnit cloneCU = getKiePMMLModelCompilationUnit(compilationDTO,
                                                                 KIE_PMML_TREE_MODEL_TEMPLATE_JAVA,
                                                                 KIE_PMML_TREE_MODEL_TEMPLATE);
        String className = compilationDTO.getSimpleClassName();
        ClassOrInterfaceDeclaration modelTemplate = cloneCU.getClassByName(className)
                .orElseThrow(() -> new KiePMMLException(MAIN_CLASS_NOT_FOUND + ": " + className));
        setConstructor(compilationDTO,
                       modelTemplate);
        Map<String, String> toReturn = new HashMap<>();
        String fullClassName = compilationDTO.getPackageCanonicalClassName();
        toReturn.put(fullClassName, cloneCU.toString());
        return toReturn;
    }
@Test
    void getKiePMMLScorecardModelSourcesMap() {
        final Map<String, KiePMMLOriginalTypeGeneratedType> fieldTypeMap = getFieldTypeMap(pmml.getDataDictionary(),
                                                                                           pmml.getTransformationDictionary(), treeModel.getLocalTransformations());
        final CommonCompilationDTO<TreeModel> compilationDTO =
                CommonCompilationDTO.fromGeneratedPackageNameAndFields(PACKAGE_NAME, pmml, treeModel,
                                                                       new PMMLCompilationContextMock(), "FILENAME");
        final DroolsCompilationDTO<TreeModel> droolsCompilationDTO =
                DroolsCompilationDTO.fromCompilationDTO(compilationDTO, fieldTypeMap);
        Map<String, String> retrieved = KiePMMLTreeModelFactory.getKiePMMLTreeModelSourcesMap(droolsCompilationDTO);
        assertThat(retrieved).isNotNull();
        assertThat(retrieved).hasSize(1);
    }