Update test to be more robust to changing dictionary order.

[htsworkflow.git] / htsworkflow / pipelines / test / test_runfolder_ipar130.py

diff --git a/htsworkflow/pipelines/test/test_runfolder_ipar130.py b/htsworkflow/pipelines/test/test_runfolder_ipar130.py
index c23ed9730808416484500e37c866a29e984e9ce0..4891015d3ff886acd3f44d7daf465d37daf3b3e9 100644 (file)
--- a/htsworkflow/pipelines/test/test_runfolder_ipar130.py
+++ b/htsworkflow/pipelines/test/test_runfolder_ipar130.py
@@ -1,10 +1,11 @@
 #!/usr/bin/env python
+from __future__ import absolute_import
 
 from datetime import datetime, date
 import os
 import tempfile
 import shutil
-from unittest2 import TestCase
+from unittest import TestCase
 
 from htsworkflow.pipelines import eland
 from htsworkflow.pipelines import ipar
@@ -13,7 +14,7 @@ from htsworkflow.pipelines import gerald
 from htsworkflow.pipelines import runfolder
 from htsworkflow.pipelines import ElementTree
 
-from htsworkflow.pipelines.test.simulate_runfolder import *
+from .simulate_runfolder import *
 
 
 def make_runfolder(obj=None):
@@ -122,7 +123,7 @@ class RunfolderTests(TestCase):
         self.failUnlessEqual(b.user,    'diane')
         self.failUnlessEqual(len(b.phasing), 8)
         self.failUnlessAlmostEqual(b.phasing[8].phasing, 0.0099)
-        self.failUnlessEqual(b.crosstalk.base.keys(), ['A','C','T','G'])
+        self.assertEqual(set(b.crosstalk.base), set(['A','C','T','G']))
         check_crosstalk(b.crosstalk)
 
         xml = b.get_elements()
@@ -337,12 +338,12 @@ class RunfolderTests(TestCase):
 
 
 def suite():
-    from unittest2 import TestSuite, defaultTestLoader
+    from unittest import TestSuite, defaultTestLoader
     suite = TestSuite()
     suite.addTests(defaultTestLoader.loadTestsFromTestCase(RunfolderTests))
     return suite
 
 
 if __name__ == "__main__":
-    from unittest2 import main
+    from unittest import main
     main(defaultTest="suite")