diff options
Diffstat (limited to 'test/lib/testing.py')
| -rw-r--r-- | test/lib/testing.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/test/lib/testing.py b/test/lib/testing.py index 36a8c8d1a..bf852de7c 100644 --- a/test/lib/testing.py +++ b/test/lib/testing.py @@ -628,12 +628,17 @@ class TestBase(object): class AssertsCompiledSQL(object): def assert_compile(self, clause, result, params=None, checkparams=None, dialect=None, - use_default_dialect=False): + use_default_dialect=False, + allow_dialect_select=False): + if use_default_dialect: dialect = default.DefaultDialect() - - if dialect is None: + elif dialect == None and not allow_dialect_select: dialect = getattr(self, '__dialect__', None) + if dialect == 'default': + dialect = default.DefaultDialect() + elif dialect is None: + dialect = db.dialect kw = {} if params is not None: |