diff options
| author | Darik Gamble <darik.gamble@gmail.com> | 2015-06-30 07:35:53 -0400 |
|---|---|---|
| committer | Darik Gamble <darik.gamble@gmail.com> | 2015-06-30 12:40:53 -0400 |
| commit | 581061440b89971432009a43e2d7a0abe321d77c (patch) | |
| tree | e46c750f9a30c95b1e741baffe74bd888b5eb91e /tests/utils.py | |
| parent | 878f6e0bdeeff951628e8e77af62cad04aa417b8 (diff) | |
pg_execute.run accepts a PGSpecial object, instead of calling special.execute directly
Diffstat (limited to 'tests/utils.py')
| -rw-r--r-- | tests/utils.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/utils.py b/tests/utils.py index 0287a2e9..93e9f258 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -58,10 +58,10 @@ def drop_tables(conn): DROP SCHEMA IF EXISTS schema2 CASCADE''') -def run(executor, sql, join=False, expanded=False): +def run(executor, sql, join=False, expanded=False, pgspecial=None): " Return string output for the sql to be run " result = [] - for title, rows, headers, status in executor.run(sql): + for title, rows, headers, status in executor.run(sql, pgspecial): result.extend(format_output(title, rows, headers, status, 'psql', expanded=expanded)) if join: |