-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):
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")