Use compat module for single Python 2/3 code base.

This change includes minor fixes and code cleanup too.

1 files changed, 1 insertions, 4 deletions

diff --git a/tests/test_tokenize.py b/tests/test_tokenize.py
index 0b23fa8..ceaf24e 100644
--- a/tests/test_tokenize.py
+++ b/tests/test_tokenize.py
@@ -9,6 +9,7 @@ import pytest
 import sqlparse
 from sqlparse import lexer
 from sqlparse import sql
+from sqlparse.compat import StringIO
 from sqlparse.tokens import *
 
 
@@ -133,8 +134,6 @@ class TestTokenList(unittest.TestCase):
 
 class TestStream(unittest.TestCase):
     def test_simple(self):
-        from cStringIO import StringIO
-
         stream = StringIO("SELECT 1; SELECT 2;")
         lex = lexer.Lexer()
 
@@ -152,8 +151,6 @@ class TestStream(unittest.TestCase):
         self.assertEqual(len(tokens), 9)
 
     def test_error(self):
-        from cStringIO import StringIO
-
         stream = StringIO("FOOBAR{")
 
         lex = lexer.Lexer()