- g = grouper.get_statements()
- statement = next(g)
- requote_names(statement)
- query = tlist2str(parsed)
- self.assertEqual(query, 'SELECT * FROM "T";')
- self.assertRaises(StopIteration, next, g)
+ for statement in grouper.get_statements():
+ requote_names(statement)
+ query = tlist2str(statement)
+ self.assertEqual(query, 'SELECT * FROM "T";')