diff --git a/os_testr/tests/test_return_codes.py b/os_testr/tests/test_return_codes.py index 591e4dd..082a74a 100644 --- a/os_testr/tests/test_return_codes.py +++ b/os_testr/tests/test_return_codes.py @@ -14,13 +14,13 @@ import os import shutil -import StringIO import subprocess import tempfile import testtools from os_testr.tests import base +from six import StringIO DEVNULL = open(os.devnull, 'wb') @@ -47,8 +47,8 @@ class TestReturnCodes(base.TestCase): shutil.copy('os_testr/tests/files/setup.cfg', self.setup_cfg_file) shutil.copy('os_testr/tests/files/__init__.py', self.init_file) - self.stdout = StringIO.StringIO() - self.stderr = StringIO.StringIO() + self.stdout = StringIO() + self.stderr = StringIO() # Change directory, run wrapper and check result self.addCleanup(os.chdir, os.path.abspath(os.curdir)) os.chdir(self.directory) diff --git a/test-requirements.txt b/test-requirements.txt index d0ca195..608d49e 100644 --- a/test-requirements.txt +++ b/test-requirements.txt @@ -11,3 +11,4 @@ oslosphinx>=2.2.0 # Apache-2.0 oslotest>=1.2.0 # Apache-2.0 testscenarios>=0.4 ddt>=0.4.0 +six>=1.9.0