diff options
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_dialect.py')
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_dialect.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_dialect.py b/lib/sqlalchemy/testing/suite/test_dialect.py index 28fd99876..daaea085d 100644 --- a/lib/sqlalchemy/testing/suite/test_dialect.py +++ b/lib/sqlalchemy/testing/suite/test_dialect.py @@ -115,7 +115,9 @@ class IsolationLevelTest(fixtures.TestBase): eq_(conn.get_isolation_level(), non_default) - conn.dialect.reset_isolation_level(conn.connection) + conn.dialect.reset_isolation_level( + conn.connection.dbapi_connection + ) eq_(conn.get_isolation_level(), existing) @@ -223,7 +225,7 @@ class AutocommitIsolationTest(fixtures.TablesTest): c2 = conn.execution_options(isolation_level="AUTOCOMMIT") self._test_conn_autocommits(c2, True) - c2.dialect.reset_isolation_level(c2.connection) + c2.dialect.reset_isolation_level(c2.connection.dbapi_connection) self._test_conn_autocommits(conn, False) |