Update tests to current Pubchem version (#496)

biotite-dev · Sep 9, 2023 · af815d1 · af815d1
1 parent 67d8016
commit af815d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/biotite/database/pubchem/query.py b/src/biotite/database/pubchem/query.py
@@ -84,7 +84,7 @@ class NameQuery(Query):
     --------
 
     >>> print(search(NameQuery("Alanine")))
-    [5950]
+    [5950, 602, 71080]
     """
 
     def __init__(self, name):
@@ -801,7 +801,7 @@ def search(query, throttle_threshold=0.5, return_throttle_status=False):
     --------
 
     >>> print(search(NameQuery("Alanine")))
-    [5950]
+    [5950, 602, 71080]
     """
     # Use POST to be compatible with the larger payloads
     # of structure searches

diff --git a/tests/database/test_pubchem.py b/tests/database/test_pubchem.py
@@ -85,7 +85,7 @@ def test_fetch_invalid():
 @pytest.mark.parametrize(
     "query, ref_ids",
     [
-        (pubchem.NameQuery("Alanine"), [5950]),
+        (pubchem.NameQuery("Alanine"), [71080, 602, 5950]),
         (pubchem.SmilesQuery("CCCC"), [7843]),
         (pubchem.InchiQuery("InChI=1S/C4H10/c1-3-4-2/h3-4H2,1-2H3"), [7843]),
         (pubchem.InchiKeyQuery("IJDNQMDRQITEOD-UHFFFAOYSA-N"), [7843]),