Initial port to python3

[htsworkflow.git] / htsworkflow / pipelines / test / test_genome_mapper.py

diff --git a/htsworkflow/pipelines/test/test_genome_mapper.py b/htsworkflow/pipelines/test/test_genome_mapper.py
index 1101305effd80ede9a09d4db99abe3566b31c8ca..e7561adf567b124ea6b06702934335d60337752f 100644 (file)
--- a/htsworkflow/pipelines/test/test_genome_mapper.py
+++ b/htsworkflow/pipelines/test/test_genome_mapper.py
@@ -1,6 +1,6 @@
-from unittest2 import TestCase
+from unittest import TestCase
 
-from StringIO import StringIO
+from io import StringIO
 from htsworkflow.pipelines import genome_mapper
 
 class testGenomeMapper(TestCase):
@@ -21,18 +21,18 @@ class testGenomeMapper(TestCase):
         self.failUnlessEqual("%(Mus musculus|mm8)s" % (genome_map), "/mm8")
         self.failUnlessEqual("%(Mus musculus|mm10)s" % (genome_map), "/mm10")
         
-        self.failUnlessEqual(len(genome_map.keys()), 6)
-        self.failUnlessEqual(len(genome_map.values()), 6)
-        self.failUnlessEqual(len(genome_map.items()), 6)
+        self.failUnlessEqual(len(list(genome_map.keys())), 6)
+        self.failUnlessEqual(len(list(genome_map.values())), 6)
+        self.failUnlessEqual(len(list(genome_map.items())), 6)
         
         
 def suite():
-    from unittest2 import TestSuite, defaultTestLoader
+    from unittest import TestSuite, defaultTestLoader
     suite = TestSuite()
     suite.addTests(defaultTestLoader.loadTestsFromTestCase(testGenomeMapper))
     return suite
 
 
 if __name__ == "__main__":
-    from unittest2 import main
+    from unittest import main
     main(defaultTest="suite")