Skip to content

Commit 08d2437

Browse files
committed
fix test
1 parent cb5364b commit 08d2437

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

src/queryparser/testing/utils.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -37,24 +37,27 @@ def _test_parsing(query_processor, test, translate=False):
3737
elif query_processor == PostgreSQLQueryProcessor:
3838
query = adt.to_postgresql()
3939

40-
if replace_schema_name is None:
41-
qp = query_processor(query)
42-
else:
43-
qp = query_processor()
44-
qp.set_query(query)
45-
qp.process_query(replace_schema_name=replace_schema_name)
40+
if replace_function_names is None:
41+
replace_function_names = []
42+
43+
qp = query_processor()
44+
qp.set_query(query)
45+
qp.process_query(
46+
replace_schema_name=replace_schema_name,
47+
replace_function_names=replace_function_names,
48+
)
4649

4750
qp_columns = [
48-
".".join([str(j) for j in i[:3]])
51+
'.'.join([str(j) for j in i[:3]])
4952
for i in qp.columns
5053
if i[0] is not None and i[1] is not None
5154
]
5255
qp_display_columns = [
53-
"%s: %s" % (str(i[0]), ".".join([str(j) for j in i[1]]))
56+
'%s: %s' % (str(i[0]), '.'.join([str(j) for j in i[1]]))
5457
for i in qp.display_columns
5558
]
5659
qp_tables = [
57-
".".join([str(j) for j in i])
60+
'.'.join([str(j) for j in i])
5861
for i in qp.tables
5962
if i[0] is not None and i[1] is not None
6063
]

0 commit comments

Comments
 (0)