From 57e27858a9b1ce8edad1feb8665028347dd722cf Mon Sep 17 00:00:00 2001 From: Vasily Nemkov Date: Sun, 6 Jul 2025 20:36:42 +0000 Subject: [PATCH] Attempt to fix test_storage_iceberg/test.py::test_cluster_table_function After merging #878 Bringing tests closer to upstream. --- .../integration/test_storage_iceberg/test.py | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/tests/integration/test_storage_iceberg/test.py b/tests/integration/test_storage_iceberg/test.py index 2b4e71291dfe..87bea8d25018 100644 --- a/tests/integration/test_storage_iceberg/test.py +++ b/tests/integration/test_storage_iceberg/test.py @@ -711,6 +711,26 @@ def add_df(mode): instance.query(f"SELECT * FROM {table_function_expr}").strip().split() ) + # Cluster Query with node1 as coordinator + table_function_expr_cluster = get_creation_expression( + storage_type, + TABLE_NAME, + started_cluster, + table_function=True, + run_on_cluster=True, + ) + + select_cluster = ( + instance.query(f"SELECT * FROM {table_function_expr_cluster}").strip().split() + ) + + # Simple size check + assert len(select_regular) == 600 + assert len(select_cluster) == 600 + + # Actual check + assert select_cluster == select_regular + def make_query_from_function( run_on_cluster=False, alt_syntax=False,