1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
import unittest
from ost import *
from ost import settings
from ost import seq
class TestWeightMatrix(unittest.TestCase):
def test_GetWeight(self):
mat = seq.alg.BLOSUM62
self.assertEqual(mat.GetWeight('A', 'A'), 4)
self.assertEqual(mat.GetWeight('A', 'B'), -2)
self.assertEqual(mat.GetWeight('a', 'A'), 4)
self.assertEqual(mat.GetWeight('A', 'b'), -2)
self.assertEqual(mat.GetWeight('A', '?'), 0)
self.assertEqual(mat.GetWeight('E', '?'), 0)
self.assertEqual(mat.GetWeight('Y', '?'), 0)
self.assertEqual(mat.GetWeight('?', 'A'), 0)
self.assertEqual(mat.GetWeight('?', 'E'), 0)
self.assertEqual(mat.GetWeight('?', 'Y'), 0)
self.assertEqual(mat.GetWeight('?', 'y'), 0)
def test_SetWeight(self):
mat = seq.alg.BLOSUM62
self.assertEqual(mat.GetWeight('A', 'A'), 4)
mat.SetWeight('A', 'A', -1)
self.assertEqual(mat.GetWeight('A', 'A'), -1)
self.assertEqual(mat.GetWeight('A', 'B'), -2)
mat.SetWeight('A', 'B', 10)
self.assertEqual(mat.GetWeight('A', 'B'), 10)
self.assertEqual(mat.GetWeight('E', '?'), 0)
mat.SetWeight('E', '?', 10)
self.assertEqual(mat.GetWeight('E', '?'), 0)
self.assertEqual(mat.GetWeight('?', 'E'), 0)
mat.SetWeight('?', 'E', 10)
self.assertEqual(mat.GetWeight('?', 'E'), 0)
if __name__ == "__main__":
from ost import testutils
testutils.RunTests()
|
