From e65206fa3a28f2e87dbd8801c233086893dfd747 Mon Sep 17 00:00:00 2001 From: Afroz Alam Date: Mon, 11 Nov 2024 13:04:57 -0800 Subject: [PATCH] Fix daily pre-commit tests --- tests/integ/test_multithreading.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tests/integ/test_multithreading.py b/tests/integ/test_multithreading.py index 23b0814e7d1..fe7db2d5e40 100644 --- a/tests/integ/test_multithreading.py +++ b/tests/integ/test_multithreading.py @@ -937,7 +937,9 @@ def call_critical_lazy_methods(df_): assert df_._plan.cumulative_node_complexity == { PlanNodeCategory.FILTER: 2, PlanNodeCategory.LITERAL: 22, - PlanNodeCategory.COLUMN: 64, + PlanNodeCategory.COLUMN: 64 + if threadsafe_session.sql_simplifier_enabled + else 62, PlanNodeCategory.LOW_IMPACT: 42, PlanNodeCategory.SET_OPERATION: 1, } @@ -947,10 +949,10 @@ def call_critical_lazy_methods(df_): PlanState.NUM_SELECTS_WITH_COMPLEXITY_MERGED: 0, PlanState.DUPLICATED_NODE_COMPLEXITY_DISTRIBUTION: [2, 0, 0, 0, 0, 0, 0], } - assert ( - df_._select_statement.encoded_node_id_with_query - == "b04d566533_SelectStatement" - ) + if threadsafe_session.sql_simplifier_enabled: + assert df_._select_statement.encoded_node_id_with_query.endswith( + "_SelectStatement" + ) with ThreadPoolExecutor(max_workers=5) as executor: futures = [executor.submit(call_critical_lazy_methods, df) for _ in range(10)]