如何编写从json文件读取spark应用的单元测试

import org.apache.spark.sql.{DataFrame, SparkSession}
import org.scalatest.{BeforeAndAfterAll, FunSuite}

class JsonFileReaderTest extends FunSuite with BeforeAndAfterAll {
  private var spark: SparkSession = _

  override def beforeAll(): Unit = {
    spark = SparkSession.builder()
                        .appName("Test")
                        .master("local[*]")
                        .getOrCreate()
  }

  override def afterAll(): Unit = {
    spark.stop()
  }

  test("Read JSON file") {
    val jsonFile = "path/to/json/file.json"
    val data: DataFrame = spark.read.json(jsonFile)

    // Perform assertions on the data
    assert(data.count() == 10)
    assert(data.columns.length == 3)
    // Add more assertions as needed
  }
}