X-Git-Url: http://woldlab.caltech.edu/gitweb/?a=blobdiff_plain;f=htsworkflow%2Fpipelines%2Ftest%2Ftest_sequences.py;h=522a9aafdf1ff7f3364a99f70cbb530a622eeae8;hb=6ea8652c57937ff48c18fc3990ae7f2ee0875ab9;hp=accd43ecf8786b4d25a96683bf861940006cf44f;hpb=3c1c31b97282ad83b38964f263398185aceb6472;p=htsworkflow.git

diff --git a/htsworkflow/pipelines/test/test_sequences.py b/htsworkflow/pipelines/test/test_sequences.py
index accd43e..522a9aa 100644
--- 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")