diff options
Diffstat (limited to 'xtesting/core/unit.py')
-rw-r--r-- | xtesting/core/unit.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/xtesting/core/unit.py b/xtesting/core/unit.py index 877cd073..9d549a2d 100644 --- a/xtesting/core/unit.py +++ b/xtesting/core/unit.py @@ -10,7 +10,7 @@ """Define the parent class to run unittest.TestSuite as TestCase.""" from __future__ import division - +from io import BytesIO import logging import os import shutil @@ -19,7 +19,6 @@ import time import unittest from subunit.run import SubunitTestRunner -import six from xtesting.core import testcase @@ -112,7 +111,7 @@ class Suite(testcase.TestCase): self.start_time = time.time() if not os.path.isdir(self.res_dir): os.makedirs(self.res_dir) - stream = six.BytesIO() + stream = BytesIO() result = SubunitTestRunner( stream=stream, verbosity=2).run(self.suite).decorated self.generate_stats(stream) |