__author__ = 'Konrad Kopciuch'

from unittest import TestCase

from Utils.Alignment import Alignment


class AlignmentTest(TestCase):

    def test_get_x(self):
        algn = Alignment("A-CGU-", "-A--CA")
        x = algn.get_template_sequence()
        self.assertEqual("ACGU", x)

    def test_change_x(self):
        algn = Alignment("ACGU-", "-A-C-")
        algn.change_template_sequence("UGCA")
        x = algn.get_template_sequence()
        self.assertEqual("UGCA", x)


