Merge pull request #713 from padix-key/database-update

Adapt test to database update
biotite-dev · Dec 9, 2024 · 05e40cb · 05e40cb
2 parents dce750d + 3fb0bf8
commit 05e40cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/database/test_rcsb.py b/tests/database/test_rcsb.py
@@ -174,8 +174,8 @@ def test_search_motif():
     # motif is taken from official RCSB search API tutorial
     MOTIF = "C-x(2,4)-C-x(3)-[LIVMFYWC]-x(8)-H-x(3,5)-H."
     query = rcsb.MotifQuery(MOTIF, "prosite", "protein")
-    test_count = rcsb.count(query)
-    assert test_count == pytest.approx(639, rel=0.1)
+    test_count = rcsb.count(query, return_type="polymer_entity")
+    assert test_count == pytest.approx(719, rel=0.1)
 
 
 @pytest.mark.skipif(cannot_connect_to(RCSB_URL), reason="RCSB PDB is not available")