annotate commons/core/seq/test/Test_Bioseq.py @ 18:94ab73e8a190

Uploaded
author m-zytnicki
date Mon, 29 Apr 2013 03:20:15 -0400
parents 769e306b7933
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1 # Copyright INRA (Institut National de la Recherche Agronomique)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
2 # http://www.inra.fr
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
3 # http://urgi.versailles.inra.fr
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
4 #
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
5 # This software is governed by the CeCILL license under French law and
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
6 # abiding by the rules of distribution of free software. You can use,
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
7 # modify and/ or redistribute the software under the terms of the CeCILL
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
8 # license as circulated by CEA, CNRS and INRIA at the following URL
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
9 # "http://www.cecill.info".
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
10 #
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
11 # As a counterpart to the access to the source code and rights to copy,
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
12 # modify and redistribute granted by the license, users are provided only
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
13 # with a limited warranty and the software's author, the holder of the
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
14 # economic rights, and the successive licensors have only limited
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
15 # liability.
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
16 #
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
17 # In this respect, the user's attention is drawn to the risks associated
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
18 # with loading, using, modifying and/or developing or reproducing the
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
19 # software by the user in light of its specific status of free software,
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
20 # that may mean that it is complicated to manipulate, and that also
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
21 # therefore means that it is reserved for developers and experienced
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
22 # professionals having in-depth computer knowledge. Users are therefore
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
23 # encouraged to load and test the software's suitability as regards their
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
24 # requirements in conditions enabling the security of their systems and/or
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
25 # data to be ensured and, more generally, to use and operate it in the
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
26 # same conditions as regards security.
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
27 #
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
28 # The fact that you are presently reading this means that you have had
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
29 # knowledge of the CeCILL license and that you accept its terms.
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
30
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
31
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
32 import unittest
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
33 import os
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
34 import sys
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
35 from commons.core.seq.Bioseq import Bioseq
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
36 from commons.core.utils.FileUtils import FileUtils
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
37 from commons.core.coord.Map import Map
18
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
38 from commons.core.checker.RepetException import RepetException
6
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
39
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
40
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
41 class Test_Bioseq( unittest.TestCase ):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
42
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
43 def setUp(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
44 self._bs = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
45
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
46
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
47 def test_isEmpty_True(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
48 self._bs.setHeader( "" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
49 self._bs.setSequence( "" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
50 exp = True
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
51 obs = self._bs.isEmpty()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
52 self.assertEquals( exp, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
53
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
54
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
55 def test_isEmpty_False(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
56 self._bs.setHeader( "seq1" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
57 self._bs.setSequence( "AGCGGACGATGCAGCATGCGAATGACGAT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
58 exp = False
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
59 obs = self._bs.isEmpty()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
60 self.assertEquals( exp, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
61
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
62
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
63 def test___eq__(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
64 self._bs.setHeader( "seq1" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
65 self._bs.setSequence( "AGCGGACGATGCAGCATGCGAATGACGAT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
66 obs = Bioseq( "seq1", "AGCGGACGATGCAGCATGCGAATGACGAT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
67 self.assertEquals( self._bs, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
68
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
69
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
70 def test___ne__Header(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
71 self._bs.setHeader( "seq2" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
72 self._bs.setSequence( "AGCGGACGATGCAGCATGCGAATGACGAT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
73 obs = Bioseq( "seq1", "AGCGGACGATGCAGCATGCGAATGACGAT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
74 self.assertNotEquals( self._bs, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
75
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
76
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
77 def test___ne__Sequence(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
78 self._bs.setHeader( "seq1" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
79 self._bs.setSequence( "GGACGATGCAGCATGCGAATGACGAT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
80 obs = Bioseq( "seq1", "AGCGGACGATGCAGCATGCGAATGACGAT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
81 self.assertNotEquals( self._bs, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
82
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
83
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
84 def test_reverse(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
85 self._bs.setHeader( "seq1" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
86 self._bs.setSequence( "TGCGGA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
87 exp = "AGGCGT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
88 self._bs.reverse()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
89 obs = self._bs.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
90 self.assertEqual( obs, exp )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
91
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
92
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
93 def test_complement(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
94 self._bs.setHeader( "seq1" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
95 self._bs.setSequence( "TGCGGA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
96 exp = "ACGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
97 self._bs.complement()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
98 obs = self._bs.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
99 self.assertEqual( obs, exp )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
100
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
101
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
102 def test_complement_with_unknown_symbol(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
103 self._bs.setHeader( "seq1" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
104 self._bs.setSequence( "TGCGGAFMRWTYSKVHDBN" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
105 exp = "ACGCCTNKYWARSMBDHVN"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
106 self._bs.complement()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
107 obs = self._bs.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
108 self.assertEqual( obs, exp )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
109
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
110
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
111 def test_reverseComplement(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
112 self._bs.setHeader( "seq1" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
113 self._bs.setSequence( "TGCGGA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
114 exp = "TCCGCA"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
115 self._bs.reverseComplement()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
116 obs = self._bs.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
117 self.assertEqual( obs, exp )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
118
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
119
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
120 def test_cleanGap(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
121 self._bs.setSequence("-ATTTTGC-AGTC--TTATTCGAG-----GCCATTGCT-")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
122 exp = "ATTTTGCAGTCTTATTCGAGGCCATTGCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
123 self._bs.cleanGap()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
124 obs = self._bs.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
125 self.assertEquals( obs, exp )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
126
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
127
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
128 def test_copyBioseqInstance(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
129 self._bs.setHeader( "seq" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
130 self._bs.setSequence( "TGCGGA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
131 obsBioseq = self._bs.copyBioseqInstance()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
132 self.assertEquals(self._bs, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
133
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
134
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
135 def test_setFrameInfoOnHeader_without_description(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
136 self._bs.setHeader( "seq" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
137 self._bs.setSequence( "TGCGGA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
138 phase = -1
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
139 expHeader = "seq_-1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
140 self._bs.setFrameInfoOnHeader(phase)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
141 self.assertEquals(expHeader, self._bs.header)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
142
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
143
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
144 def test_setFrameInfoOnHeader_with_description(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
145 self._bs.setHeader( "seq description" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
146 self._bs.setSequence( "TGCGGA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
147 phase = -1
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
148 expHeader = "seq_-1 description"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
149 self._bs.setFrameInfoOnHeader(phase)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
150 self.assertEquals(expHeader, self._bs.header)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
151
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
152
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
153 def test_read(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
154 faFile = open("dummyFaFile.fa", "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
155 faFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
156 faFile.write("ATGCGTCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
157 faFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
158 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
159 expBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
160 expBioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
161 expBioseq.sequence = "ATGCGTCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
162 obsBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
163 faFile = open("dummyFaFile.fa", "r")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
164 obsBioseq.read( faFile )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
165 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
166 os.remove("dummyFaFile.fa")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
167 self.assertEquals(expBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
168
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
169
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
170 def test_read_WithEmptyFile(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
171 faFile = open("dummyFaFile.fa", "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
172 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
173 expBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
174 expBioseq.header = None
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
175 expBioseq.sequence = None
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
176 obsBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
177 faFile = open("dummyFaFile.fa", "r")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
178 obsBioseq.read( faFile )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
179 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
180 os.remove("dummyFaFile.fa")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
181 self.assertEquals(expBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
182
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
183
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
184 def test_read_without_header(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
185 faFile = open("dummyFaFile.fa", "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
186 faFile.write("seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
187 faFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
188 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
189 expBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
190 expBioseq.header = ""
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
191 expBioseq.sequence = ""
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
192 obsBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
193 faFile = open("dummyFaFile.fa", "r")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
194 obsBioseq.read( faFile )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
195 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
196 os.remove("dummyFaFile.fa")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
197 self.assertEquals(expBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
198
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
199
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
200 def test_read_with_two_consecutive_headers(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
201 faFile = open("dummyFaFile.fa", "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
202 faFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
203 faFile.write(">ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
204 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
205 expBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
206 expBioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
207 expBioseq.sequence = ""
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
208 obsBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
209 faFile = open("dummyFaFile.fa", "r")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
210 obsBioseq.read( faFile )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
211 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
212 os.remove("dummyFaFile.fa")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
213 self.assertEquals(expBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
214
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
215
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
216 def test_read_withEmptyLines(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
217 faFile = open("dummyFaFile.fa", "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
218 faFile.write("\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
219 faFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
220 faFile.write("ATGCGTCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
221 faFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
222 faFile.write("\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
223 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
224
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
225 exp = Bioseq( "seq1 description1", "ATGCGTCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
226
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
227 obs = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
228 faFile = open("dummyFaFile.fa", "r")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
229 obs.read( faFile )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
230 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
231
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
232 os.remove("dummyFaFile.fa")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
233
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
234 self.assertEquals( exp, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
235
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
236 def test_read_with_70nt_by_line(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
237 faFile = open("dummyFaFile.fa", "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
238 faFile.write("\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
239 faFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
240 faFile.write("TGTCACATCCTGATTTTCGTTTCAGGATTTATAAATTATTTAATAAATTAATAATAGAATTTATATTAAA\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
241 faFile.write("TGTTTTTTAATTTACAAGTGAAGTTAAATGTGGGAAATAAAATTTCTTAAATCTAAAGCATGGATGGATT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
242 faFile.write("\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
243 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
244
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
245 exp = Bioseq( "seq1 description1", "TGTCACATCCTGATTTTCGTTTCAGGATTTATAAATTATTTAATAAATTAATAATAGAATTTATATTAAATGTTTTTTAATTTACAAGTGAAGTTAAATGTGGGAAATAAAATTTCTTAAATCTAAAGCATGGATGGATT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
246
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
247 obs = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
248 faFile = open("dummyFaFile.fa", "r")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
249 obs.read( faFile )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
250 faFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
251
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
252 os.remove("dummyFaFile.fa")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
253
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
254 self.assertEquals( exp, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
255
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
256 def test_appendBioseqInFile(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
257 obsFaFileName = "dummyFaFile.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
258 obsFaFile = open(obsFaFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
259 obsFaFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
260 obsFaFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
261 obsFaFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
262
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
263 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
264 bioseq.header = "seq2 description2"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
265 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
266
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
267 expFaFileName = "dummyFaFile2.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
268 expFaFile = open(expFaFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
269 expFaFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
270 expFaFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
271 expFaFile.write(">seq2 description2\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
272 expFaFile.write("GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
273 expFaFile.write("ATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
274 expFaFile.write("ATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
275 expFaFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
276
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
277 bioseq.appendBioseqInFile(obsFaFileName)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
278 self.assertTrue(FileUtils.are2FilesIdentical(expFaFileName, obsFaFileName))
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
279 os.remove(obsFaFileName)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
280 os.remove(expFaFileName)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
281
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
282
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
283 def test_writeABioseqInAFastaFile(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
284 obsFaFileName = "dummyFaFile.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
285 obsFaFile = open(obsFaFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
286 obsFaFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
287 obsFaFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
288
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
289 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
290 bioseq.header = "seq2 description2"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
291 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
292
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
293 expFaFileName = "dummyFaFile2.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
294 expFaFile = open(expFaFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
295 expFaFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
296 expFaFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
297 expFaFile.write(">seq2 description2\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
298 expFaFile.write("GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
299 expFaFile.write("ATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
300 expFaFile.write("ATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
301 expFaFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
302
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
303 bioseq.writeABioseqInAFastaFile(obsFaFile)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
304 obsFaFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
305 self.assertTrue(FileUtils.are2FilesIdentical(expFaFileName, obsFaFileName))
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
306 os.remove(obsFaFileName)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
307
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
308
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
309 def test_writeABioseqInAFastaFileWithOtherHeader(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
310 obsFaFileName = "dummyFaFile.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
311 obsFaFile = open(obsFaFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
312 obsFaFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
313 obsFaFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
314
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
315 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
316 bioseq.header = "seq2 description2"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
317 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
318
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
319 expFaFileName = "dummyFaFile2.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
320 newHeader = "seq2 New header2"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
321 expFaFile = open(expFaFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
322 expFaFile.write(">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
323 expFaFile.write("ATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
324 expFaFile.write(">" + newHeader + "\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
325 expFaFile.write("GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
326 expFaFile.write("ATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
327 expFaFile.write("ATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
328 expFaFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
329
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
330 bioseq.writeABioseqInAFastaFileWithOtherHeader(obsFaFile, newHeader)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
331 obsFaFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
332 self.assertTrue(FileUtils.are2FilesIdentical(expFaFileName, obsFaFileName))
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
333 os.remove(obsFaFileName)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
334 os.remove(expFaFileName)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
335
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
336
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
337 def test_writeSeqInFasta(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
338 iBs = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
339 iBs.header = "dummySeq"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
340 iBs.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
341
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
342 expFaFile = "dummyExpFile.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
343 expFaFileHandler = open(expFaFile, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
344 expFaFileHandler.write("GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
345 expFaFileHandler.write("ATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
346 expFaFileHandler.write("ATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
347 expFaFileHandler.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
348
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
349 obsFaFile = "dummyObsFile.fa"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
350 obsFaFileHandler = open( obsFaFile, "w" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
351
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
352 iBs.writeSeqInFasta( obsFaFileHandler )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
353
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
354 obsFaFileHandler.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
355
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
356 self.assertTrue( FileUtils.are2FilesIdentical( expFaFile, obsFaFile ) )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
357 os.remove(obsFaFile)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
358 os.remove(expFaFile)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
359
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
360
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
361 def test_subseq(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
362 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
363 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
364 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
365 start = 10
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
366 end = 30
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
367 expSubBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
368 expSubBioseq.header = "seq1 description1 fragment " + str(start) + ".." + str(end)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
369 expSubBioseq.sequence = bioseq.sequence[(start - 1) : end]
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
370 obsBioseq = bioseq.subseq(start, end)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
371 self.assertEquals(expSubBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
372
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
373
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
374 def test_subseq_no_end(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
375 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
376 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
377 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
378 start = 10
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
379 expSubBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
380 expEnd = len(bioseq.sequence)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
381 expSubBioseq.header = "seq1 description1 fragment " + str(start) + ".." + str(expEnd)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
382 expSubBioseq.sequence = bioseq.sequence[(start - 1) : expEnd]
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
383 obsBioseq = bioseq.subseq(start)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
384 self.assertEquals(expSubBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
385
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
386
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
387 def test_subseq_start_gt_end(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
388 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
389 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
390 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
391 start = 30
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
392 end = 10
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
393 expSubBioseq = None
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
394 obsBioseq = bioseq.subseq(start, end)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
395 self.assertEquals(expSubBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
396
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
397
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
398 def test_subseq_start_eq_end(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
399 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
400 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
401 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
402 start = 10
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
403 end = 10
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
404 expSubBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
405 expSubBioseq.header = "seq1 description1 fragment " + str(start) + ".." + str(end)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
406 expSubBioseq.sequence = bioseq.sequence[(start - 1) : end]
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
407 obsBioseq = bioseq.subseq(start, end)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
408 self.assertEquals(expSubBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
409
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
410
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
411 def test_subseq_negative_start(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
412 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
413 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
414 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
415 start = -10
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
416 end = 10
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
417 expSubBioseq = None
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
418 obsBioseq = bioseq.subseq(start, end)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
419 self.assertEquals(expSubBioseq, obsBioseq)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
420
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
421
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
422 def test_getNtFromPosition_1(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
423 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
424 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
425 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
426 expNt = "G"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
427 obsNt = bioseq.getNtFromPosition(1)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
428 self.assertEquals(expNt, obsNt)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
429
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
430
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
431 def test_getNtFromPosition_10(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
432 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
433 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
434 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
435 expNt = "C"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
436 obsNt = bioseq.getNtFromPosition(10)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
437 self.assertEquals(expNt, obsNt)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
438
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
439
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
440 def test_getNtFromPosition_last(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
441 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
442 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
443 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
444 expNt = "G"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
445 obsNt = bioseq.getNtFromPosition(146)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
446 self.assertEquals(expNt, obsNt)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
447
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
448
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
449 def test_getNtFromPosition_position_outside_range_0(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
450 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
451 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
452 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
453 expNt = None
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
454 obsNt = bioseq.getNtFromPosition(0)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
455 self.assertEquals(expNt, obsNt)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
456
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
457
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
458 def test_getNtFromPosition_position_outside_range_negative(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
459 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
460 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
461 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
462 expNt = None
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
463 obsNt = bioseq.getNtFromPosition(-10)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
464 self.assertEquals(expNt, obsNt)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
465
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
466
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
467 def test_getNtFromPosition_position_outside_range_positive(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
468 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
469 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
470 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
471 expNt = None
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
472 obsNt = bioseq.getNtFromPosition(147)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
473 self.assertEquals(expNt, obsNt)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
474
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
475
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
476 def test_view(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
477 obsFileName = "obsdummy_Bioseq_view"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
478 expFileName = "expDummy_Bioseq_View"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
479
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
480 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
481 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
482 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
483
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
484 obsFile = open(obsFileName,"w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
485 expFile = open(expFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
486
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
487 expFile.write ( ">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
488 expFile.write ( "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
489 expFile.write ( "ATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
490 expFile.write ( "ATAGCAGACGCATATTATATTGCGCG\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
491
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
492 stdoutRef = sys.stdout
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
493 sys.stdout = obsFile
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
494 bioseq.view()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
495 obsFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
496 expFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
497 self.assertTrue( FileUtils.are2FilesIdentical( expFileName, obsFileName ) )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
498 sys.stdout = stdoutRef
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
499 os.remove ( obsFileName )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
500 os.remove ( expFileName )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
501
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
502
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
503 def test_view_with_l(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
504 obsFileName = "obsdummy_Bioseq_view"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
505 expFileName = "expDummy_Bioseq_View"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
506 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
507 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
508 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
509 obsFile = open(obsFileName,"w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
510 expFile = open(expFileName, "w")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
511 expFile.write ( ">seq1 description1\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
512 expFile.write ( "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
513 expFile.write ( "ATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTAT\n")
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
514 stdoutRef = sys.stdout
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
515 sys.stdout = obsFile
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
516 bioseq.view(120)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
517 obsFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
518 expFile.close()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
519 self.assertTrue( FileUtils.are2FilesIdentical( expFileName, obsFileName ) )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
520 sys.stdout = stdoutRef
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
521 os.remove ( obsFileName )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
522 os.remove ( expFileName )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
523
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
524
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
525 def test_getLength(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
526 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
527 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
528 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
529 expLength = 146
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
530 obsLength = bioseq.getLength()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
531 self.assertEquals(expLength, obsLength)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
532
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
533
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
534 def test_getLength_empty_seq(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
535 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
536 expLength = 0
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
537 obsLength = bioseq.getLength()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
538 self.assertEquals(expLength, obsLength)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
539
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
540
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
541 def test_getLength_WithoutN(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
542 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
543 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
544 bioseq.sequence = "GCGANCGCTGCTTTATTAAGCGCTAGATGNNNNNNNNNNNNNNNCGACGCTGCATTTATTAAGCGCTAGCGATTATANNNNNNNNNTAGCAGACGCATATTATATTGCGCGATGCGACGCTGCTTTATTANAGCGCTAGCGNNATTATATAGCANGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
545 expLength = 146
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
546 obsLength = bioseq.getLength(False)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
547 self.assertEquals(expLength, obsLength)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
548
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
549
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
550 def test_getLength_WithoutN_empty_seq(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
551 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
552 expLength = 0
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
553 obsLength = bioseq.getLength(False)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
554 self.assertEquals(expLength, obsLength)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
555
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
556
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
557 def test_countNt(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
558 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
559 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
560 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
561 expCount = 3
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
562 obsCount = bioseq.countNt('N')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
563 self.assertEquals(expCount, obsCount)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
564
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
565
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
566 def test_countNt_withCharacterNotExisting(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
567 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
568 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
569 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
570 expCount = 0
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
571 obsCount = bioseq.countNt('W')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
572 self.assertEquals(expCount, obsCount)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
573
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
574
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
575 def test_countAllNt(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
576 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
577 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
578 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
579 dExpCount = {'A': 34, 'C': 31, 'T': 43, 'G': 35, 'N': 3}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
580 dObsCount = bioseq.countAllNt()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
581 self.assertEquals(dExpCount, dObsCount)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
582
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
583
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
584 def test_occ_word_size_1(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
585 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
586 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
587 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
588 dExpOccWord = {'A': 34, 'C': 31, 'T': 43, 'G': 35}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
589 ExpNbWord = 143
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
590 dObsOccWord, ObsNbWord = bioseq.occ_word(1)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
591 self.assertEquals(dExpOccWord, dObsOccWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
592 self.assertEquals(ExpNbWord, ObsNbWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
593
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
594
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
595 def test_occ_word_size_0(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
596 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
597 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
598 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
599 dExpOccWord = {}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
600 ExpNbWord = 0
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
601 dObsOccWord, ObsNbWord = bioseq.occ_word(0)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
602 self.assertEquals(dExpOccWord, dObsOccWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
603 self.assertEquals(ExpNbWord, ObsNbWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
604
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
605
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
606 def test_occ_word_size_n(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
607 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
608 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
609 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
610 dExpOccWord = {'ACC': 0, 'ATG': 2, 'AAG': 3, 'AAA': 0, 'ATC': 0, 'AAC': 0, 'ATA': 8, 'AGG': 0, 'CCT': 0, 'CTC': 0, 'AGC': 8, 'ACA': 0, 'AGA': 2, 'CAT': 3, 'AAT': 0, 'ATT': 9, 'CTG': 3, 'CTA': 3, 'ACT': 0, 'CAC': 0, 'ACG': 2, 'CAA': 0, 'AGT': 0, 'CAG': 2, 'CCG': 0, 'CCC': 0, 'CTT': 3, 'TAT': 13, 'GGT': 0, 'TGT': 0, 'CGA': 3, 'CCA': 0, 'TCT': 0, 'GAT': 3, 'CGG': 0, 'TTT': 3, 'TGC': 7, 'GGG': 0, 'TAG': 5, 'GGA': 0, 'TAA': 3, 'GGC': 0, 'TAC': 0, 'TTC': 0, 'TCG': 0, 'TTA': 10, 'TTG': 2, 'TCC': 0, 'GAA': 0, 'TGG': 0, 'GCA': 5, 'GTA': 0, 'GCC': 0, 'GTC': 0, 'GCG': 12, 'GTG': 0, 'GAG': 0, 'GTT': 0, 'GCT': 9, 'TGA': 0, 'GAC': 2, 'CGT': 0, 'TCA': 0, 'CGC': 10}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
611 ExpNbWord = 135
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
612 dObsOccWord, ObsNbWord = bioseq.occ_word(3)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
613 self.assertEquals(dExpOccWord, dObsOccWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
614 self.assertEquals(ExpNbWord, ObsNbWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
615
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
616
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
617 def test_freq_word_size_1(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
618 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
619 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
620 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
621 dExpFreqWord = {'A': 0.23776223776223776, 'C': 0.21678321678321677, 'T': 0.30069930069930068, 'G': 0.24475524475524477}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
622 dObsFreqWord = bioseq.freq_word(1)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
623 self.assertEquals(dExpFreqWord, dObsFreqWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
624
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
625
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
626 def test_freq_word_size_0(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
627 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
628 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
629 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
630 dExpFreqWord = {}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
631 dObsFreqWord = bioseq.freq_word(0)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
632 self.assertEquals(dExpFreqWord, dObsFreqWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
633
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
634
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
635 def test_freq_word_size_n(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
636 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
637 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
638 bioseq.sequence = "GCGNCGCTGCTTTATTAAGCGCTAGCATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCGATGCGNCGCTGCTTTATTAAGCGCTAGCGATTATATAGCAGACGCATATTATATTGCGCG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
639 dExpFreqWord = {'ACC': 0.0, 'ATG': 0.014814814814814815, 'AAG': 0.022222222222222223, 'AAA': 0.0, 'ATC': 0.0, 'AAC': 0.0, 'ATA': 0.059259259259259262, 'AGG': 0.0, 'CCT': 0.0, 'CTC': 0.0, 'AGC': 0.059259259259259262, 'ACA': 0.0, 'AGA': 0.014814814814814815, 'CAT': 0.022222222222222223, 'AAT': 0.0, 'ATT': 0.066666666666666666, 'CTG': 0.022222222222222223, 'CTA': 0.022222222222222223, 'ACT': 0.0, 'CAC': 0.0, 'ACG': 0.014814814814814815, 'CAA': 0.0, 'AGT': 0.0, 'CAG': 0.014814814814814815, 'CCG': 0.0, 'CCC': 0.0, 'TAT': 0.096296296296296297, 'GGT': 0.0, 'TGT': 0.0, 'CGA': 0.022222222222222223, 'CCA': 0.0, 'TCT': 0.0, 'GAT': 0.022222222222222223, 'CGG': 0.0, 'CTT': 0.022222222222222223, 'TGC': 0.05185185185185185, 'GGG': 0.0, 'TAG': 0.037037037037037035, 'GGA': 0.0, 'TAA': 0.022222222222222223, 'GGC': 0.0, 'TAC': 0.0, 'TTC': 0.0, 'TCG': 0.0, 'TTT': 0.022222222222222223, 'TTG': 0.014814814814814815, 'TCC': 0.0, 'GAA': 0.0, 'TGG': 0.0, 'GCA': 0.037037037037037035, 'GTA': 0.0, 'GCC': 0.0, 'GTC': 0.0, 'TGA': 0.0, 'GCG': 0.088888888888888892, 'GTG': 0.0, 'GAG': 0.0, 'GTT': 0.0, 'GCT': 0.066666666666666666, 'TTA': 0.07407407407407407, 'GAC': 0.014814814814814815, 'CGT': 0.0, 'TCA': 0.0, 'CGC': 0.07407407407407407}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
640 dObsFreqWord = bioseq.freq_word(3)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
641 self.assertEquals(dExpFreqWord, dObsFreqWord)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
642
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
643
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
644 def test_findORF_no_ORF_in_sequence (self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
645 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
646 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
647 bioseq.sequence = "GCGNCGCTGCTTTATT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
648 expORF = {0:[],1:[],2:[]}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
649 obsORF = bioseq.findORF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
650 self.assertEquals (expORF,obsORF)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
651
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
652
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
653 def test_findORF_one_ORF_in_first_phase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
654 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
655 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
656 bioseq.sequence = "TAAGCGNCGCTGCTTTATT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
657 expORF = {0:[0],1:[],2:[]}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
658 obsORF = bioseq.findORF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
659 self.assertEquals (expORF,obsORF)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
660
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
661
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
662 def test_findORF_three_ORF_in_first_phase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
663 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
664 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
665 bioseq.sequence = "TAAGCGTAGNCGTGACTGCTTTATT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
666 expORF = {0:[0,6,12],1:[],2:[]}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
667 obsORF = bioseq.findORF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
668 self.assertEquals (expORF,obsORF)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
669
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
670
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
671 def test_findORF_two_ORF_in_first_phase_one_ORF_in_second_phase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
672 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
673 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
674 bioseq.sequence = "TAAGTAGAGNCGTGACTGCTTTATT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
675 expORF = {0:[0,12],1:[4],2:[]}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
676 obsORF = bioseq.findORF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
677 self.assertEquals (expORF,obsORF)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
678
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
679
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
680 def test_findORF_two_ORF_in_first_phase_three_ORF_in_second_phase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
681 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
682 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
683 bioseq.sequence = "TAAGTAGAGNCGTGACTGATAGTATT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
684 expORF = {0:[0,12],1:[4,16,19],2:[]}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
685 obsORF = bioseq.findORF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
686 self.assertEquals (expORF,obsORF)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
687
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
688
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
689 def test_findORF_one_ORF_in_second_phase_three_ORF_in_third_phase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
690 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
691 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
692 bioseq.sequence = "AATATTAGTGGAGTAGTTGATGATTTT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
693 expORF = {0:[], 1:[13], 2:[5,17,20]}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
694 obsORF = bioseq.findORF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
695 self.assertEquals (expORF,obsORF)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
696
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
697
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
698 def test_findORF_three_ORF_in_second_phase_one_ORF_in_third_phase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
699 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
700 bioseq.header = "seq1 description1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
701 bioseq.sequence = "TTTGAAGTGGAGGAGTTGATGATTTTAAT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
702 expORF = {0:[], 1:[16, 19, 25], 2:[2]}
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
703 obsORF = bioseq.findORF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
704 self.assertEquals (expORF,obsORF)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
705
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
706
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
707 def test_upCase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
708 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
709 bioseq.header = "seq description"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
710 bioseq.sequence = "taattcggcct"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
711 expSeq = "TAATTCGGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
712 bioseq.upCase()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
713 obsSeq = bioseq.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
714 self.assertEquals( expSeq, obsSeq )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
715
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
716
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
717 def test_lowCase(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
718 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
719 bioseq.header = "seq description"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
720 bioseq.sequence = "TAATTCGGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
721 expSeq = "taattcggcct"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
722 bioseq.lowCase()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
723 obsSeq = bioseq.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
724 self.assertEquals( expSeq, obsSeq )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
725
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
726
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
727 def test_getClusterID(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
728 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
729 bioseq.header = "MbQ58Gr2Cl0 chunk1 {Fragment} 74091..74624"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
730 bioseq.sequence = "TAATTCGGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
731 expID = "0"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
732 obsID = bioseq.getClusterID()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
733 self.assertEquals( expID, obsID )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
734
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
735
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
736 def test_getGroupID(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
737 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
738 bioseq.header = "MbQ58Gr2Cl0 chunk1 {Fragment} 74091..74624"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
739 bioseq.sequence = "TAATTCGGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
740 expID = "2"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
741 obsID = bioseq.getGroupID()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
742 self.assertEquals( expID, obsID )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
743
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
744
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
745 def test_getHeaderFullSeq(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
746 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
747 bioseq.header = "MbQ58Gr2Cl0 chunk1 {Fragment} 74091..74624"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
748 bioseq.sequence = "TAATTCGGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
749 expHeader = "chunk1"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
750 obsHeader = bioseq.getHeaderFullSeq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
751 self.assertEquals( expHeader, obsHeader )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
752
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
753
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
754 def test_getFragStrand_plus_strand(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
755 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
756 bioseq.header = "MbQ58Gr2Cl0 chunk1 {Fragment} 74091..74624"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
757 bioseq.sequence = "TAATTCGGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
758 expStrand = '+'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
759 obsStrand = bioseq.getFragStrand()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
760 self.assertEquals(expStrand, obsStrand)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
761
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
762
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
763 def test_getFragStrand_minus_strand(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
764 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
765 bioseq.header = "MbQ58Gr2Cl0 chunk1 {Fragment} 74624..74091"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
766 bioseq.sequence = "TAATTCGGCCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
767 expStrand = '-'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
768 obsStrand = bioseq.getFragStrand()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
769 self.assertEquals(expStrand, obsStrand)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
770
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
771
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
772 def test_getATGCNFromIUPAC_A(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
773 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
774 expNucl = 'A'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
775 obsNucl = bioseq.getATGCNFromIUPAC('A')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
776 self.assertEquals(expNucl, obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
777
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
778
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
779 def test_getATGCNFromIUPAC_T(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
780 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
781 expNucl = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
782 obsNucl = bioseq.getATGCNFromIUPAC('T')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
783 self.assertEquals(expNucl, obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
784
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
785
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
786 def test_getATGCNFromIUPAC_C(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
787 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
788 expNucl = 'C'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
789 obsNucl = bioseq.getATGCNFromIUPAC('C')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
790 self.assertEquals(expNucl, obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
791
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
792
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
793 def test_getATGCNFromIUPAC_G(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
794 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
795 expNucl = 'G'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
796 obsNucl = bioseq.getATGCNFromIUPAC('G')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
797 self.assertEquals(expNucl, obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
798
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
799
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
800 def test_getATGCNFromIUPAC_N(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
801 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
802 expNucl = 'N'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
803 obsNucl = bioseq.getATGCNFromIUPAC('N')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
804 self.assertEquals(expNucl, obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
805
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
806
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
807 def test_getATGCNFromIUPAC_U(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
808 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
809 expNucl = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
810 obsNucl = bioseq.getATGCNFromIUPAC('U')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
811 self.assertEquals(expNucl, obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
812
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
813
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
814 def test_getATGCNFromIUPAC_R(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
815 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
816 expNucl1 = 'A'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
817 expNucl2 = 'G'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
818 obsNucl = bioseq.getATGCNFromIUPAC('R')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
819 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
820
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
821
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
822 def test_getATGCNFromIUPAC_Y(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
823 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
824 expNucl1 = 'C'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
825 expNucl2 = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
826 obsNucl = bioseq.getATGCNFromIUPAC('Y')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
827 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
828
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
829
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
830 def test_getATGCNFromIUPAC_M(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
831 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
832 expNucl1 = 'C'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
833 expNucl2 = 'A'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
834 obsNucl = bioseq.getATGCNFromIUPAC('M')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
835 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
836
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
837
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
838 def test_getATGCNFromIUPAC_K(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
839 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
840 expNucl1 = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
841 expNucl2 = 'G'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
842 obsNucl = bioseq.getATGCNFromIUPAC('K')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
843 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
844
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
845
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
846 def test_getATGCNFromIUPAC_W(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
847 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
848 expNucl1 = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
849 expNucl2 = 'A'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
850 obsNucl = bioseq.getATGCNFromIUPAC('W')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
851 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
852
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
853
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
854 def test_getATGCNFromIUPAC_S(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
855 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
856 expNucl1 = 'C'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
857 expNucl2 = 'G'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
858 obsNucl = bioseq.getATGCNFromIUPAC('S')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
859 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
860
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
861
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
862 def test_getATGCNFromIUPAC_B(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
863 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
864 expNucl1 = 'C'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
865 expNucl2 = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
866 expNucl3 = 'G'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
867 obsNucl = bioseq.getATGCNFromIUPAC('B')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
868 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl or expNucl3 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
869
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
870
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
871 def test_getATGCNFromIUPAC_D(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
872 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
873 expNucl1 = 'A'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
874 expNucl2 = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
875 expNucl3 = 'G'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
876 obsNucl = bioseq.getATGCNFromIUPAC('D')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
877 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl or expNucl3 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
878
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
879
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
880 def test_getATGCNFromIUPAC_H(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
881 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
882 expNucl1 = 'C'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
883 expNucl2 = 'T'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
884 expNucl3 = 'A'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
885 obsNucl = bioseq.getATGCNFromIUPAC('H')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
886 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl or expNucl3 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
887
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
888
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
889 def test_getATGCNFromIUPAC_V(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
890 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
891 expNucl1 = 'C'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
892 expNucl2 = 'A'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
893 expNucl3 = 'G'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
894 obsNucl = bioseq.getATGCNFromIUPAC('V')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
895 self.assertTrue(expNucl1 == obsNucl or expNucl2 == obsNucl or expNucl3 == obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
896
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
897
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
898 def test_getATGCNFromIUPAC_Z(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
899 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
900 expNucl = 'N'
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
901 obsNucl = bioseq.getATGCNFromIUPAC('Z')
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
902 self.assertEquals(expNucl, obsNucl)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
903
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
904
18
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
905 def test_getATGCNFromIUPACandATGCN(self):
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
906 self.assertEquals("G", Bioseq().getATGCNFromIUPACandATGCN('R', 'A'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
907 self.assertEquals("A", Bioseq().getATGCNFromIUPACandATGCN('R', 'G'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
908 self.assertEquals("C", Bioseq().getATGCNFromIUPACandATGCN('Y', 'T'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
909 self.assertEquals("T", Bioseq().getATGCNFromIUPACandATGCN('Y', 'C'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
910 self.assertEquals("C", Bioseq().getATGCNFromIUPACandATGCN('M', 'A'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
911 self.assertEquals("A", Bioseq().getATGCNFromIUPACandATGCN('M', 'C'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
912 self.assertEquals("T", Bioseq().getATGCNFromIUPACandATGCN('K', 'G'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
913 self.assertEquals("G", Bioseq().getATGCNFromIUPACandATGCN('K', 'T'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
914 self.assertEquals("A", Bioseq().getATGCNFromIUPACandATGCN('W', 'T'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
915 self.assertEquals("T", Bioseq().getATGCNFromIUPACandATGCN('W', 'A'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
916 self.assertEquals("G", Bioseq().getATGCNFromIUPACandATGCN('S', 'C'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
917 self.assertEquals("C", Bioseq().getATGCNFromIUPACandATGCN('S', 'G'))
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
918
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
919 def test_getATGCNFromIUPACandATGCN_invalid_combination(self):
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
920 obsMsg = ""
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
921 try:
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
922 Bioseq().getATGCNFromIUPACandATGCN('R', 'T')
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
923 except RepetException as e:
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
924 obsMsg = e.getMessage()
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
925
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
926 self.assertEqual("IUPAC code 'R' and nucleotide 'T' are not compatible", obsMsg)
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
927
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
928 def test_getATGCNFromIUPACandATGCN_invalid_IUPAC(self):
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
929 bioseq = Bioseq()
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
930 obsMsg = ""
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
931 try:
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
932 bioseq.getATGCNFromIUPACandATGCN('B', 'T')
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
933 except RepetException as e:
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
934 obsMsg = e.getMessage()
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
935
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
936 self.assertEqual("Can't retrieve the third nucleotide from IUPAC code 'B' and nucleotide 'T'", obsMsg)
94ab73e8a190 Uploaded
m-zytnicki
parents: 6
diff changeset
937
6
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
938 def test_partialIUPAC(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
939 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
940 bioseq.sequence = "ATGCNRATGCN"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
941 expSequence1 = "ATGCNAATGCN"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
942 expSequence2 = "ATGCNGATGCN"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
943 bioseq.partialIUPAC()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
944 obsSequence = bioseq.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
945 self.assertTrue(expSequence1 == obsSequence or expSequence2 == obsSequence)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
946
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
947
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
948 def test_checkEOF(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
949 bioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
950 bioseq.sequence = "ATGCNRATGCN\rATGCAAT\rTATA\r"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
951 bioseq.checkEOF()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
952 obsSequence = bioseq.sequence
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
953 expSequence = "ATGCNRATGCNATGCAATTATA"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
954
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
955 self.assertEquals(expSequence, obsSequence)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
956
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
957
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
958 def test_getLMapWhithoutGap(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
959 iBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
960 iBioseq.header = "header"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
961 iBioseq.sequence = "ATGC-RA-GCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
962 obsLMap = iBioseq.getLMapWhithoutGap()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
963 expLMap = [Map( "header_subSeq1", "header", 1, 4 ), Map( "header_subSeq2", "header", 6, 7 ), Map( "header_subSeq3", "header", 9, 11 )]
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
964
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
965 self.assertEquals(expLMap, obsLMap)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
966
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
967
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
968 def test_getLMapWhithoutGap_seqStartsWithGap(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
969 iBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
970 iBioseq.header = "header"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
971 iBioseq.sequence = "-TGC-RA-GCT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
972 obsLMap = iBioseq.getLMapWhithoutGap()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
973 expLMap = [Map( "header_subSeq1", "header", 2, 4 ), Map( "header_subSeq2", "header", 6, 7 ), Map( "header_subSeq3", "header", 9, 11 )]
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
974
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
975 self.assertEquals(expLMap, obsLMap)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
976
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
977
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
978 def test_getLMapWhithoutGap_seqEndsWithGap(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
979 iBioseq = Bioseq()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
980 iBioseq.header = "header"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
981 iBioseq.sequence = "ATGC-RA-GC-"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
982 obsLMap = iBioseq.getLMapWhithoutGap()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
983 expLMap = [Map( "header_subSeq1", "header", 1, 4 ), Map( "header_subSeq2", "header", 6, 7 ), Map( "header_subSeq3", "header", 9, 10 )]
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
984
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
985 self.assertEquals(expLMap, obsLMap)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
986
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
987 def test_getGCpercentage_onlyATGC( self ):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
988 iBs = Bioseq( "seq", "TGCAGCT" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
989 exp = 100 * 4 / 7.0
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
990 obs = iBs.getGCpercentage()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
991 self.assertEqual( exp, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
992
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
993 def test_getGCpercentageInSequenceWithoutCountNInLength( self ):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
994 iBs = Bioseq( "seq", "TGCAGCTNNNNN" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
995 exp = 100 * 4 / 7.0
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
996 obs = iBs.getGCpercentageInSequenceWithoutCountNInLength()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
997 self.assertEqual( exp, obs )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
998
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
999 def test_get5PrimeFlank(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1000 bs = Bioseq( "line1", "AACTTTCCAGAA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1001 position = 7
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1002 obsFlank = bs.get5PrimeFlank(position, 3)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1003 expFlank = "TTT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1004 self.assertEquals(expFlank, obsFlank)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1005
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1006 def test_get5PrimeFlank_flank_length_truncated(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1007 bs = Bioseq( "line1", "AACTTTCCAGAA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1008 position = 7
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1009 obsFlank = bs.get5PrimeFlank(position, 15)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1010 expFlank = "AACTTT"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1011 self.assertEquals(expFlank, obsFlank)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1012
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1013 def test_get5PrimeFlank_flank_of_first_base(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1014 bs = Bioseq( "line1", "AACTTTCCAGAA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1015 position = 1
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1016 obsFlank = bs.get5PrimeFlank(position, 15)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1017 expFlank = ""
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1018 self.assertEquals(expFlank, obsFlank)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1019
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1020 def test_get3PrimeFlank(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1021 bs = Bioseq( "line1", "AACTTTCCAGAA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1022 position = 7
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1023 obsFlank = bs.get3PrimeFlank(position, 3)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1024 expFlank = "CAG"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1025 self.assertEquals(expFlank, obsFlank)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1026
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1027 def test_get3PrimeFlank_flank_length_truncated(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1028 bs = Bioseq( "line1", "AACTTTCCAGAA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1029 position = 7
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1030 obsFlank = bs.get3PrimeFlank(position, 15)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1031 expFlank = "CAGAA"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1032 self.assertEquals(expFlank, obsFlank)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1033
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1034 def test_get3PrimeFlank_flank_of_last_base(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1035 bs = Bioseq( "line1", "AACTTTCCAGAA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1036 position = 12
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1037 obsFlank = bs.get3PrimeFlank(position, 15)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1038 expFlank = ""
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1039 self.assertEquals(expFlank, obsFlank)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1040
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1041 def test_get3PrimeFlank_polymLength_different_of_1(self):
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1042 bs = Bioseq( "line1", "AACTTTCCAGAA" )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1043 position = 7
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1044 obsFlank = bs.get3PrimeFlank(position, 3, 2)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1045 expFlank = "AGA"
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1046 self.assertEquals(expFlank, obsFlank)
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1047
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1048 test_suite = unittest.TestSuite()
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1049 test_suite.addTest( unittest.makeSuite( Test_Bioseq ) )
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1050 if __name__ == "__main__":
769e306b7933 Change the repository level.
yufei-luo
parents:
diff changeset
1051 unittest.TextTestRunner(verbosity=2).run( test_suite )