@Test
void classDeclaration() {

    ClassOrInterfaceDeclaration retrieved = predictionContainerGenerator.classDeclaration();
    assertNotNull(retrieved);
    String retrievedString = retrieved.toString();
    String expected = PMML_RESOURCES
            .stream()
            .map(pmmlResource ->  "\"" + pmmlResource.getModelPath() + "\"")
            .collect(Collectors.joining(", "));
    expected = String.format("org.kie.kogito.pmml.PMMLKogito.createKieRuntimeFactories(%s);", expected);
    assertTrue(retrievedString.contains(expected));
}