TypeError when using --include-ctes with DATE, DATETIME, or TIMESTAMP columns in sqlmesh create_test

[BK1147]

Hi team!

When creating unit tests, if a column has a data type of DATE, DATETIME, or TIMESTAMP, the command
sqlmesh create_test {model} --include-ctes --query {} throws the following error:

TypeError: list indices must be integers or slices, not str

Without the --include-ctes option, it works perfectly (I believe there are no structural issues or incorrect column types). However, when using --include-ctes, it only works if those columns are cast to STRING. (I’m using the BigQuery engine with the default DuckDB test engine.)

Here is a simple code example based on the SQLMesh documentation.

model_unit_test.sql

MODEL (
 name sqlmesh_example.filtered_cte_view,
 kind VIEW,
 audits (
 NOT_NULL(columns = [id])
 ),
 ignored_rules ['ambiguousorinvalidcolumn']
);

WITH filtered_seed_cte AS (
 SELECT
 id,
 item_id,
 event_date
 FROM sqlmesh_example.seed_model
 WHERE
 item_id = 1
)
SELECT
 item_id,
 COUNT(DISTINCT id) AS num_orders
FROM filtered_seed_cte
GROUP BY
 item_id

seed_model.sql

MODEL (
 name sqlmesh_example.seed_model,
 kind SEED (
 path '../seeds/seed_data.csv'
 ),
 audits (
 unique_values(columns := (
 id
 ))
 )
)

Command for creating the unit test:

sqlmesh create_test "sqlmesh_example.filtered_cte_view" \
 --include-ctes \
 --query "sqlmesh_example.seed_model" \
 "SELECT
 4 AS id,
 1 AS item_id
 UNION ALL
 SELECT
 5,
 1 
 UNION ALL
 SELECT
 7,
 2 
 UNION ALL
 SELECT
 8,
 2 " \
 --overwrite

Could someone from the SQLMesh team take a look at this issue?

Context: https://tobiko-data.slack.com/archives/C044BRE5W4S/p1760719851382139

[brucearctor]

Confirmed addressed, Thanks @treysp !