如何在Databricks中运行Spark-Scala单元测试笔记本？

// 导入所需的库
import org.apache.spark.sql.functions._
import org.scalatest.FunSuite

// 定义测试套件
class MyTestSuite extends FunSuite {
  // 创建本地Spark上下文
  val spark = SparkSession.builder()
    .appName("Test")
    .master("local")
    .getOrCreate()
    
  // 编写测试用例
  test("测试示例") {
    import spark.implicits._
    
    // 创建测试数据
    val data = Seq(("Alice", 25), ("Bob", 30), ("Charlie", 35))
    val df = data.toDF("Name", "Age")
    
    // 执行测试操作
    val result = df.select($"Name", $"Age" + 1).collect()
    
    // 验证测试结果
    assert(result.length == 3)
    assert(result(0)(1) == 26)
    assert(result(1)(1) == 31)
    assert(result(2)(1) == 36)
  }
}

// 运行测试套件
val suite = new MyTestSuite
suite.run()