def test_get_subalignment_sequence_order_maintained(self):
"""
Sequences given rearranged are still output in input order
"""
result = AlignedSeq.get_sub_alignment_by_list_id(["s3", "s1"], self.alignment)
expected = MultipleSeqAlignment([self.alignment[0], self.alignment[2]])
self.assertTrue(msas_equal(expected, result))
def test_get_subalignment_with_interval(self):
result = AlignedSeq.get_sub_alignment_by_list_id(
["s2", "s3"], self.alignment, [0, 2]
)
expected = MultipleSeqAlignment(
[SeqRecord(Seq("C--"), id="s2"), SeqRecord(Seq("AAT"), id="s3"),]
)
self.assertTrue(msas_equal(expected, result))
def test_get_subalignment_sequence_order_maintained2(self):
result = AlignedSeq.get_sub_alignment_by_list_id(["s1", "s3"], self.alignment)
expected = MultipleSeqAlignment([self.alignment[0], self.alignment[2]])
self.assertTrue(msas_equal(expected, result))