From baeea12b776b1f4477ee5af80aaa2e7c44fadf0e Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Wed, 6 Apr 2016 08:43:53 +0200 Subject: Code cleanup --- tests/test_grouping.py | 7 ++++--- 1 file 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) -- cgit v1.2.1