|
10 | 10 | class TestGrouping(TestCaseBase): |
11 | 11 |
|
12 | 12 | def test_parenthesis(self): |
13 | | - s ='select (select (x3) x2) and (y2) bar' |
| 13 | + s = 'select (select (x3) x2) and (y2) bar' |
14 | 14 | parsed = sqlparse.parse(s)[0] |
15 | 15 | self.ndiffAssertEqual(s, str(parsed)) |
16 | 16 | self.assertEqual(len(parsed.tokens), 9) |
@@ -39,18 +39,18 @@ def test_assignment(self): |
39 | 39 | def test_identifiers(self): |
40 | 40 | s = 'select foo.bar from "myscheme"."table" where fail. order' |
41 | 41 | parsed = sqlparse.parse(s)[0] |
42 | | - self.ndiffAssertEqual(s, parsed.to_unicode()) |
| 42 | + self.ndiffAssertEqual(s, unicode(parsed)) |
43 | 43 | self.assert_(isinstance(parsed.tokens[2], sql.Identifier)) |
44 | 44 | self.assert_(isinstance(parsed.tokens[6], sql.Identifier)) |
45 | 45 | self.assert_(isinstance(parsed.tokens[8], sql.Where)) |
46 | 46 | s = 'select * from foo where foo.id = 1' |
47 | 47 | parsed = sqlparse.parse(s)[0] |
48 | | - self.ndiffAssertEqual(s, parsed.to_unicode()) |
| 48 | + self.ndiffAssertEqual(s, unicode(parsed)) |
49 | 49 | self.assert_(isinstance(parsed.tokens[-1].tokens[-1].tokens[0], |
50 | 50 | sql.Identifier)) |
51 | 51 | s = 'select * from (select "foo"."id" from foo)' |
52 | 52 | parsed = sqlparse.parse(s)[0] |
53 | | - self.ndiffAssertEqual(s, parsed.to_unicode()) |
| 53 | + self.ndiffAssertEqual(s, unicode(parsed)) |
54 | 54 | self.assert_(isinstance(parsed.tokens[-1].tokens[3], sql.Identifier)) |
55 | 55 |
|
56 | 56 | def test_identifier_wildcard(self): |
@@ -117,44 +117,44 @@ def test_identifier_list_other(self): # issue2 |
117 | 117 | def test_where(self): |
118 | 118 | s = 'select * from foo where bar = 1 order by id desc' |
119 | 119 | p = sqlparse.parse(s)[0] |
120 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 120 | + self.ndiffAssertEqual(s, unicode(p)) |
121 | 121 | self.assertTrue(len(p.tokens), 16) |
122 | 122 | s = 'select x from (select y from foo where bar = 1) z' |
123 | 123 | p = sqlparse.parse(s)[0] |
124 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 124 | + self.ndiffAssertEqual(s, unicode(p)) |
125 | 125 | self.assertTrue(isinstance(p.tokens[-3].tokens[-2], sql.Where)) |
126 | 126 |
|
127 | 127 | def test_typecast(self): |
128 | 128 | s = 'select foo::integer from bar' |
129 | 129 | p = sqlparse.parse(s)[0] |
130 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 130 | + self.ndiffAssertEqual(s, unicode(p)) |
131 | 131 | self.assertEqual(p.tokens[2].get_typecast(), 'integer') |
132 | 132 | self.assertEqual(p.tokens[2].get_name(), 'foo') |
133 | 133 | s = 'select (current_database())::information_schema.sql_identifier' |
134 | 134 | p = sqlparse.parse(s)[0] |
135 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 135 | + self.ndiffAssertEqual(s, unicode(p)) |
136 | 136 | self.assertEqual(p.tokens[2].get_typecast(), |
137 | 137 | 'information_schema.sql_identifier') |
138 | 138 |
|
139 | 139 | def test_alias(self): |
140 | 140 | s = 'select foo as bar from mytable' |
141 | 141 | p = sqlparse.parse(s)[0] |
142 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 142 | + self.ndiffAssertEqual(s, unicode(p)) |
143 | 143 | self.assertEqual(p.tokens[2].get_real_name(), 'foo') |
144 | 144 | self.assertEqual(p.tokens[2].get_alias(), 'bar') |
145 | 145 | s = 'select foo from mytable t1' |
146 | 146 | p = sqlparse.parse(s)[0] |
147 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 147 | + self.ndiffAssertEqual(s, unicode(p)) |
148 | 148 | self.assertEqual(p.tokens[6].get_real_name(), 'mytable') |
149 | 149 | self.assertEqual(p.tokens[6].get_alias(), 't1') |
150 | 150 | s = 'select foo::integer as bar from mytable' |
151 | 151 | p = sqlparse.parse(s)[0] |
152 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 152 | + self.ndiffAssertEqual(s, unicode(p)) |
153 | 153 | self.assertEqual(p.tokens[2].get_alias(), 'bar') |
154 | 154 | s = ('SELECT DISTINCT ' |
155 | 155 | '(current_database())::information_schema.sql_identifier AS view') |
156 | 156 | p = sqlparse.parse(s)[0] |
157 | | - self.ndiffAssertEqual(s, p.to_unicode()) |
| 157 | + self.ndiffAssertEqual(s, unicode(p)) |
158 | 158 | self.assertEqual(p.tokens[4].get_alias(), 'view') |
159 | 159 |
|
160 | 160 | def test_alias_case(self): # see issue46 |
|
0 commit comments