pub_soft/AlphaPulldown
3
0
Fork 0
mirror of https://github.com/KosinskiLab/AlphaPulldown.git synced 2026-06-04 14:14:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle grouped identical chains in AF3 trimer test

Browse Source
This commit is contained in:
Dima
2026-04-10 17:48:33 +02:00
parent f7176e87af
commit 7a8afc7c7a
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/cluster/check_alphafold3_predictions.py
View File

@@ -1835,8 +1835,13 @@ class TestAlphaFold3MmseqsIssue588Inference(_TestBase):
self.assertLen(input_json_paths, 1)
written = json.loads(input_json_paths[0].read_text(encoding="utf-8"))
protein_entries = _protein_entries_from_af3_input(written)
self.assertLen(protein_entries, 3)
self.assertLen(protein_entries, 2)
all_chain_ids = []
for protein_entry in protein_entries:
entry_ids = protein_entry["id"]
if isinstance(entry_ids, str):
entry_ids = [entry_ids]
all_chain_ids.extend(entry_ids)
self.assertEqual(
_a3m_query_sequence(protein_entry["pairedMsa"]),
protein_entry["sequence"],
@@ -1845,6 +1850,7 @@ class TestAlphaFold3MmseqsIssue588Inference(_TestBase):
_a3m_query_sequence(protein_entry["unpairedMsa"]),
protein_entry["sequence"],
)
self.assertCountEqual(all_chain_ids, ["A", "B", "C"])
# --------------------------------------------------------------------------- #