diff options
Diffstat (limited to 'test/dialect/postgresql/test_compiler.py')
| -rw-r--r-- | test/dialect/postgresql/test_compiler.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/dialect/postgresql/test_compiler.py b/test/dialect/postgresql/test_compiler.py index 316f0c240..4cc9c837d 100644 --- a/test/dialect/postgresql/test_compiler.py +++ b/test/dialect/postgresql/test_compiler.py @@ -87,8 +87,9 @@ class SequenceTest(fixtures.TestBase, AssertsCompiledSQL): Column(cname[:57], Integer, primary_key=True), ) t.create(engine) - r = engine.execute(t.insert()) - assert r.inserted_primary_key == [1] + with engine.begin() as conn: + r = conn.execute(t.insert()) + assert r.inserted_primary_key == [1] class CompileTest(fixtures.TestBase, AssertsCompiledSQL): |