Initial port to python3

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

diff --git a/htsworkflow/util/test/test_makebed.py b/htsworkflow/util/test/test_makebed.py
index b701e08a5fec656f0705449e3b9a4352e1269d2a..2c9ea4f66bdff80ac9ca6c87d2271331acb3866a 100644 (file)
--- a/htsworkflow/util/test/test_makebed.py
+++ b/htsworkflow/util/test/test_makebed.py
@@ -1,6 +1,6 @@
 import os
-from StringIO import StringIO
-from unittest2 import TestCase
+from io import StringIO
+from unittest import TestCase
 
 from htsworkflow.util import makebed
 
@@ -43,17 +43,16 @@ class testMakeBed(TestCase):
       out = StringIO()
 
       out = list(makebed.make_bed_from_multi_eland_generator(instream, 'name', 'desc', 'mm9_chr', 1))
-      print out
       self.failUnlessEqual(len(out), 1)
 
 
 def suite():
-    from unittest2 import TestSuite, defaultTestLoader
+    from unittest import TestSuite, defaultTestLoader
     suite = TestSuite()
     suite.addTests(defaultTestLoader.loadTestsFromTestCase(testMakeBed))
     return suite
 
 
 if __name__ == "__main__":
-    from unittest2 import main
+    from unittest import main
     main(defaultTest="suite")