Change unittest2 back into unittest.

[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..50d2c64a638f5f8b84bff93f0418090c3fc7094d 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 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")