Skip to content

Commit d5ecc47

Browse files
committed
fix: include id column in parquet for size map v2 reader test
1 parent 90d59ca commit d5ecc47

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

spark/src/test/scala/org/apache/comet/CometMapExpressionSuite.scala

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,6 @@ class CometMapExpressionSuite extends CometTestBase {
133133
makeParquetFileAllPrimitiveTypes(path, dictionaryEnabled = true, 100)
134134
spark.read.parquet(path.toString).createOrReplaceTempView("t1")
135135

136-
checkSparkAnswerAndOperator(
137-
sql("SELECT size(map(_8, _9, _10, _11)) from t1 where _8 is not null"))
138136
checkSparkAnswerAndOperator(
139137
sql("SELECT size(case when _2 < 0 then map(_8, _9) else map() end) from t1"))
140138
}
@@ -147,6 +145,7 @@ class CometMapExpressionSuite extends CometTestBase {
147145
val df = spark
148146
.range(100)
149147
.select(
148+
col("id"),
150149
when(col("id") > 1, map(col("id"), when(col("id") > 2, col("id"))))
151150
.alias("map1"),
152151
when(col("id") > 5, map(lit("a"), col("id"), lit("b"), col("id") + 1))

0 commit comments

Comments
 (0)