diff --git a/elk-snomed-owl/src/test/java/dev/ikm/elk/snomed/owl/IncrementalClassifierCldUs20230301TestIT.java b/elk-snomed-owl/src/test/java/dev/ikm/elk/snomed/owl/IncrementalClassifierCldUs20230301TestIT.java index 8e895d28f..f659788ec 100644 --- a/elk-snomed-owl/src/test/java/dev/ikm/elk/snomed/owl/IncrementalClassifierCldUs20230301TestIT.java +++ b/elk-snomed-owl/src/test/java/dev/ikm/elk/snomed/owl/IncrementalClassifierCldUs20230301TestIT.java @@ -101,13 +101,13 @@ public SnomedOwlOntology loadOntology() throws Exception { return ontology; } - private HashMap> getSuperClassMap(SnomedOwlOntology ontology) { + private MutableLongObjectMap getSuperClassMap(SnomedOwlOntology ontology) { HashMap> superConcepts = new HashMap<>(); for (OWLClass clazz : ontology.getOntology().getClassesInSignature()) { long id = SnomedOwlOntology.getId(clazz); superConcepts.put(id, ontology.getSuperClasses(id)); } - return superConcepts; + return convertToEclipseCollections(superConcepts); } private MutableLongObjectMap convertToEclipseCollections(HashMap> map) { @@ -139,8 +139,8 @@ private SnomedOwlOntology ontologyEquivalentClasses() throws Exception { private SnomedIsa classifyEquivalentClasses() throws Exception { SnomedOwlOntology ontology = ontologyEquivalentClasses(); ontology.classify(); - HashMap> supers = getSuperClassMap(ontology); - return SnomedIsa.init(convertToEclipseCollections(supers)); + MutableLongObjectMap supers = getSuperClassMap(ontology); + return SnomedIsa.init(supers); } private void toSubClassOf(SnomedOwlOntology ontology) throws Exception { @@ -162,8 +162,8 @@ private SnomedOwlOntology ontologySubClassOf() throws Exception { private SnomedIsa classifySubClassOf() throws Exception { SnomedOwlOntology ontology = ontologySubClassOf(); ontology.classify(); - HashMap> supers = getSuperClassMap(ontology); - return SnomedIsa.init(convertToEclipseCollections(supers)); + MutableLongObjectMap supers = getSuperClassMap(ontology); + return SnomedIsa.init(supers); } @Test