import parasail
from unittest import TestCase, main


class Tests(TestCase):

    def test1(self):
        result = parasail.sw("asdf", "asdf", 10, 1, parasail.blosum62)
        self.assertEqual(result.score, 20)

    def test2(self):
        result = parasail.sw("asdf", "asdf", 10, 1, parasail.pam50)
        self.assertEqual(result.score, 27)

    def test3(self):
        matrix = parasail.matrix_create("acgt", 1, -1)
        result = parasail.sw("acgt", "acgt", 10, 1, matrix)
        self.assertEqual(result.score, 4)

    import pytest
    import platform
    @pytest.mark.skipif(platform.uname().machine=='aarch64', reason="FIXME: test4 ends up in Segmentation fault on arm64")
    def test4(self):
        profile = parasail.profile_create_8("asdf", parasail.blosum62)
        result = parasail.sw_striped_profile_8(profile, "asdf", 10, 1)
        self.assertEqual(result.score, 20)


if __name__ == '__main__':
    main()