31
|
1 from commons.core.utils.FileUtils import FileUtils
|
|
2 from commons.core.parsing.WigParser import WigParser
|
|
3 import unittest
|
|
4
|
|
5 class Test_WigParser(unittest.TestCase):
|
|
6
|
|
7 def tearDown(self):
|
|
8 FileUtils.removeFilesByPattern("data/.chr*.index")
|
|
9
|
|
10 def test_GetRange1(self):
|
|
11 self.parser = WigParser("data/test.wig")
|
|
12 outputRange = [0.0, 1.1, 1.2, 0.0, 1.4, 1.5, 0.0, 1.7, 0.0, 1.9, 0.0]
|
|
13 self.assertEqual(self.parser.getRange("chr1", 10, 20), outputRange)
|
|
14 outputRange = [0.0, 9.5, 9.6, 0.0]
|
|
15 self.assertEqual(self.parser.getRange("chrX", 4, 7), outputRange)
|
|
16
|
|
17 def test_GetRange2(self):
|
|
18 self.parser = WigParser("data/test1.wig")
|
|
19 outputRange = [0.0, 1.1, 1.2, 0.0, 1.4, 1.5, 0.0, 1.7, 0.0, 1.9, 0.0]
|
|
20 self.assertEqual(self.parser.getRange("chr2", 10, 20), outputRange)
|
|
21
|
|
22 def test_GetRange3(self):
|
|
23 self.parser = WigParser("data/test2.wig")
|
|
24 outputRange = [1.4, 1.5]
|
|
25 self.assertEqual(self.parser.getRange("chr3", 14, 15), outputRange)
|
|
26
|
|
27 if __name__ == '__main__':
|
|
28 unittest.main()
|