make it more explicit in tests which dialect we want to use for things

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: