diff options
Diffstat (limited to 'lib/sqlalchemy/testing')
| -rw-r--r-- | lib/sqlalchemy/testing/assertions.py | 4 | ||||
| -rw-r--r-- | lib/sqlalchemy/testing/assertsql.py | 8 |
2 files changed, 12 insertions, 0 deletions
diff --git a/lib/sqlalchemy/testing/assertions.py b/lib/sqlalchemy/testing/assertions.py index 1ea366dac..998dde66b 100644 --- a/lib/sqlalchemy/testing/assertions.py +++ b/lib/sqlalchemy/testing/assertions.py @@ -343,6 +343,7 @@ class AssertsCompiledSQL(object): result, params=None, checkparams=None, + for_executemany=False, check_literal_execute=None, check_post_param=None, dialect=None, @@ -391,6 +392,9 @@ class AssertsCompiledSQL(object): if render_postcompile: compile_kwargs["render_postcompile"] = True + if for_executemany: + kw["for_executemany"] = True + if render_schema_translate: kw["render_schema_translate"] = True diff --git a/lib/sqlalchemy/testing/assertsql.py b/lib/sqlalchemy/testing/assertsql.py index ef324635e..caf61a806 100644 --- a/lib/sqlalchemy/testing/assertsql.py +++ b/lib/sqlalchemy/testing/assertsql.py @@ -325,6 +325,14 @@ class EachOf(AssertRule): super(EachOf, self).no_more_statements() +class Conditional(EachOf): + def __init__(self, condition, rules, else_rules): + if condition: + super(Conditional, self).__init__(*rules) + else: + super(Conditional, self).__init__(*else_rules) + + class Or(AllOf): def process_statement(self, execute_observed): for rule in self.rules: |