From 7ced756287757a7e4ff070188fe4789854fd3b2c Mon Sep 17 00:00:00 2001 From: shalili Date: Mon, 19 May 2025 21:40:03 +0800 Subject: [PATCH] fix ColumnarBasicFunctionSuite --- .../sql/execution/forsql/ColumnarBasicFunctionSuite.scala | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/omnioperator/omniop-spark-extension/java/src/test/scala/org/apache/spark/sql/execution/forsql/ColumnarBasicFunctionSuite.scala b/omnioperator/omniop-spark-extension/java/src/test/scala/org/apache/spark/sql/execution/forsql/ColumnarBasicFunctionSuite.scala index c19c11e00..0af7ebe9c 100644 --- a/omnioperator/omniop-spark-extension/java/src/test/scala/org/apache/spark/sql/execution/forsql/ColumnarBasicFunctionSuite.scala +++ b/omnioperator/omniop-spark-extension/java/src/test/scala/org/apache/spark/sql/execution/forsql/ColumnarBasicFunctionSuite.scala @@ -68,10 +68,10 @@ class ColumnarBasicFunctionSuite extends QueryTest with SharedSparkSession { insert = spark.sql("insert into table target_table select UPPER(name) as name, SUM(amount) as total_amount from " + "source_table where amount >= 10 GROUP BY UPPER(name)") insert.collect() - assert(insert.queryExecution.executedPlan.asInstanceOf[CommandResultExec] - .commandPhysicalPlan.toString().contains("OmniColumnarHashAggregate"), "use columnar data writing command") - val columnarDataWrite = insert.queryExecution.executedPlan.asInstanceOf[CommandResultExec] - .commandPhysicalPlan.find({ + assert(insert.queryExecution.executedPlan.toString().contains("OmniColumnarHashAggregate"), + "use columnar data writing command") + val columnarDataWrite = insert.queryExecution.executedPlan + .find({ case _: HashAggregateExec => true case _ => false }) -- Gitee