X-Git-Url: http://woldlab.caltech.edu/gitweb/?p=htsworkflow.git;a=blobdiff_plain;f=test%2Ftest_copier.py;fp=test%2Ftest_copier.py;h=e30f60452eb833ff0787822067d8e348b28e22f0;hp=3e26cc679605f36d45d5f24016a99f959df63b4e;hb=53ebb47e86e81ce349d4ef4e16aef94be93486ea;hpb=4262586d10cc0cc227390873b301b55244204c11

diff --git a/test/test_copier.py b/test/test_copier.py
index 3e26cc6..e30f604 100644
--- a/test/test_copier.py
+++ b/test/test_copier.py
@@ -1,6 +1,6 @@
-from unittest2 import TestCase
+from unittest import TestCase
 
-from StringIO import StringIO
+from io import StringIO
 from htsworkflow.automation import copier
 from htsworkflow.automation.solexa import is_runfolder
 
@@ -76,12 +76,12 @@ notify_users: user3@example.fake
 
 
 def suite():
-    from unittest2 import TestSuite, defaultTestLoader
+    from unittest import TestSuite, defaultTestLoader
     suite = TestSuite()
     suite.addTests(defaultTestLoader.loadTestsFromTestCase(testCopier))
     return suite
 
 
 if __name__ == "__main__":
-    from unittest2 import main
+    from unittest import main
     main(defaultTest="suite")