Scala-单元测试

我想为这个函数编写单元测试，有人能帮我吗

    def train(sc: SparkContext, data: PreparedData): TopViewModel = {

    val itemStringIntMap = BiMap.stringInt(data.itemsView.keys)

    // collect Item as Map and convert ID to Int index
    val items: Map[Int, Item] = data.itemsView.map { 
        case (id, item) =>
        (itemStringIntMap(id), item)
    }.collectAsMap.toMap

    val topView = trainTopView(
        events = data.viewEvents,
        n = ap.n,
        itemStringIntMap = itemStringIntMap
    )
    logger.info(topView)

    new TopViewModel(
        topView = topView,
        itemStringIntMap = itemStringIntMap,
        items = items
    )
}

---

非常感谢。

您需要选择一个测试框架，如ScalaTest或Specs2。看看这里的例子。为什么你把sc作为参数传递，你甚至不用它。。。