Return 0 count for frequency query when term is not matched in SPARQL…

… query. Fixes #191.
phenoscape · Mar 26, 2020 · 2543398 · 2543398
1 parent 6924182
commit 2543398
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/scala/org/phenoscape/kb/Similarity.scala b/src/main/scala/org/phenoscape/kb/Similarity.scala
@@ -303,7 +303,11 @@ object Similarity {
     }
     App.executeSPARQLQueryString(query.text, qs =>
       IRI.create(qs.getResource("term").getURI) ->
-        qs.getLiteral("count").getInt).map(_.toMap)
+        qs.getLiteral("count").getInt).map(_.toMap).map { result =>
+      val foundTerms = result.keySet
+      // add in 0 counts for terms not returned by the query
+      result ++ (terms -- foundTerms).map(_ -> 0)
+    }
   }
 
   type TermFrequencyTable = Map[IRI, Int]