Fix test_shlex: Use io.StringIO.

doerwalter · doerwalter · commit 2c849f2f2071 · 2007-06-12T17:43:43.000Z
diff --git 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 @@ def splitTest(self, data, comments):
 
     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)
