tests/test_print_tokens.py

   1 # -*- coding: utf-8 -*-
   2
   3 try:
   4     from cStringIO import StringIO
   5 except ImportError:
   6     try:
   7         from StringIO import StringIO
   8     except ImportError:
   9         from io import StringIO
  10
  11 from sqlparse import parse
  12 from sqlconvert.print_tokens import print_tokens
  13
  14
  15 def test_print_tokens():
  16     sio = StringIO()
  17     parsed = parse("select * from T")[0]
  18     print_tokens(parsed, outfile=sio)
  19     assert sio.getvalue() == u"SELECT * FROM T"