diff --git a/src/biotite/database/rcsb/query.py b/src/biotite/database/rcsb/query.py index 972cf1423..c6dba94ec 100644 --- a/src/biotite/database/rcsb/query.py +++ b/src/biotite/database/rcsb/query.py @@ -439,9 +439,9 @@ class StructureQuery(SingleQuery): Examples -------- - >>> query = StructureQuery("1L2Y", chain="A") + >>> query = StructureQuery("1GYA", assembly="1") >>> print(sorted(search(query))) - ['1L2Y', '1RIJ', '2JOF', '2LDJ', '2M7D', '7MQS', '9DPF'] + ['1CDB', '1GYA', '1QA9'] """ def __init__(self, pdb_id, chain=None, assembly=None, strict=True): @@ -452,7 +452,7 @@ def __init__(self, pdb_id, chain=None, assembly=None, strict=True): ): raise TypeError("Either the chain ID or assembly ID must be set") elif chain is None: - self._value = {"entry_id": pdb_id, "asssembly_id": assembly} + self._value = {"entry_id": pdb_id, "assembly_id": assembly} else: self._value = {"entry_id": pdb_id, "asym_id": chain} diff --git a/tests/database/test_rcsb.py b/tests/database/test_rcsb.py index c20a5eca9..24dffc057 100644 --- a/tests/database/test_rcsb.py +++ b/tests/database/test_rcsb.py @@ -196,8 +196,9 @@ def test_search_sequence(): @pytest.mark.skipif(cannot_connect_to(RCSB_URL), reason="RCSB PDB is not available") -def test_search_structure(): - query = rcsb.StructureQuery("1L2Y", chain="A") +@pytest.mark.parametrize("chain, assembly", [("A", None), (None, "1")]) +def test_search_structure(chain, assembly): + query = rcsb.StructureQuery("1L2Y", chain=chain, assembly=assembly) test_ids = rcsb.search(query) assert "1L2Y" in test_ids