Fix test_shlex: Use io.StringIO.

1 files changed, 2 insertions, 7 deletions

diff --git a/Lib/test/test_shlex.py b/Lib/test/test_shlex.py
index 5c6572a..b18155e 100644
--- a/Lib/test/test_shlex.py
+++ b/Lib/test/test_shlex.py
@@ -1,15 +1,10 @@
 # -*- coding: iso-8859-1 -*-
 import unittest
-import os, sys
+import os, sys, io
 import shlex
 
 from test import test_support
 
-try:
-    from cStringIO import StringIO
-except ImportError:
-    from StringIO import StringIO
-
 
 # The original test data set was from shellwords, by Hartmut Goebel.
 
@@ -160,7 +155,7 @@ class ShlexTest(unittest.TestCase):
 
     def oldSplit(self, s):
         ret = []
-        lex = shlex.shlex(StringIO(s))
+        lex = shlex.shlex(io.StringIO(s))
         tok = lex.get_token()
         while tok:
             ret.append(tok)