import unittest

import alignementseq_multiple

from Bio.SeqRecord import SeqRecord

class MultipleAlignTestCase(unittest.TestCase):
    def test_align_from_tree(self):
        self.assertEqual(
            alignementseq_multiple.multiple_align_from_tree((("CHAT", "CAT"), "HER")),
            ['CHAT', 'C-AT', 'H-ER']
            )
    def test_align_from_linkage_matrix(self):
        import tree_generator
        import alignementseq

        seqs = [ SeqRecord(s) for s in ("CHAT", "CAT", "HER") ]
        tree = tree_generator.tree_build(seqs, alignementseq.vec_align)
        aseqs = alignementseq_multiple.multiple_align_from_linkage_matrix(
                seqs,
                tree
            )
        aseqs_ref = [ "CHAT", "C-AT", "H-ER" ]
        aseqs_ref.sort()
        aseqs.sort()
        
        self.assertEqual(aseqs_ref, aseqs)