From 427a3346f126bb7d734b41567c957c12521fddf0 Mon Sep 17 00:00:00 2001 From: Max Gekk Date: Thu, 15 Feb 2024 22:53:45 +0300 Subject: [PATCH] Fix TryCastSuite --- .../spark/sql/catalyst/expressions/TryCastSuite.scala | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/TryCastSuite.scala b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/TryCastSuite.scala index 9ead075663540..f138d9642d1e1 100644 --- a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/TryCastSuite.scala +++ b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/TryCastSuite.scala @@ -19,7 +19,7 @@ package org.apache.spark.sql.catalyst.expressions import scala.reflect.ClassTag -import org.apache.spark.SparkFunSuite +import org.apache.spark.{SparkFunSuite, SparkThrowable} import org.apache.spark.sql.catalyst.InternalRow import org.apache.spark.sql.catalyst.util.DateTimeTestUtils.UTC_OPT import org.apache.spark.sql.types._ @@ -44,6 +44,14 @@ class TryCastSuite extends CastWithAnsiOnSuite { checkEvaluation(expression, null, inputRow) } + override def checkErrorInExpression[T <: SparkThrowable : ClassTag]( + expression: => Expression, + inputRow: InternalRow, + errorClass: String, + parameters: Map[String, String]): Unit = { + checkEvaluation(expression, null, inputRow) + } + override def checkCastToBooleanError(l: Literal, to: DataType, tryCastResult: Any): Unit = { checkEvaluation(cast(l, to), tryCastResult, InternalRow(l.value)) }