示例#1
0
    def test_gotoh_sim(self):
        """Test abydos.distance.Gotoh.sim."""
        self.assertEqual(Gotoh().sim('', ''), 1.0)

        # https://en.wikipedia.org/wiki/Needleman–Wunsch_algorithm
        self.assertEqual(Gotoh(1, 1, _sim_nw).sim('GATTACA', 'GCATGCU'), 0)
        self.assertGreaterEqual(
            Gotoh(1, 0.5, _sim_nw).sim('GATTACA', 'GCATGCU'),
            NeedlemanWunsch(1, _sim_nw).sim('GATTACA', 'GCATGCU'),
        )
        self.assertEqual(
            Gotoh(5, 5, _sim_wikipedia).sim('AGACTAGTTAC', 'CGAGACGT'),
            0.19950186722152657,
        )
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_wikipedia).sim('AGACTAGTTAC', 'CGAGACGT'),
            NeedlemanWunsch(5, _sim_wikipedia).sim('AGACTAGTTAC', 'CGAGACGT'),
        )
示例#2
0
 def test_gotoh_sim_score_nialls(self):
     """Test abydos.distance.Gotoh.sim_score (Nialls set)."""
     # checked against http://ds9a.nl/nwunsch/ (mismatch=1, gap=2, skew=2)
     nw_vals = (5, 0, -2, 3, 1, 1, -2, -2, -1, -3, -3, -5, -3, -7, -7, -19)
     g22 = Gotoh(2, 2, _sim_nw)
     for i in range(len(NIALL)):
         self.assertEqual(g22.sim_score(NIALL[0], NIALL[i]), nw_vals[i])
     nw_vals2 = (5, 0, -2, 3, 1, 1, -2, -2, -1, -2, -3, -3, -2, -6, -6, -8)
     g21 = Gotoh(2, 1, _sim_nw)
     g205 = Gotoh(2, 0.5, _sim_nw)
     nw2 = NeedlemanWunsch(2, _sim_nw)
     for i in range(len(NIALL)):
         self.assertEqual(g21.sim_score(NIALL[0], NIALL[i]), nw_vals2[i])
         self.assertGreaterEqual(
             g205.sim_score(NIALL[0], NIALL[i]),
             nw2.sim_score(NIALL[0], NIALL[i]),
         )
示例#3
0
 def test_gotoh_dist_abs_nialls(self):
     """Test abydos.distance.Gotoh.dist_abs (Nialls set)."""
     # checked against http://ds9a.nl/nwunsch/ (mismatch=1, gap=2, skew=2)
     nw_vals = (5, 0, -2, 3, 1, 1, -2, -2, -1, -3, -3, -5, -3, -7, -7, -19)
     g22 = Gotoh(2, 2, _sim_nw)
     for i in range(len(NIALL)):
         self.assertEqual(g22.dist_abs(NIALL[0], NIALL[i]), nw_vals[i])
     nw_vals2 = (5, 0, -2, 3, 1, 1, -2, -2, -1, -2, -3, -3, -2, -6, -6, -8)
     g21 = Gotoh(2, 1, _sim_nw)
     g205 = Gotoh(2, 0.5, _sim_nw)
     nw2 = NeedlemanWunsch(2, _sim_nw)
     for i in range(len(NIALL)):
         self.assertEqual(g21.dist_abs(NIALL[0], NIALL[i]), nw_vals2[i])
         self.assertGreaterEqual(
             g205.dist_abs(NIALL[0], NIALL[i]),
             nw2.dist_abs(NIALL[0], NIALL[i]),
         )
