Convert to unittest2

[htsworkflow.git] / htsworkflow / util / test / test_api.py

diff --git a/htsworkflow/util/test/test_api.py b/htsworkflow/util/test/test_api.py
index 4cbe3831405916266493a3c62cd7f5af67f61749..dadd6afd198fa841c19f81a56f3f6249becbdd47 100644 (file)
--- a/htsworkflow/util/test/test_api.py
+++ b/htsworkflow/util/test/test_api.py
@@ -1,10 +1,10 @@
 import copy
 import os
-import unittest
+from unittest2 import TestCase
 
 from htsworkflow.util import api
 
-class testApi(unittest.TestCase):
+class testApi(TestCase):
     def test_make_key(self):
         k1 = api.make_django_secret_key()
         k2 = api.make_django_secret_key()
@@ -14,7 +14,12 @@ class testApi(unittest.TestCase):
         self.failUnless(k1 != k2)
 
 def suite():
-    return unittest.makeSuite(testApi, 'test')
+    from unittest2 import TestSuite, defaultTestLoader
+    suite = TestSuite()
+    suite.addTests(defaultTestLoader.loadTestsFromTestCase(TestApi))
+    return suite
+
 
 if __name__ == "__main__":
-    unittest.main(defaultTest='suite')
+    from unittest2 import main
+    main(defaultTest="suite")