diff options
Diffstat (limited to 'tests/test_parser.py')
-rw-r--r-- | tests/test_parser.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tests/test_parser.py b/tests/test_parser.py index 2cefc07..d6849c3 100644 --- a/tests/test_parser.py +++ b/tests/test_parser.py @@ -106,6 +106,7 @@ class TestParser(unittest.TestCase): expr = parse_one("SELECT foo IN UNNEST(bla) AS bar") self.assertIsInstance(expr.selects[0], exp.Alias) self.assertEqual(expr.selects[0].output_name, "bar") + self.assertIsNotNone(parse_one("select unnest(x)").find(exp.Unnest)) def test_unary_plus(self): self.assertEqual(parse_one("+15"), exp.Literal.number(15)) @@ -880,10 +881,12 @@ class TestParser(unittest.TestCase): self.assertIsInstance(parse_one("a IS DISTINCT FROM b OR c IS DISTINCT FROM d"), exp.Or) def test_trailing_comments(self): - expressions = parse(""" + expressions = parse( + """ select * from x; -- my comment - """) + """ + ) self.assertEqual( ";\n".join(e.sql() for e in expressions), "SELECT * FROM x;\n/* my comment */" |