diff --git a/tests/integration/dbapi/async/test_queries_async.py b/tests/integration/dbapi/async/test_queries_async.py index 1d64bfdb64d..90c1b665e9d 100644 --- a/tests/integration/dbapi/async/test_queries_async.py +++ b/tests/integration/dbapi/async/test_queries_async.py @@ -382,15 +382,11 @@ async def test_multi_statement_query(connection: Connection) -> None: " primary index i" ) - assert ( - await c.execute( - "INSERT INTO test_tb_async_multi_statement values (1, 'a'), (2, 'b');" - "SELECT * FROM test_tb_async_multi_statement;" - "SELECT * FROM test_tb_async_multi_statement WHERE i <= 1" - ) - == 1 # Insert always returns 1 row with num of rows modified - ), "Invalid row count returned for insert" - assert c.rowcount == 1, "Invalid row count" + await c.execute( + "INSERT INTO test_tb_async_multi_statement values (1, 'a'), (2, 'b');" + "SELECT * FROM test_tb_async_multi_statement;" + "SELECT * FROM test_tb_async_multi_statement WHERE i <= 1" + ) assert c.description is None, "Invalid description" assert await c.nextset() diff --git a/tests/integration/dbapi/sync/test_queries.py b/tests/integration/dbapi/sync/test_queries.py index 55ab830d955..a409dea4729 100644 --- a/tests/integration/dbapi/sync/test_queries.py +++ b/tests/integration/dbapi/sync/test_queries.py @@ -327,15 +327,11 @@ def test_multi_statement_query(connection: Connection) -> None: "CREATE FACT TABLE test_tb_multi_statement(i int, s string) primary index i" ) - assert ( - c.execute( - "INSERT INTO test_tb_multi_statement values (1, 'a'), (2, 'b');" - "SELECT * FROM test_tb_multi_statement;" - "SELECT * FROM test_tb_multi_statement WHERE i <= 1" - ) - == 1 # Insert always returns 1 row with num of rows modified - ), "Invalid row count returned for insert" - assert c.rowcount == 1, "Invalid row count" + c.execute( + "INSERT INTO test_tb_multi_statement values (1, 'a'), (2, 'b');" + "SELECT * FROM test_tb_multi_statement;" + "SELECT * FROM test_tb_multi_statement WHERE i <= 1" + ) assert c.description is None, "Invalid description" assert c.nextset()