From 4af95c1a0a719f0749744c99f594ae0ad5450729 Mon Sep 17 00:00:00 2001 From: Dereck Li Date: Fri, 1 Apr 2022 11:50:53 +0800 Subject: [PATCH] [SPARK-38754][SQL][TEST] Using EquivalentExpressions getEquivalentExprs function instead of getExprState at SubexpressionEliminationSuite --- .../catalyst/expressions/SubexpressionEliminationSuite.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/SubexpressionEliminationSuite.scala b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/SubexpressionEliminationSuite.scala index 6071b4bd296cc..09007659728bd 100644 --- a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/SubexpressionEliminationSuite.scala +++ b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/SubexpressionEliminationSuite.scala @@ -330,13 +330,13 @@ class SubexpressionEliminationSuite extends SparkFunSuite with ExpressionEvalHel test("SPARK-38333: PlanExpression expression should skip addExprTree function in Executor") { try { // suppose we are in executor - val context1 = new TaskContextImpl(0, 0, 0, 0, 0, null, null, null, cpus = 0) + val context1 = new TaskContextImpl(0, 0, 0, 0, 0, null, null, null) TaskContext.setTaskContext(context1) val equivalence = new EquivalentExpressions val expression = DynamicPruningExpression(Exists(LocalRelation())) equivalence.addExprTree(expression) - assert(equivalence.getExprState(expression).isEmpty) + assert(equivalence.getEquivalentExprs(expression).size == 0) } finally { TaskContext.unset() }