diff --git a/velox/exec/tests/utils/QueryAssertions.cpp b/velox/exec/tests/utils/QueryAssertions.cpp index dfc996e2063b6..73d3937350ec4 100644 --- a/velox/exec/tests/utils/QueryAssertions.cpp +++ b/velox/exec/tests/utils/QueryAssertions.cpp @@ -39,6 +39,30 @@ ::duckdb::Value duckValueAt(const VectorPtr& vector, vector_size_t index) { return ::duckdb::Value(vector->as>()->valueAt(index)); } +template <> +::duckdb::Value duckValueAt( + const VectorPtr& vector, + vector_size_t index) { + return ::duckdb::Value::TINYINT( + vector->as>()->valueAt(index)); +} + +template <> +::duckdb::Value duckValueAt( + const VectorPtr& vector, + vector_size_t index) { + return ::duckdb::Value::TINYINT( + vector->as>()->valueAt(index)); +} + +template <> +::duckdb::Value duckValueAt( + const VectorPtr& vector, + vector_size_t index) { + return ::duckdb::Value::BOOLEAN( + vector->as>()->valueAt(index)); +} + template <> ::duckdb::Value duckValueAt( const VectorPtr& vector,