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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
#!/usr/bin/env python
# This code is part of the Biopython distribution and governed by its
# license. Please see the LICENSE file that should have been included
# as part of this package.
__version__ = "$Revision: 1.11 $"
import cStringIO
import doctest, unittest
import sys
if sys.modules.has_key('requires_wise'):
del sys.modules['requires_wise']
import requires_wise
from Bio import Wise
class TestWiseDryRun(unittest.TestCase):
def setUp(self):
self.old_stdout = sys.stdout
sys.stdout = cStringIO.StringIO()
def test_dnal(self):
"""Call dnal, and do a trivial check on its output."""
Wise.align(["dnal"], ("seq1.fna", "seq2.fna"), kbyte=100000, dry_run=True)
self.assert_(sys.stdout.getvalue().startswith("dnal -kbyte 100000 seq1.fna seq2.fna"))
def test_psw(self):
"""Call psw, and do a trivial check on its output."""
Wise.align(["psw"], ("seq1.faa", "seq2.faa"), dry_run=True, kbyte=4)
self.assert_(sys.stdout.getvalue().startswith("psw -kbyte 4 seq1.faa seq2.faa"))
def tearDown(self):
sys.stdout = self.old_stdout
class TestWise(unittest.TestCase):
def test_align(self):
"""Call dnal with optional arguments, and do a trivial check on the output."""
temp_file = Wise.align(["dnal"], ("Wise/human_114_g01_exons.fna_01", "Wise/human_114_g02_exons.fna_01"), kbyte=100000, force_type="DNA", quiet=True)
line = temp_file.readline().rstrip()
if line == "Score 114":
#Wise 2.4.1 includes a score line, even in quiet mode, ignore this
line = temp_file.readline().rstrip()
if line == "ENSG00000172135 AGGGAAAGCCCCTAAGCTC--CTGATCTATGCTGCATCCAGTTTGCAAAGTGGGGTCCC":
#This is what we expect from wise 2.2.0 (and earlier)
pass
elif line == "ENSG00000172135 AGGGAAAGCCCCTAAGCTC--CTGATCTATGCTGCATCCAGTTTGCAAAG-TGGGGTCC":
#This is what we expect from wise 2.4.1
pass
else:
#Bad!
self.assert_(False, line)
if __name__ == "__main__":
unittest_suite = unittest.TestLoader().loadTestsFromName("test_Wise")
doctest_suite = doctest.DocTestSuite(Wise)
suite = unittest.TestSuite((unittest_suite, doctest_suite))
runner = unittest.TextTestRunner(sys.stdout, verbosity = 2)
runner.run(suite)
|
