Skip to content

Commit a1338f6

Browse files
author
云峤
committed
Update python dataFrame show test and add empty df unit test.
1 parent 734369c commit a1338f6

File tree

1 file changed

+13
-11
lines changed

1 file changed

+13
-11
lines changed

sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -599,20 +599,22 @@ class DataFrameSuite extends QueryTest {
599599
}
600600

601601
test("SPARK-7319 showString") {
602-
assert(testData.select($"*").showString(1) === """+---+-----+
603-
||key|value|
604-
|+---+-----+
605-
|| 1| 1|
606-
|+---+-----+
607-
|""".stripMargin)
602+
val expectedAnswer = """+---+-----+
603+
||key|value|
604+
|+---+-----+
605+
|| 1| 1|
606+
|+---+-----+
607+
|""".stripMargin
608+
assert(testData.select($"*").showString(1) === expectedAnswer)
608609
}
609610

610611
test("SPARK-7327 show with empty dataFrame") {
611-
assert(testData.select($"*").filter($"key" < 0).showString(1) === """+---+-----+
612-
||key|value|
613-
|+---+-----+
614-
|+---+-----+
615-
|""".stripMargin)
612+
val expectedAnswer = """+---+-----+
613+
||key|value|
614+
|+---+-----+
615+
|+---+-----+
616+
|""".stripMargin
617+
assert(testData.select($"*").filter($"key" < 0).showString(1) === expectedAnswer)
616618
}
617619

618620
test("createDataFrame(RDD[Row], StructType) should convert UDTs (SPARK-6672)") {

0 commit comments

Comments
 (0)