diff --git a/tests/integration/test_sqlalchemy_integration.py b/tests/integration/test_sqlalchemy_integration.py index 1dc8f05a..fe9a788e 100644 --- a/tests/integration/test_sqlalchemy_integration.py +++ b/tests/integration/test_sqlalchemy_integration.py @@ -102,7 +102,9 @@ def test_insert(trino_connection): schema="test") metadata.create_all(engine) ins = users.insert() - conn.execute(ins, {"id": 2, "name": "wendy", "fullname": "Wendy Williams"}) + result = conn.execute(ins, {"id": 2, "name": "wendy", "fullname": "Wendy Williams"}) + # TODO: Can we get rid of this? + result.fetchall() query = sqla.select(users) result = conn.execute(query) rows = result.fetchall() @@ -126,11 +128,13 @@ def test_insert_multiple_statements(trino_connection): schema="test") metadata.create_all(engine) ins = users.insert() - conn.execute(ins, [ + result = conn.execute(ins, [ {"id": 2, "name": "wendy", "fullname": "Wendy Williams"}, {"id": 3, "name": "john", "fullname": "John Doe"}, {"id": 4, "name": "mary", "fullname": "Mary Hopkins"}, ]) + # TODO: Can we get rid of this? + result.fetchall() query = sqla.select(users) result = conn.execute(query) rows = result.fetchall()