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]