2 from pytest import raises
3 from sqlconvert.print_tokens import tlist2str
4 from sqlconvert.process_tokens import StatementGrouper
8 grouper = StatementGrouper()
9 grouper.process_line("select * from `T`")
10 assert not grouper.statements
11 assert len(grouper.statements) == 0
12 raises(ValueError, grouper.close)
15 def test_statements():
16 grouper = StatementGrouper()
17 grouper.process_line("select * from T;")
18 assert grouper.statements
19 assert len(grouper.statements) == 1
20 for statement in grouper.get_statements():
21 query = tlist2str(statement)
22 assert query == 'SELECT * FROM T;'
23 assert len(grouper.statements) == 0
24 assert grouper.close() is None