Skip to content

Commit 99422b4

Browse files
committed
test(db2): expect None rowcount for CTAS full_model (ibm_db_dbi DDL rowcount)
1 parent 5755b16 commit 99422b4

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

‎tests/core/engine_adapter/integration/test_integration.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2531,9 +2531,13 @@ def capture_execution_stats(
25312531

25322532
if ctx.engine_adapter.SUPPORTS_QUERY_EXECUTION_TRACKING:
25332533
assert actual_execution_stats["incremental_model"].total_rows_processed == 7
2534-
# snowflake and redshift don't track rows for CTAS
2534+
# snowflake, redshift, and db2 don't track rows for CTAS (ibm_db_dbi returns -1 rowcount for DDL)
25352535
assert actual_execution_stats["full_model"].total_rows_processed == (
2536-
None if ctx.mark.startswith("snowflake") or ctx.mark.startswith("redshift") else 3
2536+
None
2537+
if ctx.mark.startswith("snowflake")
2538+
or ctx.mark.startswith("redshift")
2539+
or ctx.mark.startswith("db2")
2540+
else 3
25372541
)
25382542
assert actual_execution_stats["seed_model"].total_rows_processed == (
25392543
None if ctx.mark.startswith("snowflake") else 7

0 commit comments

Comments
 (0)