Update python dataFrame show test and add empty df unit test.

nemccarthy · May 4, 2015 · a1338f6 · a1338f6
1 parent 734369c
commit a1338f6
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala
@@ -599,20 +599,22 @@ class DataFrameSuite extends QueryTest {
   }
 
   test("SPARK-7319 showString") {
-    assert(testData.select($"*").showString(1) === """+---+-----+
-                                                     ||key|value|
-                                                     |+---+-----+
-                                                     ||  1|    1|
-                                                     |+---+-----+
-                                                     |""".stripMargin)
+    val expectedAnswer = """+---+-----+
+                           ||key|value|
+                           |+---+-----+
+                           ||  1|    1|
+                           |+---+-----+
+                           |""".stripMargin
+    assert(testData.select($"*").showString(1) === expectedAnswer)
   }
 
   test("SPARK-7327 show with empty dataFrame") {
-    assert(testData.select($"*").filter($"key" < 0).showString(1) === """+---+-----+
-                                                                        ||key|value|
-                                                                        |+---+-----+
-                                                                        |+---+-----+
-                                                                        |""".stripMargin)
+    val expectedAnswer = """+---+-----+
+                           ||key|value|
+                           |+---+-----+
+                           |+---+-----+
+                           |""".stripMargin
+    assert(testData.select($"*").filter($"key" < 0).showString(1) === expectedAnswer)
   }
 
   test("createDataFrame(RDD[Row], StructType) should convert UDTs (SPARK-6672)") {