psqlpy-python
diff --git a/‎Cargo.lock
Lines changed: 16 additions & 0 deletions b/‎Cargo.lock
Lines changed: 16 additions & 0 deletions
diff --git a/‎Cargo.toml
Lines changed: 1 addition & 1 deletion b/‎Cargo.toml
Lines changed: 1 addition & 1 deletion
diff --git a/‎python/tests/test_transaction.py
Lines changed: 8 additions & 10 deletions b/‎python/tests/test_transaction.py
Lines changed: 8 additions & 10 deletions
@@ -10,7 +10,7 @@ crate-type = ["cdylib"]
 
 [dependencies]
 deadpool-postgres = { git = "https://github.com/chandr-andr/deadpool.git", branch = "psqlpy" }
-pyo3 = { version = "0.23.4", features = ["chrono", "experimental-async", "rust_decimal", "py-clone", "macros"] }
+pyo3 = { version = "0.23.4", features = ["chrono", "experimental-async", "rust_decimal", "py-clone", "macros", "multiple-pymethods"] }
 pyo3-async-runtimes = { git = "https://github.com/chandr-andr/pyo3-async-runtimes.git", branch = "psqlpy", features = [
     "tokio-runtime",
 ] }
 
@@ -9,11 +9,10 @@
     IsolationLevel,
     ReadVariant,
 )
+from psqlpy._internal.exceptions import TransactionClosedError
 from psqlpy.exceptions import (
     InterfaceError,
-    TransactionBeginError,
     TransactionExecuteError,
-    TransactionSavepointError,
 )
 
 from tests.helpers import count_rows_in_test_table
@@ -64,10 +63,10 @@ async def test_transaction_begin(
     connection = await psql_pool.connection()
     transaction = connection.transaction()
 
-    with pytest.raises(expected_exception=TransactionBeginError):
-        await transaction.execute(
-            f"SELECT * FROM {table_name}",
-        )
+    # with pytest.raises(expected_exception=TransactionBeginError):
+    await transaction.execute(
+        f"SELECT * FROM {table_name}",
+    )
 
     await transaction.begin()
 
@@ -170,7 +169,7 @@ async def test_transaction_rollback(
 
     await transaction.rollback()
 
-    with pytest.raises(expected_exception=TransactionBeginError):
+    with pytest.raises(expected_exception=TransactionClosedError):
         await transaction.execute(
             f"SELECT * FROM {table_name} WHERE name = $1",
             parameters=[test_name],
@@ -198,9 +197,8 @@ async def test_transaction_release_savepoint(
     sp_name_2 = "sp2"
 
     await transaction.create_savepoint(sp_name_1)
-
-    with pytest.raises(expected_exception=TransactionSavepointError):
-        await transaction.create_savepoint(sp_name_1)
+    # There is no problem in creating the same sp_name
+    await transaction.create_savepoint(sp_name_1)
 
     await transaction.create_savepoint(sp_name_2)