#!/usr/bin/python3

import pglast

query = 'select foo . bar from moo where (1=2)'
prettified_query = pglast.prettify(query)
expected_query = """SELECT foo.bar
FROM moo
WHERE 1 = 2"""

print(query)
print(prettified_query)
assert prettified_query == expected_query
