From 87002643407f886f13a3b53283ea0b6dafa695cc Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Wed, 16 Jan 2013 20:10:55 -0500 Subject: - preparing for #695, modernize constraint/index tests --- lib/sqlalchemy/testing/assertsql.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) (limited to 'lib/sqlalchemy/testing/assertsql.py') diff --git a/lib/sqlalchemy/testing/assertsql.py b/lib/sqlalchemy/testing/assertsql.py index d955d1554..864ce5b4d 100644 --- a/lib/sqlalchemy/testing/assertsql.py +++ b/lib/sqlalchemy/testing/assertsql.py @@ -144,7 +144,7 @@ class RegexSQL(SQLMatchRule): class CompiledSQL(SQLMatchRule): - def __init__(self, statement, params): + def __init__(self, statement, params=None): SQLMatchRule.__init__(self) self.statement = statement self.params = params @@ -153,14 +153,19 @@ class CompiledSQL(SQLMatchRule): executemany): if not context: return + from sqlalchemy.schema import _DDLCompiles _received_parameters = list(context.compiled_parameters) # recompile from the context, using the default dialect - compiled = \ - context.compiled.statement.compile(dialect=DefaultDialect(), + if isinstance(context.compiled.statement, _DDLCompiles): + compiled = \ + context.compiled.statement.compile(dialect=DefaultDialect()) + else: + compiled = \ + context.compiled.statement.compile(dialect=DefaultDialect(), column_keys=context.compiled.column_keys) - _received_statement = re.sub(r'\n', '', str(compiled)) + _received_statement = re.sub(r'[\n\t]', '', str(compiled)) equivalent = self.statement == _received_statement if self.params: if util.callable(self.params): -- cgit v1.2.1