companion类使用来自companion对象的变量,如何用scalatest测试companion类中的方法

wgeznvg7  于 2021-05-27  发布在  Spark
关注(0)|答案(0)|浏览(277)

我有一个如下的实现,在这个实现中,我在companion对象中调用了companion类中的一些方法。companion类中的方法使用来自companion对象的一些私有变量。

class Foo{

  def loadFinalDF(df:DataFrame): DataFrame ={
    val finaldf: DataFrame = df
      .withColumn("Date", lit(date)).distinct()
    finaldf
  }
}

object Foo {

  import org.apache.spark.sql.SparkSession
  val spark = SparkSession.builder().getOrCreate()
  import spark.implicits._

  val someDF = Seq(
          (8, "bat"),
          (64, "mouse"),
          (-27, "horse")
          ).toDF("number", "word")

  val obj = new Foo
  private val date = "2019-03-01"
  val finalDF = obj.loadFinalDF(someDF)

}

我想为loadfinaldf方法编写测试用例,但是我不确定如何使用在伴随对象中定义的date变量来完成。我猜我可能不得不使用某种模拟测试,但我从来没有做过模拟测试之前和网上资源是非常混乱的理解。如何为这个方法编写测试用例?请帮帮我!谢谢您!

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题