diff options
Diffstat (limited to 'lib/sqlalchemy/testing/replay_fixture.py')
| -rw-r--r-- | lib/sqlalchemy/testing/replay_fixture.py | 24 |
1 files changed, 15 insertions, 9 deletions
diff --git a/lib/sqlalchemy/testing/replay_fixture.py b/lib/sqlalchemy/testing/replay_fixture.py index 9832b07a2..613d7572b 100644 --- a/lib/sqlalchemy/testing/replay_fixture.py +++ b/lib/sqlalchemy/testing/replay_fixture.py @@ -1,13 +1,14 @@ -from . import fixtures -from . import profiling -from .. import util -import types from collections import deque import contextlib +import types + from . import config -from sqlalchemy import MetaData -from sqlalchemy import create_engine -from sqlalchemy.orm import Session +from . import fixtures +from . import profiling +from .. import create_engine +from .. import MetaData +from .. import util +from ..orm import Session class ReplayFixtureTest(fixtures.TestBase): @@ -19,7 +20,10 @@ class ReplayFixtureTest(fixtures.TestBase): dbapi_session = ReplayableSession() creator = config.db.pool._creator - recorder = lambda: dbapi_session.recorder(creator()) + + def recorder(): + return dbapi_session.recorder(creator()) + engine = create_engine( config.db.url, creator=recorder, use_native_hstore=False ) @@ -34,7 +38,9 @@ class ReplayFixtureTest(fixtures.TestBase): self.teardown_engine() engine.dispose() - player = lambda: dbapi_session.player() + def player(): + return dbapi_session.player() + engine = create_engine( config.db.url, creator=player, use_native_hstore=False ) |
