import os
-import unittest
-from StringIO import StringIO
+from unittest import TestCase
+from io import StringIO
-from simulate_runfolder import TESTDATA_DIR
+from .simulate_runfolder import TESTDATA_DIR
from htsworkflow.pipelines.runfolder import load_pipeline_run_xml
from htsworkflow.pipelines.eland import SampleKey
-class testLoadRunXML(unittest.TestCase):
+
+class testLoadRunXML(TestCase):
def _check_run_xml(self, run_xml_name, results, eland_results=8):
run_xml_path = os.path.join(TESTDATA_DIR, run_xml_name)
run = load_pipeline_run_xml(run_xml_path)
self.failUnlessEqual(run.runfolder_name, runfolder_name)
self.failUnlessEqual(run.gerald.runfolder_name, runfolder_name)
- for (end, lane), lane_results in results['lane_results'].items():
- for name, test_value in lane_results.items():
+ for (end, lane), lane_results in list(results['lane_results'].items()):
+ for name, test_value in list(lane_results.items()):
xml_value = getattr(run.gerald.summary[end][lane], name)
self.failUnlessEqual(xml_value, test_value,
}
self._check_run_xml(run_xml_name, results, eland_results=8)
+
def suite():
- return unittest.makeSuite(testLoadRunXML,'test')
+ from unittest import TestSuite, defaultTestLoader
+ suite = TestSuite()
+ suite.addTests(defaultTestLoader.loadTestsFromTestCase(testLoadRunXML))
+ return suite
+
if __name__ == "__main__":
- unittest.main(defaultTest="suite")
+ from unittest import main
+ main(defaultTest="suite")