Skip to content
This repository was archived by the owner on May 17, 2024. It is now read-only.

Commit 06959d3

Browse files
committed
Continue converting tests to new query-builder
1 parent 6b533b2 commit 06959d3

File tree

3 files changed

+260
-262
lines changed

3 files changed

+260
-262
lines changed

data_diff/databases/base.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,10 @@ def query(self, sql_ast: Union[Expr, Generator], res_type: type = list):
122122
compiler = Compiler(self)
123123
if isinstance(sql_ast, Generator):
124124
sql_code = ThreadLocalInterpreter(compiler, sql_ast)
125+
elif isinstance(sql_ast, list):
126+
for i in sql_ast[:-1]:
127+
self.query(i)
128+
return self.query(sql_ast[-1], res_type)
125129
else:
126130
sql_code = compiler.compile(sql_ast)
127131
if sql_code is SKIP:

0 commit comments

Comments
 (0)