示例#4
0
    def test_gotoh_sim_score(self):
        """Test abydos.distance.Gotoh.sim_score."""
        self.assertEqual(Gotoh().sim_score('', ''), 0)

        # https://en.wikipedia.org/wiki/Needleman–Wunsch_algorithm
        self.assertEqual(
            Gotoh(1, 1, _sim_nw).sim_score('GATTACA', 'GCATGCU'), 0)
        self.assertGreaterEqual(
            Gotoh(1, 0.5, _sim_nw).sim_score('GATTACA', 'GCATGCU'),
            NeedlemanWunsch(1, _sim_nw).sim_score('GATTACA', 'GCATGCU'),
        )
        self.assertEqual(
            Gotoh(5, 5, _sim_wikipedia).sim_score('AGACTAGTTAC', 'CGAGACGT'),
            16,
        )
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_wikipedia).sim_score('AGACTAGTTAC', 'CGAGACGT'),
            NeedlemanWunsch(5,
                            _sim_wikipedia).sim_score('AGACTAGTTAC',
                                                      'CGAGACGT'),
        )

        # checked against http://ds9a.nl/nwunsch/ (mismatch=1, gap=5, skew=5)
        self.assertEqual(
            Gotoh(5, 5, _sim_nw).sim_score('CGATATCAG', 'TGACGSTGC'), -5)
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_nw).sim_score('CGATATCAG', 'TGACGSTGC'),
            NeedlemanWunsch(5, _sim_nw).sim_score('CGATATCAG', 'TGACGSTGC'),
        )
        self.assertEqual(
            Gotoh(5, 5, _sim_nw).sim_score('AGACTAGTTAC', 'TGACGSTGC'), -7)
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_nw).sim_score('AGACTAGTTAC', 'TGACGSTGC'),
            NeedlemanWunsch(5, _sim_nw).sim_score('AGACTAGTTAC', 'TGACGSTGC'),
        )
        self.assertEqual(
            Gotoh(5, 5, _sim_nw).sim_score('AGACTAGTTAC', 'CGAGACGT'), -15)
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_nw).sim_score('AGACTAGTTAC', 'CGAGACGT'),
            NeedlemanWunsch(5, _sim_nw).sim_score('AGACTAGTTAC', 'CGAGACGT'),
        )

        # Test cases where one of the strings has length 1
        self.assertAlmostEqual(Gotoh().sim_score('', 'a'), -1.0)
        self.assertAlmostEqual(Gotoh().sim_score('abc', 'a'), -0.4)
        self.assertAlmostEqual(Gotoh().sim_score('a', ''), -1.0)
        self.assertAlmostEqual(Gotoh().sim_score('a', 'abc'), -0.4)
示例#5
0
class GotohTestCases(unittest.TestCase):
    """Test Gotoh functions.

    abydos.distance.Gotoh
    """

    cmp = Gotoh()
    nw = NeedlemanWunsch()

    def test_gotoh_dist_abs(self):
        """Test abydos.distance.Gotoh.dist_abs."""
        self.assertEqual(gotoh('', ''), 0)

        # https://en.wikipedia.org/wiki/Needleman–Wunsch_algorithm
        self.assertEqual(
            self.cmp.dist_abs('GATTACA', 'GCATGCU', 1, 1, _sim_nw), 0)
        self.assertGreaterEqual(
            self.cmp.dist_abs('GATTACA', 'GCATGCU', 1, 0.5, _sim_nw),
            self.nw.dist_abs('GATTACA', 'GCATGCU', 1, _sim_nw),
        )
        self.assertEqual(
            self.cmp.dist_abs('AGACTAGTTAC', 'CGAGACGT', 5, 5, _sim_wikipedia),
            16,
        )
        self.assertGreaterEqual(
            self.cmp.dist_abs('AGACTAGTTAC', 'CGAGACGT', 5, 2, _sim_wikipedia),
            self.nw.dist_abs('AGACTAGTTAC', 'CGAGACGT', 5, _sim_wikipedia),
        )

        # checked against http://ds9a.nl/nwunsch/ (mismatch=1, gap=5, skew=5)
        self.assertEqual(
            self.cmp.dist_abs('CGATATCAG', 'TGACGSTGC', 5, 5, _sim_nw), -5)
        self.assertGreaterEqual(
            self.cmp.dist_abs('CGATATCAG', 'TGACGSTGC', 5, 2, _sim_nw),
            self.nw.dist_abs('CGATATCAG', 'TGACGSTGC', 5, _sim_nw),
        )
        self.assertEqual(gotoh('AGACTAGTTAC', 'TGACGSTGC', 5, 5, _sim_nw), -7)
        self.assertGreaterEqual(
            self.cmp.dist_abs('AGACTAGTTAC', 'TGACGSTGC', 5, 2, _sim_nw),
            self.nw.dist_abs('AGACTAGTTAC', 'TGACGSTGC', 5, _sim_nw),
        )
        self.assertEqual(
            self.cmp.dist_abs('AGACTAGTTAC', 'CGAGACGT', 5, 5, _sim_nw), -15)
        self.assertGreaterEqual(
            self.cmp.dist_abs('AGACTAGTTAC', 'CGAGACGT', 5, 2, _sim_nw),
            self.nw.dist_abs('AGACTAGTTAC', 'CGAGACGT', 5, _sim_nw),
        )

        # Test wrapper
        self.assertEqual(
            gotoh('AGACTAGTTAC', 'CGAGACGT', 5, 5, _sim_wikipedia), 16)

    def test_gotoh_dist_abs_nialls(self):
        """Test abydos.distance.Gotoh.dist_abs (Nialls set)."""
        # checked against http://ds9a.nl/nwunsch/ (mismatch=1, gap=2, skew=2)
        nw_vals = (5, 0, -2, 3, 1, 1, -2, -2, -1, -3, -3, -5, -3, -7, -7, -19)
        for i in range(len(NIALL)):
            self.assertEqual(
                self.cmp.dist_abs(NIALL[0], NIALL[i], 2, 2, _sim_nw),
                nw_vals[i],
            )
        nw_vals2 = (5, 0, -2, 3, 1, 1, -2, -2, -1, -2, -3, -3, -2, -6, -6, -8)
        for i in range(len(NIALL)):
            self.assertEqual(
                self.cmp.dist_abs(NIALL[0], NIALL[i], 2, 1, _sim_nw),
                nw_vals2[i],
            )
            self.assertGreaterEqual(
                self.cmp.dist_abs(NIALL[0], NIALL[i], 2, 0.5, _sim_nw),
                self.nw.dist_abs(NIALL[0], NIALL[i], 2, _sim_nw),
            )
