diff --git a/tests/test_prec.py b/tests/test_prec.py index cf186c4..c9ae004 100644 --- a/tests/test_prec.py +++ b/tests/test_prec.py @@ -161,6 +161,14 @@ def test_unmodelled_residues(self, pdb_id): aa_seq_no_ligands = prec.aa_seq.replace(UNKNOWN_AA, "") assert aa_seq_no_ligands == canonical_seq + # Make sure UNP alignment is correct and includes both modelled and + # unmodelled residues + unp_seq = prec.unp_rec.sequence + assert not all(r.unp_idx is None for r in prec) + assert all( + [unp_seq[r.unp_idx] == r.name for r in prec if r.unp_idx is not None] + ) + class TestFromUnp: def test_from_unp_default_selector(self):