From 5c4733c03674268f749a1a6fdaefcad2cf786ac4 Mon Sep 17 00:00:00 2001 From: Dongjoon Hyun Date: Wed, 16 Oct 2024 09:14:23 -0700 Subject: [PATCH] Address comments --- .../spark/sql/execution/adaptive/AdaptiveQueryExecSuite.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/adaptive/AdaptiveQueryExecSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/adaptive/AdaptiveQueryExecSuite.scala index 2b8d65839a71e..141b211ff9249 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/execution/adaptive/AdaptiveQueryExecSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/execution/adaptive/AdaptiveQueryExecSuite.scala @@ -936,8 +936,8 @@ class AdaptiveQueryExecSuite val error = intercept[SparkException] { joined.collect() } - assert(error.getMessage().contains("coalesce test error") || - error.getMessage().contains("Multiple failures in stage materialization")) + assert((Seq(error) ++ Option(error.getCause) ++ error.getSuppressed()).exists( + e => e.getMessage() != null && e.getMessage().contains("coalesce test error"))) val adaptivePlan = joined.queryExecution.executedPlan.asInstanceOf[AdaptiveSparkPlanExec]