annotate commons/core/seq/test/Test_Bioseq.py @ 6:769e306b7933

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