Initial port to python3

[htsworkflow.git] / htsworkflow / submission / test / test_ucsc.py

diff --git a/htsworkflow/submission/test/test_ucsc.py b/htsworkflow/submission/test/test_ucsc.py
index a003594310069b5108073890abe1d99ec409f2bf..2cc237c94b54cf0e918521946d4f9e15e7982c40 100644 (file)
--- a/htsworkflow/submission/test/test_ucsc.py
+++ b/htsworkflow/submission/test/test_ucsc.py
@@ -1,5 +1,5 @@
-import unittest
-from StringIO import StringIO
+from unittest import TestCase, TestSuite, defaultTestLoader
+from io import StringIO
 
 from htsworkflow.submission import ucsc
 
@@ -9,21 +9,22 @@ wgEncodeCaltechRnaSeqGm12878R2x75Il200FastqRd2Rep1.fastq.tgz  project=wgEncode; g
 """
 
 
-class TestUCSCInfo(unittest.TestCase):
+class TestUCSCInfo(TestCase):
     def test_parse_encodedcc_file(self):
         stream = StringIO(ENCODE_FILES)
-        file_index = ucsc.parse_ucsc_file_index(stream)
+        file_index = ucsc.parse_ucsc_file_index(stream, 'http://example.com/files')
         self.assertEquals(len(file_index), 2)
 
-        for attributes in file_index.values():
+        for attributes in list(file_index.values()):
             self.failUnless('subId' in attributes)
             self.failUnless('project' in attributes)
             self.assertEquals(attributes['project'], 'wgEncode')
 
 def suite():
-    suite = unittest.makeSuite(TestUCSCInfo, 'test')
+    suite = TestSuite()
+    suite.addTests(defaultTestLoader.loadTestsFromTestCase(TestUCSCInfo))
     return suite
 
 if __name__ == "__main__":
-    unittest.main(defaultTest='suite')
-
+    from unittest import main
+    main(defaultTest='suite')