Update DataFrameSuite.scala

Incorporated review comments, Make the code more concise
apache · Feb 8, 2016 · d576496 · d576496
1 parent a9e3b32
commit d576496
Showing 1 changed file with 1 addition and 4 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
@@ -1293,10 +1293,7 @@ class DataFrameSuite extends QueryTest with SharedSQLContext {
   }
 
   test("SPARK-12982: Add table name validation in temp table registration") {
-    val rows = List(Row("foo"), Row("bar"))
-    val schema = StructType(Seq(StructField("col", StringType)))
-    val rdd = sparkContext.parallelize(rows)
-    val df = sqlContext.createDataFrame(rdd, schema)
+    val df = Seq("foo", "bar").map(Tuple1.apply).toDF("col")
     //invalid table name test as below
     intercept[AnalysisException](df.registerTempTable("t~"))
     //valid table name test as below