Skip to content
Snippets Groups Projects
Commit c9dbfcc6 authored by Liang-Chi Hsieh's avatar Liang-Chi Hsieh Committed by Reynold Xin
Browse files

[SPARK-11743][SQL] Move the test for arrayOfUDT

A following pr for #9712. Move the test for arrayOfUDT.

Author: Liang-Chi Hsieh <viirya@gmail.com>

Closes #10538 from viirya/move-udt-test.
parent 5adec63a
No related branches found
No related tags found
No related merge requests found
...@@ -108,7 +108,8 @@ class RowEncoderSuite extends SparkFunSuite { ...@@ -108,7 +108,8 @@ class RowEncoderSuite extends SparkFunSuite {
.add("arrayOfArrayOfString", ArrayType(arrayOfString)) .add("arrayOfArrayOfString", ArrayType(arrayOfString))
.add("arrayOfArrayOfInt", ArrayType(ArrayType(IntegerType))) .add("arrayOfArrayOfInt", ArrayType(ArrayType(IntegerType)))
.add("arrayOfMap", ArrayType(mapOfString)) .add("arrayOfMap", ArrayType(mapOfString))
.add("arrayOfStruct", ArrayType(structOfString))) .add("arrayOfStruct", ArrayType(structOfString))
.add("arrayOfUDT", arrayOfUDT))
encodeDecodeTest( encodeDecodeTest(
new StructType() new StructType()
...@@ -130,18 +131,6 @@ class RowEncoderSuite extends SparkFunSuite { ...@@ -130,18 +131,6 @@ class RowEncoderSuite extends SparkFunSuite {
new StructType().add("array", arrayOfString).add("map", mapOfString)) new StructType().add("array", arrayOfString).add("map", mapOfString))
.add("structOfUDT", structOfUDT)) .add("structOfUDT", structOfUDT))
test(s"encode/decode: arrayOfUDT") {
val schema = new StructType()
.add("arrayOfUDT", arrayOfUDT)
val encoder = RowEncoder(schema)
val input: Row = Row(Seq(new ExamplePoint(0.1, 0.2), new ExamplePoint(0.3, 0.4)))
val row = encoder.toRow(input)
val convertedBack = encoder.fromRow(row)
assert(input.getSeq[ExamplePoint](0) == convertedBack.getSeq[ExamplePoint](0))
}
test(s"encode/decode: Product") { test(s"encode/decode: Product") {
val schema = new StructType() val schema = new StructType()
.add("structAsProduct", .add("structAsProduct",
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment