Convert to unittest2

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

diff --git a/htsworkflow/pipelines/test/test_sequences.py b/htsworkflow/pipelines/test/test_sequences.py
index accd43ecf8786b4d25a96683bf861940006cf44f..522a9aafdf1ff7f3364a99f70cbb530a622eeae8 100644 (file)
--- a/htsworkflow/pipelines/test/test_sequences.py
+++ b/htsworkflow/pipelines/test/test_sequences.py
@@ -2,7 +2,7 @@
 import os
 import shutil
 import tempfile
-import unittest
+from unittest2 import TestCase
 
 import RDF
 
@@ -10,7 +10,7 @@ from htsworkflow.pipelines import sequences
 from htsworkflow.util.rdfhelp import get_model, load_string_into_model, \
      rdfNS, libraryOntology, dump_model, fromTypedNode
 
-class SequenceFileTests(unittest.TestCase):
+class SequenceFileTests(TestCase):
     """
     Make sure the sequence archive class works
     """
@@ -612,7 +612,12 @@ class SimulateSimpleTree(SimulateTree):
 
 
 def suite():
-    return unittest.makeSuite(SequenceFileTests,'test')
+    from unittest2 import TestSuite, defaultTestLoader
+    suite = TestSuite()
+    suite.addTests(defaultTestLoader.loadTestsFromTestCase(SequenceFileTests))
+    return suite
+
 
 if __name__ == "__main__":
-    unittest.main(defaultTest="suite")
+    from unittest2 import main
+    main(defaultTest="suite")