From 16e14008faf8a35fd42c0f4a5c2e36b17309c649 Mon Sep 17 00:00:00 2001 From: Sander Mathijs van Veen Date: Wed, 29 Jul 2026 11:20:50 +0200 Subject: [PATCH] Replace stmt_length with query --- src/pystatement.cpp | 4 +--- tests/fast/api/test_duckdb_connection.py | 4 ++-- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/pystatement.cpp b/src/pystatement.cpp index 18a37b37..cef9ae32 100644 --- a/src/pystatement.cpp +++ b/src/pystatement.cpp @@ -28,9 +28,7 @@ unique_ptr DuckDBPyStatement::GetStatement() { } string DuckDBPyStatement::Query() const { - auto &loc = statement->stmt_location; - auto &length = statement->stmt_length; - return statement->query.substr(loc, length); + return statement->query; } nb::set DuckDBPyStatement::NamedParameters() const { diff --git a/tests/fast/api/test_duckdb_connection.py b/tests/fast/api/test_duckdb_connection.py index 1c70abaf..0617bea6 100644 --- a/tests/fast/api/test_duckdb_connection.py +++ b/tests/fast/api/test_duckdb_connection.py @@ -134,12 +134,12 @@ def test_pystatement(self): statements = duckdb.extract_statements("select $1; select 21") assert len(statements) == 2 - assert statements[0].query.startswith("select $1") + assert statements[0].query == "select $1; " assert statements[0].type == duckdb.StatementType.SELECT assert statements[0].named_parameters == set("1") assert statements[0].expected_result_type == [duckdb.ExpectedResultType.QUERY_RESULT] - assert statements[1].query.startswith("select 21") + assert statements[1].query == "select 21" assert statements[1].type == duckdb.StatementType.SELECT assert statements[1].named_parameters == set()