From 4976d84daedabb41d1d59ba5766c8c1c2206b043 Mon Sep 17 00:00:00 2001 From: David Li Date: Tue, 4 Aug 2026 13:40:18 +0900 Subject: [PATCH] test: allow overriding another query --- adbc_drivers_validation/tests/statement.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/adbc_drivers_validation/tests/statement.py b/adbc_drivers_validation/tests/statement.py index 8e3e07f..45db0ae 100644 --- a/adbc_drivers_validation/tests/statement.py +++ b/adbc_drivers_validation/tests/statement.py @@ -146,9 +146,11 @@ def test_parameter_null_typed( cursor.adbc_statement.execute_update() with conn.cursor() as cursor: - cursor.adbc_statement.set_sql_query( - f"SELECT {value} FROM {sample_table} WHERE {id_} = 7001" + query = driver.query_override( + "TestStatement.test_parameter_null_typed", + f"SELECT {value} FROM {sample_table} WHERE {id_} = 7001", ) + cursor.adbc_statement.set_sql_query(query) handle, _ = cursor.adbc_statement.execute_query() result = pyarrow.RecordBatchReader._import_from_c(handle.address).read_all() assert result[0].to_pylist() == [None]