use withTempTable

apache · Feb 1, 2016 · 898cf20 · 898cf20
1 parent 21f29a5
commit 898cf20
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/SQLQuerySuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/SQLQuerySuite.scala
@@ -2046,6 +2046,15 @@ class SQLQuerySuite extends QueryTest with SharedSQLContext {
     )
   }
 
+  test("SPARK-13056: Null in map value causes NPE") {
+    val df = Seq(1 -> Map("abc" -> "somestring", "cba" -> null)).toDF("key", "value")
+    withTempTable("maptest") {
+      df.registerTempTable("maptest")
+      checkAnswer(sql("SELECT value['abc'] FROM maptest"), Row("somestring"))
+      checkAnswer(sql("SELECT value['cba'] FROM maptest"), Row(null))
+    }
+  }
+
   test("hash function") {
     val df = Seq(1 -> "a", 2 -> "b").toDF("i", "j")
     withTempTable("tbl") {
@@ -2056,11 +2065,4 @@ class SQLQuerySuite extends QueryTest with SharedSQLContext {
       )
     }
   }
-
-  test("SPARK-13056: Null in map value causes NPE") {
-    val df = Seq(1 -> Map("abc" -> "somestring", "cba" -> null)).toDF("key", "value")
-    df.registerTempTable("maptest")
-    checkAnswer(sql("SELECT value['abc'] FROM maptest"), Row("somestring"))
-    checkAnswer(sql("SELECT value['cba'] FROM maptest"), Row(null))
-  }
 }