1313 os .path .dirname (__file__ ), "data" , "4eiy_anisouchunk.pdb"
1414)
1515TESTDATA_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
2421TESTDATA_FILENAME_MMCIF = os .path .join (os .path .dirname (__file__ ), "data" , "3eiy.cif" )
2522TESTDATA_FILENAME_MMCIF_GZ = os .path .join (os .path .dirname (__file__ ), "data" , "3eiy.cif.gz" )
@@ -44,11 +41,10 @@ def test_add_pdb_gz():
4441
4542def 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
5349def test_add_mmcif ():
5450 stack = PandasPdbStack ()
@@ -86,7 +82,7 @@ def test_fetch_multiple_pdbs_mixed():
8682def 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