-import unittest
+from unittest import TestCase
-from StringIO import StringIO
+from io import StringIO
from htsworkflow.pipelines import genome_mapper
-class testGenomeMapper(unittest.TestCase):
+class testGenomeMapper(TestCase):
def test_construct_mapper(self):
genomes = {
'Arabidopsis thaliana': {'v01212004': '/arabidopsis'},
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():
- return unittest.makeSuite(testGenomeMapper,'test')
+ from unittest import TestSuite, defaultTestLoader
+ suite = TestSuite()
+ suite.addTests(defaultTestLoader.loadTestsFromTestCase(testGenomeMapper))
+ return suite
+
if __name__ == "__main__":
- unittest.main(defaultTest="suite")
+ from unittest import main
+ main(defaultTest="suite")