improve test

apache · Nov 11, 2024 · ca06f05 · ca06f05
1 parent 9d4885d
commit ca06f05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spark/src/test/scala/org/apache/comet/CometCastSuite.scala b/spark/src/test/scala/org/apache/comet/CometCastSuite.scala
@@ -888,8 +888,8 @@ class CometCastSuite extends CometTestBase with AdaptiveSparkPlanHelper {
         makeParquetFileAllTypes(path, dictionaryEnabled = dictionaryEnabled, 10000)
         withParquetTable(path.toString, "tbl") {
           checkSparkAnswerAndOperator(
-            "SELECT CAST(struct(_1, _2) as " +
-              "struct<_1:string, _2:string>) FROM tbl")
+            "SELECT CAST(CASE WHEN _1 THEN struct(_1, _2, _3, _4) ELSE null END as " +
+              "struct<_1:string, _2:string, _3:string, _4:string>) FROM tbl")
         }
       }
     }