Skip to content
Snippets Groups Projects
Commit 51030e4f authored by Tobias Schmidt's avatar Tobias Schmidt
Browse files

add missing weight matrix test file

parent 61acb788
Branches
Tags
No related merge requests found
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__":
try:
unittest.main()
except Exception, e:
print e
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment