def format_sql(s, debug=False): tokens = list(tokenize(s)) if debug: print_non_data('Tokens: %s' % tokens) parsed = list(parse(tokens)) if debug: print_non_data('Statements: %s' % parsed) styled = style(parsed) if debug: print_non_data('Output: %s' % styled) return styled
def assert_style(statements1, styled2): assert style(statements1) == styled2