From c7dccd1d8a8d87b73b28ab0e4d364b9f835514dd Mon Sep 17 00:00:00 2001
From: Carlos Vega <carlos.vega@uni.lu>
Date: Tue, 19 Nov 2019 12:24:09 +0100
Subject: [PATCH] added main for mini tests

---
 biokb/biokb.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/biokb/biokb.py b/biokb/biokb.py
index 0f2ac0e..503425f 100644
--- a/biokb/biokb.py
+++ b/biokb/biokb.py
@@ -100,3 +100,10 @@ class BioKBservice(TextMiningService):
             co_occur = CoOccurrence(entity_code, count)
             values.append(co_occur)
         return values
+
+
+if __name__ == "__main__":
+    bkb = BioKBservice()
+    print(bkb.get_co_occurrences('DOID:2841'))
+    print('')
+    print(bkb.get_mentions(['DOID:2841', 'DOID:1205']))
-- 
GitLab