|
12 | 12 | sys.path.insert(0, '..') |
13 | 13 |
|
14 | 14 | from sqlparse.filters import Compact |
15 | | -from sqlparse.functions import GetColumns, GetLimit, IsType |
| 15 | +from sqlparse.functions import getcolumns, getlimit, IsType |
16 | 16 |
|
17 | 17 |
|
18 | 18 | class Test_SQL(TestCase): |
@@ -86,31 +86,31 @@ def test_compact3(self): |
86 | 86 |
|
87 | 87 | class Test_GetColumns(Test_SQL): |
88 | 88 | def test_getcolumns1(self): |
89 | | - columns = GetColumns(tokenize(self.sql)) |
| 89 | + columns = getcolumns(tokenize(self.sql)) |
90 | 90 | self.assertEqual(columns, []) |
91 | 91 |
|
92 | 92 | def test_getcolumns2(self): |
93 | | - columns = GetColumns(tokenize(self.sql2)) |
| 93 | + columns = getcolumns(tokenize(self.sql2)) |
94 | 94 | self.assertEqual(columns, ['child_entry', 'inode', 'creation']) |
95 | 95 |
|
96 | 96 | def test_getcolumns3(self): |
97 | | - columns = GetColumns(tokenize(self.sql3)) |
| 97 | + columns = getcolumns(tokenize(self.sql3)) |
98 | 98 | self.assertEqual(columns, ['st_dev', 'st_uid', 'st_gid', 'st_mode', |
99 | 99 | 'st_ino', 'st_nlink', 'st_ctime', |
100 | 100 | 'st_atime', 'st_mtime', 'st_size', 'size']) |
101 | 101 |
|
102 | 102 |
|
103 | 103 | class Test_GetLimit(Test_SQL): |
104 | 104 | def test_getlimit1(self): |
105 | | - limit = GetLimit(tokenize(self.sql)) |
| 105 | + limit = getlimit(tokenize(self.sql)) |
106 | 106 | self.assertEqual(limit, 1) |
107 | 107 |
|
108 | 108 | def test_getlimit2(self): |
109 | | - limit = GetLimit(tokenize(self.sql2)) |
| 109 | + limit = getlimit(tokenize(self.sql2)) |
110 | 110 | self.assertEqual(limit, 1) |
111 | 111 |
|
112 | 112 | def test_getlimit3(self): |
113 | | - limit = GetLimit(tokenize(self.sql3)) |
| 113 | + limit = getlimit(tokenize(self.sql3)) |
114 | 114 | self.assertEqual(limit, 1) |
115 | 115 |
|
116 | 116 |
|
|
0 commit comments