X-Git-Url: http://woldlab.caltech.edu/gitweb/?a=blobdiff_plain;f=htsworkflow%2Fpipelines%2Ftest%2Ftest_genome_mapper.py;h=e7561adf567b124ea6b06702934335d60337752f;hb=refs%2Fheads%2Fpython3-django1.5;hp=1101305effd80ede9a09d4db99abe3566b31c8ca;hpb=4262586d10cc0cc227390873b301b55244204c11;p=htsworkflow.git

diff --git a/htsworkflow/pipelines/test/test_genome_mapper.py b/htsworkflow/pipelines/test/test_genome_mapper.py
index 1101305..e7561ad 100644
--- 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")