diff options
| -rw-r--r-- | tests/test_grouping.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/test_grouping.py b/tests/test_grouping.py index a6c4028..9a9dd6a 100644 --- a/tests/test_grouping.py +++ b/tests/test_grouping.py @@ -229,7 +229,8 @@ class TestGrouping(TestCaseBase): class TestStatement(TestCaseBase): def test_get_type(self): - f = lambda sql: sqlparse.parse(sql)[0] + def f(sql): + return sqlparse.parse(sql)[0] self.assertEqual(f('select * from foo').get_type(), 'SELECT') self.assertEqual(f('update foo').get_type(), 'UPDATE') self.assertEqual(f(' update foo').get_type(), 'UPDATE') @@ -332,14 +333,14 @@ def test_comparison_with_functions(): # issue230 assert len(p.tokens[0].tokens) == 5 assert p.tokens[0].left.value == 'foo' assert p.tokens[0].right.value == 'DATE(bar.baz)' - + p = sqlparse.parse('DATE(foo.bar) = DATE(bar.baz)')[0] assert len(p.tokens) == 1 assert isinstance(p.tokens[0], sql.Comparison) assert len(p.tokens[0].tokens) == 5 assert p.tokens[0].left.value == 'DATE(foo.bar)' assert p.tokens[0].right.value == 'DATE(bar.baz)' - + p = sqlparse.parse('DATE(foo.bar) = bar.baz')[0] assert len(p.tokens) == 1 assert isinstance(p.tokens[0], sql.Comparison) |