Fix fallback test

In one case there is a zero in index and test fails due to spark error
apache · Nov 19, 2024 · 4770fce · 4770fce
1 parent 659ab7a
commit 4770fce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spark/src/test/scala/org/apache/comet/CometExpressionSuite.scala b/spark/src/test/scala/org/apache/comet/CometExpressionSuite.scala
@@ -2385,7 +2385,7 @@ class CometExpressionSuite extends CometTestBase with AdaptiveSparkPlanHelper {
       val df = spark.read
         .parquet(path.toString)
         .withColumn("arr", array(col("_4"), lit(null), col("_4")))
-        .withColumn("idx", udf((x: Int) => x).apply(col("_4")))
+        .withColumn("idx", udf((_: Int) => 1).apply(col("_4")))
         .withColumn("arrUnsupportedArgs", expr("array_insert(arr, idx, 1)"))
       checkSparkAnswer(df.select("arrUnsupportedArgs"))
     }