Merge pull request #86 from PathwayCommons/ncbi_tests

Ncbi tests
PathwayCommons · Apr 26, 2021 · 657716e · 657716e
2 parents 613032c + 16bad25
commit 657716e
Show file tree

Hide file tree

Showing 2 changed files with 2,155 additions and 6 deletions.
diff --git a/semantic_search/ncbi.py b/semantic_search/ncbi.py
@@ -72,11 +72,11 @@ def _parse_medline(text: str) -> List[dict]:
     return medline_records
 
 
-def _get_eutil_records(eutil: str, id: List[str], **opts) -> List[Dict[Any, Any]]:
+def _get_eutil_records(eutil: str, _id: List[str], **opts) -> List[Dict[Any, Any]]:
     """Call one of the NCBI EUTILITIES and returns data as Python objects."""
     eutils_params = {
         "db": "pubmed",
-        "id": ",".join(id),
+        "id": ",".join(_id),
         "retstart": 0,
         "retmode": "xml",
         "api_key": settings.ncbi_eutils_api_key,
@@ -116,10 +116,10 @@ def uids_to_docs(uids: List[str]) -> Generator[List[Dict[str, str]], None, None]
     for i in range(num_queries):
         lower = i * MAX_EFETCH_RETMAX
         upper = min([lower + MAX_EFETCH_RETMAX, num_uids])
-        id = uids[lower:upper]
+        _id = uids[lower:upper]
         try:
             start_time = time.time()
-            eutil_response = _get_eutil_records("efetch", id, rettype="medline", retmode="text")
+            eutil_response = _get_eutil_records("efetch", _id, rettype="medline", retmode="text")
             duration = time.time() - start_time
             logging.info(
                 f"Retrieved docs {lower} through {upper - 1} of {num_uids - 1} in {duration}s"