diff options
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_insert.py')
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_insert.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_insert.py b/lib/sqlalchemy/testing/suite/test_insert.py index 2cc8761b8..931b0ef65 100644 --- a/lib/sqlalchemy/testing/suite/test_insert.py +++ b/lib/sqlalchemy/testing/suite/test_insert.py @@ -109,7 +109,8 @@ class InsertBehaviorTest(fixtures.TablesTest): else: engine = config.db - r = engine.execute(self.tables.autoinc_pk.insert(), data="some data") + with engine.begin() as conn: + r = conn.execute(self.tables.autoinc_pk.insert(), data="some data") assert r._soft_closed assert not r.closed assert r.is_insert @@ -278,9 +279,10 @@ class ReturningTest(fixtures.TablesTest): def test_explicit_returning_pk_autocommit(self): engine = config.db table = self.tables.autoinc_pk - r = engine.execute( - table.insert().returning(table.c.id), data="some data" - ) + with engine.begin() as conn: + r = conn.execute( + table.insert().returning(table.c.id), data="some data" + ) pk = r.first()[0] fetched_pk = config.db.scalar(select([table.c.id])) eq_(fetched_pk, pk) |