8 from erange import bedtoregion
11 class TestBedToRegion(unittest.TestCase):
13 testBedFile = "erangeTestBedFile"
14 testRegionFile = "erangeTestRegionFile"
18 bedfile = open(self.testBedFile, "w")
19 bedfile.write("tab\tdelimited\tfields\n")
20 bedfile.write("space delimited fields\n")
21 bedfile.write("track\n")
22 bedfile.write("line after track will not be processed\n")
28 os.remove(self.testBedFile)
30 print "bed file does not exist"
33 os.remove(self.testRegionFile)
35 print "region file does not exist"
38 def testBedToRegion(self):
39 bedtoregion.bedToRegion("regionLabel", self.testBedFile, self.testRegionFile)
40 resultFile = open(self.testRegionFile)
41 regionList = resultFile.readlines()
42 self.assertEquals(2, len(regionList))
43 self.assertEquals("regionLabel1\ttab\tdelimited\tfields\n", regionList[0])
44 self.assertEquals("regionLabel2\tspace\tdelimited\tfields\n", regionList[1])
48 argv = ["bedtoregion"]
49 self.assertRaises(SystemExit, bedtoregion.main, argv)
50 argv = ["bedtoregion", "regionLabel", self.testBedFile, self.testRegionFile]
51 bedtoregion.main(argv)
52 resultFile = open(self.testRegionFile)
53 self.assertEquals(2, len(resultFile.readlines()))
57 suite = unittest.TestSuite()
58 suite.addTest(unittest.makeSuite(TestBedToRegion))
63 if __name__ == "__main__":
64 #import sys;sys.argv = ['', 'Test.testName']