Skip to content

Commit ed1c6a2

Browse files
committed
updated af database version
1 parent 5896ab5 commit ed1c6a2

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

tests/stack/test_read_stack.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,10 @@
1313
os.path.dirname(__file__), "data", "4eiy_anisouchunk.pdb"
1414
)
1515
TESTDATA_FILENAME_GZ = os.path.join(os.path.dirname(__file__), "data", "3eiy.pdb.gz")
16-
TESTDATA_FILENAME_AF2_V4 = os.path.join(
17-
os.path.dirname(__file__), "data", "AF-Q5VSL9-F1-model_v4.pdb"
16+
TESTDATA_FILENAME_AF2_V6 = os.path.join(
17+
os.path.dirname(__file__), "data", "AF-Q5VSL9-F1-model_v6.pdb"
1818
)
1919

20-
TESTDATA_FILENAME_AF2_V3 = os.path.join(
21-
os.path.dirname(__file__), "data", "AF-Q5VSL9-F1-model_v3.pdb"
22-
)
2320

2421
TESTDATA_FILENAME_MMCIF = os.path.join(os.path.dirname(__file__), "data", "3eiy.cif")
2522
TESTDATA_FILENAME_MMCIF_GZ = os.path.join(os.path.dirname(__file__), "data", "3eiy.cif.gz")
@@ -44,11 +41,10 @@ def test_add_pdb_gz():
4441

4542
def test_add_multiple_pdbs():
4643
stack = PandasPdbStack()
47-
stack.add_pdbs([TESTDATA_FILENAME, TESTDATA_FILENAME_AF2_V4, TESTDATA_FILENAME_AF2_V3])
48-
assert len(stack.pdbs.keys()) == 3
44+
stack.add_pdbs([TESTDATA_FILENAME, TESTDATA_FILENAME_AF2_V6])
45+
assert len(stack.pdbs.keys()) == 2
4946
assert '3eiy' in stack.pdbs
50-
assert 'AF-Q5VSL9-F1-model_v4' in stack.pdbs
51-
assert 'AF-Q5VSL9-F1-model_v3' in stack.pdbs
47+
assert 'AF-Q5VSL9-F1-model_v6' in stack.pdbs
5248

5349
def test_add_mmcif():
5450
stack = PandasPdbStack()
@@ -86,7 +82,7 @@ def test_fetch_multiple_pdbs_mixed():
8682
def test_add_multiple_structures_mixed():
8783
stack = PandasPdbStack()
8884
stack.add_pdbs([TESTDATA_FILENAME, TESTDATA_FILENAME_GZ,
89-
TESTDATA_FILENAME_AF2_V4,
85+
TESTDATA_FILENAME_AF2_V6,
9086
TESTDATA_FILENAME_MMCIF, TESTDATA_FILENAME_MMCIF_GZ,
9187
'Q5VSL9', 'P99999', '1YCR', '6V4E'])
9288
# There are multiple 3eiy, they overwrite each other, hence the length of dict is not 9

0 commit comments

Comments
 (0)