From f746a9a0005cc5b4341be1b7b8f2ec0b82ebd089 Mon Sep 17 00:00:00 2001
From: Carlos Vega <carlos.vega@uni.lu>
Date: Tue, 19 Nov 2019 15:06:29 +0100
Subject: [PATCH] implemented get_co_occurrences for the DummyService
---
dummy-service/dummyService.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/dummy-service/dummyService.py b/dummy-service/dummyService.py
index a30bf81..e29c494 100644
--- a/dummy-service/dummyService.py
+++ b/dummy-service/dummyService.py
@@ -12,10 +12,16 @@ class DummyService(TextMiningService):
super().__init__("DummyService", "Service that returns dummy results")
def get_mentions(self, entities: List, limit: int = 20) -> List[Publication]:
- return [Publication(pm_id="00000" + str(i)) for i in range(20)]
+ return [Publication(pm_id="00000" + str(i)) for i in range(limit)]
def get_co_occurrences(self, entity: str, limit: int = 20, types: List[str] = None) -> List[CoOccurrence]:
- pass
+ values = []
+ for idx in range(limit):
+ entity_code = str(idx)
+ count = int(idx*idx)
+ co_occur = CoOccurrence(entity_code, count)
+ values.append(co_occur)
+ return values
if __name__ == '__main__':
--
GitLab