Convert to unittest2

[htsworkflow.git] / test / test_srf2fastq.py

diff --git a/test/test_srf2fastq.py b/test/test_srf2fastq.py
index 74e031ca7324174362de8695fb99ba4c50d830c7..9185fffde9d516d8651a5ade40723645ab82fa26 100644 (file)
--- a/test/test_srf2fastq.py
+++ b/test/test_srf2fastq.py
@@ -1,7 +1,7 @@
 import os
 from StringIO import StringIO
 import sys
-import unittest
+from unittest2 import TestCase
 
 _module_path, _module_name = os.path.split(__file__)
 sys.path.append(os.path.join(_module_path, '..', 'scripts'))
@@ -10,7 +10,7 @@ from htsworkflow.pipelines.test.simulate_runfolder import TESTDATA_DIR
 
 from htsworkflow.pipelines import srf2fastq
 
-class testSrf2Fastq(unittest.TestCase):
+class testSrf2Fastq(TestCase):
     def test_split_good(self):
         source = StringIO("""@header
 AGCTTTTT
@@ -135,8 +135,14 @@ IIIIB+++
         self.failUnlessEqual(is_cnf1(cnf4_path), False)
         self.failUnlessEqual(is_cnf1(cnf1_path), True)
 
+
 def suite():
-    return unittest.makeSuite(testSrf2Fastq,'test')
+    from unittest2 import TestSuite, defaultTestLoader
+    suite = TestSuite()
+    suite.addTests(defaultTestLoader.loadTestsFromTestCase(testSrf2Fastq))
+    return suite
+
 
 if __name__ == "__main__":
-    unittest.main(defaultTest="suite")
+    from unittest2 import main
+    main(defaultTest="suite")