X-Git-Url: http://woldlab.caltech.edu/gitweb/?a=blobdiff_plain;f=htsworkflow%2Fpipelines%2Ftest%2Ftest_genomemap.py;h=7640910afd0ae6078a658237049ac89d75f62957;hb=d078bd2653d1d702d54a0ff12f91ef2bd85d8d0e;hp=aa2f618f9324faca633dae27a0053ad76c14b54b;hpb=6ea8652c57937ff48c18fc3990ae7f2ee0875ab9;p=htsworkflow.git

diff --git a/htsworkflow/pipelines/test/test_genomemap.py b/htsworkflow/pipelines/test/test_genomemap.py
index aa2f618..7640910 100644
--- a/htsworkflow/pipelines/test/test_genomemap.py
+++ b/htsworkflow/pipelines/test/test_genomemap.py
@@ -5,9 +5,9 @@ import os
 from StringIO import StringIO
 import shutil
 import tempfile
-from unittest2 import TestCase
+from unittest import TestCase
 
-from htsworkflow.pipelines.runfolder import ElementTree
+from htsworkflow.pipelines import ElementTree
 from htsworkflow.pipelines import genomemap
 
 MINI_GENOME_XML = '''<sequenceSizes>
@@ -77,12 +77,12 @@ class TestGenomeMap(TestCase):
 
 
 def suite():
-    from unittest2 import TestSuite, defaultTestLoader
+    from unittest import TestSuite, defaultTestLoader
     suite = TestSuite()
     suite.addTests(defaultTestLoader.loadTestsFromTestCase(TestGenomeMap))
     return suite
 
 
 if __name__ == "__main__":
-    from unittest2 import main
+    from unittest import main
     main(defaultTest="suite")