示例#6
0
    def test_gotoh_dist_abs(self):
        """Test abydos.distance.Gotoh.dist_abs."""
        self.assertEqual(Gotoh().dist_abs('', ''), 0)

        # https://en.wikipedia.org/wiki/Needleman–Wunsch_algorithm
        self.assertEqual(
            Gotoh(1, 1, _sim_nw).dist_abs('GATTACA', 'GCATGCU'), 0)
        self.assertGreaterEqual(
            Gotoh(1, 0.5, _sim_nw).dist_abs('GATTACA', 'GCATGCU'),
            NeedlemanWunsch(1, _sim_nw).dist_abs('GATTACA', 'GCATGCU'),
        )
        self.assertEqual(
            Gotoh(5, 5, _sim_wikipedia).dist_abs('AGACTAGTTAC', 'CGAGACGT'),
            16)
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_wikipedia).dist_abs('AGACTAGTTAC', 'CGAGACGT'),
            NeedlemanWunsch(5,
                            _sim_wikipedia).dist_abs('AGACTAGTTAC',
                                                     'CGAGACGT'),
        )

        # checked against http://ds9a.nl/nwunsch/ (mismatch=1, gap=5, skew=5)
        self.assertEqual(
            Gotoh(5, 5, _sim_nw).dist_abs('CGATATCAG', 'TGACGSTGC'), -5)
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_nw).dist_abs('CGATATCAG', 'TGACGSTGC'),
            NeedlemanWunsch(5, _sim_nw).dist_abs('CGATATCAG', 'TGACGSTGC'),
        )
        self.assertEqual(
            Gotoh(5, 5, _sim_nw).dist_abs('AGACTAGTTAC', 'TGACGSTGC'), -7)
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_nw).dist_abs('AGACTAGTTAC', 'TGACGSTGC'),
            NeedlemanWunsch(5, _sim_nw).dist_abs('AGACTAGTTAC', 'TGACGSTGC'),
        )
        self.assertEqual(
            Gotoh(5, 5, _sim_nw).dist_abs('AGACTAGTTAC', 'CGAGACGT'), -15)
        self.assertGreaterEqual(
            Gotoh(5, 2, _sim_nw).dist_abs('AGACTAGTTAC', 'CGAGACGT'),
            NeedlemanWunsch(5, _sim_nw).dist_abs('AGACTAGTTAC', 'CGAGACGT'),
        )

        # Test wrapper
        self.assertEqual(
            gotoh('AGACTAGTTAC', 'CGAGACGT', 5, 5, _sim_wikipedia), 16)