Make most tests run unmodified under Python 3.4.

1 files changed, 4 insertions, 3 deletions

diff --git a/test/unittest_lint.py b/test/unittest_lint.py
index c0064062e..c96dcf70e 100644
--- a/test/unittest_lint.py
+++ b/test/unittest_lint.py
@@ -13,7 +13,6 @@
 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 
 from contextlib import contextmanager
-import cStringIO
 import sys
 import os
 import tempfile
@@ -22,6 +21,8 @@ from os import getcwd, chdir
 from os.path import join, basename, dirname, isdir, abspath, sep
 import unittest
 
+import six
+
 from logilab.common.compat import reload
 
 from pylint import config, lint
@@ -468,7 +469,7 @@ class PyLinterTC(unittest.TestCase):
             self.linter.reporter.messages)
 
     def test_html_reporter_missing_files(self):
-        output = cStringIO.StringIO()
+        output = six.StringIO()
         self.linter.set_reporter(html.HTMLReporter(output))
         self.linter.set_option('output-format', 'html')
         self.linter.check('troppoptop.py')
@@ -661,7 +662,7 @@ class MessagesStoreTC(unittest.TestCase):
             msg, checkerref=False)
 
     def test_list_messages(self):
-        sys.stdout = cStringIO.StringIO()
+        sys.stdout = six.StringIO()
         try:
             self.store.list_messages()
             output = sys.stdout.getvalue()