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