使用类型参数编写函数

Question

我有一个单元测试，它测试一些解决方案。但是这个测试代码也可以用于测试其他非常相似的解决方案。我想要的是测试代码是通用的，适用于两种解决方案，如下所示：

describe("when table contains all correct rows") {
      it("should be empty") {
        def check[T](func: T => List[Row]) = {
          val tableGen = new TableGenerator()
          val table: Vector[Row] = tableGen.randomTable(100)
            .sortWith(_.time isBefore _.time).distinct
          val result: List[Row] = func(table)
          assert(result.isEmpty)
        }

        check(Solution.solution1)
        check(Solution.solution2)
      }
    }

解决方案有类型：

solution1: IndexedSeq[Row] => List[Row]
solution2: Seq[Row] => List[Row]

必须如何编写 check() 函数才能做到这一点？ 在消除代码重复的情况下，编写这个（可能是其他方式）的最佳方法是什么？

更新： 当我尝试编译此代码时，func(table) 出现类型不匹配错误：

Error:(36, 29) type mismatch;
 found   : table.type (with underlying type scala.collection.immutable.Vector[com.vmalov.tinkoff.Row])
 required: T
          val result = func(table)

Answer 1

为此，您需要能够将Vector[Row] 传递给func，因此任何Vector[Row] 都必须是T；也就是说，T 是Vector[Row] 的超类型。您可以通过使用类型参数绑定告诉编译器：

def check[T >: Vector[Row]](func: T => List[Row])

或者，根据上述推理，函数T => List[Row] 也将是函数Vector[Row] => List[Row]，恰好当T 是Vector[Row] 的超类型时，Scala 编译器知道这一点（函数是逆变的 在他们的参数类型中）。所以这个签名等价于更简单

def check(func: Vector[Row] => List[Row])

当然，您可以概括这一点，但具体程度取决于您的具体愿望。例如。您可以将List[Row] 替换为Seq[Row]（无处不在），或者使用类型参数并将额外的函数传递给check：

def check[A](func: Vector[Row] => A)(test: A => Boolean) = {
  val table = ...
  val result = func(table)
  assert(test(result))
}

check(Solution.solution1)(_.isEmpty) // the compiler infers A is List[Row] 

Answer 2

您的情况可能足以以更具体的方式抽象类型，例如定义您期望 Travesable。

  def check[S[_] : Traversable](func: S[Row] => List[Row])

这将接受 Seq 或 IndexedSeq 作为有效参数，同时它也会限制它。

希望对你有帮助

已编辑：检查 Alexey Romanov 答案，因为这样您将无法以您的方式调用 func 。对此感到抱歉

def 检查(func: Vector[Row] => List[Row])