chore: fix tests to include dataframe

eakmanrq · eakmanrq · commit 5583459d0747 · 2025-09-22T16:20:27.000-07:00
diff --git a/sqlmesh/core/engine_adapter/clickhouse.py b/sqlmesh/core/engine_adapter/clickhouse.py
@@ -113,7 +113,9 @@ def query_factory() -> Query:
                     **kwargs,
                 )
 
-                self.cursor.client.insert_df(temp_table.sql(dialect=self.dialect), df=df)
+                self.cursor.client.insert_df(
+                    temp_table.sql(dialect=self.dialect), df=df, column_names=source_columns
+                )
 
             return exp.select(*self._casted_columns(target_columns_to_types, source_columns)).from_(
                 temp_table
diff --git a/tests/core/engine_adapter/integration/conftest.py b/tests/core/engine_adapter/integration/conftest.py
@@ -148,7 +148,7 @@ def ctx_df(
     yield from create_test_context(*request.param)
 
 
-@pytest.fixture(params=list(generate_pytest_params(ENGINES, query=True, df=False)))
+@pytest.fixture(params=list(generate_pytest_params(ENGINES, query=True, df=True)))
 def ctx_query_and_df(
     request: FixtureRequest,
     create_test_context: t.Callable[[IntegrationTestEngine, str], t.Iterable[TestContext]],

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,9 @@ def query_factory() -> Query:`
`113`	`113`	`**kwargs,`
`114`	`114`	`)`
`115`	`115`
`116`		`- self.cursor.client.insert_df(temp_table.sql(dialect=self.dialect), df=df)`
	`116`	`+ self.cursor.client.insert_df(`
	`117`	`+ temp_table.sql(dialect=self.dialect), df=df, column_names=source_columns`
	`118`	`+ )`
`117`	`119`
`118`	`120`	`return exp.select(*self._casted_columns(target_columns_to_types, source_columns)).from_(`
`119`	`121`	`temp_table`