Convert to unittest2

[htsworkflow.git] / htsworkflow / pipelines / test / test_genomemap.py

diff --git a/htsworkflow/pipelines/test/test_genomemap.py b/htsworkflow/pipelines/test/test_genomemap.py
index d0c1e660e5a786c50794d278f164678026a5cb74..aa2f618f9324faca633dae27a0053ad76c14b54b 100644 (file)
--- a/htsworkflow/pipelines/test/test_genomemap.py
+++ b/htsworkflow/pipelines/test/test_genomemap.py
@@ -5,7 +5,7 @@ import os
 from StringIO import StringIO
 import shutil
 import tempfile
-import unittest
+from unittest2 import TestCase
 
 from htsworkflow.pipelines.runfolder import ElementTree
 from htsworkflow.pipelines import genomemap
@@ -15,7 +15,7 @@ MINI_GENOME_XML = '''<sequenceSizes>
         <chromosome fileName="chr1.fa" contigName="chr1" totalBases="197195432"/>
 </sequenceSizes>
 '''
-class TestGenomeMap(unittest.TestCase):
+class TestGenomeMap(TestCase):
     def test_genomesizes_xml(self):
         xml = ElementTree.fromstring(MINI_GENOME_XML)
         g = genomemap.GenomeMap()
@@ -76,5 +76,13 @@ class TestGenomeMap(unittest.TestCase):
         self.assertEqual(g['chr1.fa'], '{0}/chr1.fa'.format(tempgenome))
 
 
+def suite():
+    from unittest2 import TestSuite, defaultTestLoader
+    suite = TestSuite()
+    suite.addTests(defaultTestLoader.loadTestsFromTestCase(TestGenomeMap))
+    return suite
+
+
 if __name__ == "__main__":
-    unittest.main()
+    from unittest2 import main
+    main(defaultTest="suite")