X-Git-Url: http://woldlab.caltech.edu/gitweb/?a=blobdiff_plain;f=htsworkflow%2Fautomation%2Ftest%2Ftest_runner.py;h=a60c458401b2f0e8129c72d1a5864b0c11f045a5;hb=afe61390da5434277effe31f7c73ae41f55f0e00;hp=1457f9effbb44e764f119184a572cd48d232e940;hpb=c79ee97e6c1345658e2a48d79081258249268936;p=htsworkflow.git

diff --git a/htsworkflow/automation/test/test_runner.py b/htsworkflow/automation/test/test_runner.py
index 1457f9e..a60c458 100644
--- a/htsworkflow/automation/test/test_runner.py
+++ b/htsworkflow/automation/test/test_runner.py
@@ -1,5 +1,4 @@
-import unittest
-
+from unittest2 import TestCase
 
 import os
 from htsworkflow.automation.solexa import is_runfolder
@@ -17,13 +16,14 @@ def extract_runfolder_path(watchdir, event):
       return runfolder_path
   return None
 
+
 class Event(object):
   def __init__(self, path=None, name=None):
     self.path = path
     self.name = name
 
-class testRunner(unittest.TestCase):
 
+class testRunner(TestCase):
     def test_extract_runfolder(self):
         watchdir = os.path.join('root', 'server', 'mount')
         runfolder = os.path.join(watchdir, '080909_HWI-EAS229_0052_1234ABCD')
@@ -38,9 +38,15 @@ class testRunner(unittest.TestCase):
 
         event = Event( path=other)
         self.failUnlessEqual(extract_runfolder_path(watchdir, event), None )
-        
+
+
 def suite():
-    return unittest.makeSuite(testRunner,'test')
+    from unittest2 import TestSuite, defaultTestLoader
+    suite = TestSuite()
+    suite.addTests(defaultTestLoader.loadTestsFromTestCase(testRunner))
+    return suite
+
 
 if __name__ == "__main__":
-    unittest.main(defaultTest="suite")
+    from unittest2 import main
+    main(defaultTest="suite")