Moving badRecords to JsonSuite

Author: MaxGekk

sql/core/src/test/scala/org/apache/spark/sql/execution/datasources/json/JsonSuite.scala

@@ -2566,6 +2566,9 @@ class JsonSuite extends QueryTest with SharedSQLContext with TestJsonData {
 
   test("return partial result for bad records") {
     val schema = "a double, b array<int>, c string, _corrupt_record string"
+    val badRecords = Seq(
+      """{"a":"-","b":[0, 1, 2],"c":"abc"}""",
+      """{"a":0.1,"b":{},"c":"def"}""").toDS()
     val df = spark.read.schema(schema).json(badRecords)
 
     checkAnswer(

sql/core/src/test/scala/org/apache/spark/sql/execution/datasources/json/TestJsonData.scala

@@ -229,11 +229,6 @@ private[json] trait TestJsonData {
       """{"date": "27/10/2014 18:30"}""" ::
       """{"date": "28/01/2016 20:00"}""" :: Nil))(Encoders.STRING)
 
-  def badRecords: Dataset[String] =
-    spark.createDataset(spark.sparkContext.parallelize(
-      """{"a":"-","b":[0, 1, 2],"c":"abc"}""" ::
-        """{"a":0.1,"b":{},"c":"def"}""" :: Nil))(Encoders.STRING)
-
   lazy val singleRow: Dataset[String] =
     spark.createDataset(spark.sparkContext.parallelize("""{"a":123}""" :: Nil))(Encoders.STRING)