我正在使用ScalaTest为Spark编写测试用例。
import org.apache.spark.sql.SparkSession
import org.scalatest.{BeforeAndAfterAll, FlatSpec}
class ClassNameSpec extends FlatSpec with BeforeAndAfterAll {
var spark: SparkSession = _
var className: ClassName = _
override def beforeAll(): Unit = {
spark = SparkSession.builder().master("local").appName("class-name-test").getOrCreate()
className = new ClassName(spark)
}
it should "return data" in {
import spark.implicits._
val result = className.getData(input)
assert(result.count() == 3)
}
override def afterAll(): Unit = {
spark.stop()
}
}
当我尝试编译测试套件时,它会出现以下错误:
stable identifier required, but ClassNameSpec.this.spark.implicits found.
[error] import spark.implicits._
[error] ^
[error] one error found
[error] (test:compileIncremental) Compilation failed
我无法理解为什么我不能在测试套件中import spark.implicits._
。
感谢任何帮助!
答案 0 :(得分:25)
要进行导入,您需要一个"稳定的标识符"正如错误消息所说。这意味着你需要一个val,而不是var。 由于您将spark定义为var,因此scala无法正确导入。
要解决此问题,您可以执行以下操作:
val spark2 = spark
import spark2.implicits._
或者将原始var更改为val,例如:
lazy val spark: SparkSession = SparkSession.builder().master("local").appName("class-name-test").getOrCreate()