diff options
Diffstat (limited to 'lib/sqlalchemy/testing/engines.py')
| -rw-r--r-- | lib/sqlalchemy/testing/engines.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/lib/sqlalchemy/testing/engines.py b/lib/sqlalchemy/testing/engines.py index 910af5876..280e6901e 100644 --- a/lib/sqlalchemy/testing/engines.py +++ b/lib/sqlalchemy/testing/engines.py @@ -238,10 +238,13 @@ def reconnecting_engine(url=None, options=None): return engine -def testing_engine(url=None, options=None): +def testing_engine(url=None, options=None, future=False): """Produce an engine configured by --options with optional overrides.""" - from sqlalchemy import create_engine + if future or config.db and config.db._is_future: + from sqlalchemy.future import create_engine + else: + from sqlalchemy import create_engine from sqlalchemy.engine.url import make_url if not options: |