Github user ala commented on a diff in the pull request:
https://github.com/apache/spark/pull/19367#discussion_r141372248
--- Diff:
sql/core/src/test/scala/org/apache/spark/sql/execution/vectorized/ColumnarBatchSuite.scala
---
@@ -829,13 +823,15 @@ class ColumnarBatchSuite extends SparkFunSuite {
assert(column.getArray(2).getArray(1).getInt(1) === 4)
assert(column.getArray(2).getArray(1).getInt(2) === 5)
assert(column.isNullAt(3))
- }
}
- test("Nest Struct in Array.") {
- (MemoryMode.ON_HEAP :: MemoryMode.OFF_HEAP :: Nil).foreach { memMode =>
- val schema = new StructType().add("int", IntegerType).add("long",
LongType)
- val column = allocate(10, new ArrayType(schema, true), memMode)
+ private val structType: StructType = new StructType().add("i",
IntegerType).add("l", LongType)
+
+ testVector(
+ "Nest Struct in Array",
+ 10,
+ new ArrayType(new StructType().add("int", IntegerType).add("long",
LongType), true),
--- End diff --
Did you mean `new ArrayType(structType)`?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
