50

我正在尝试对一些收集量很大的 Scala 进行单元测试。这些集合返回为Iterable[T]，因此我对集合的内容感兴趣，即使基础类型不同。这实际上是两个相关的问题：

  1. 如何断言两个有序集合包含相同的元素序列？
  2. 如何断言两个无序集合包含相同的元素集？

总之，我在 ScalaTest 中寻找 NUnit CollectionAssert.AreEqual（有序）和（无序）的 Scala 等效项：CollectionAssert.AreEquivalent

Set(1, 2) should equal (List(1, 2))          // ordered, pass
Iterable(2, 1) should equal (Iterable(1, 2)) // unordered, pass
4

2 回答 2

103

同时你可以使用

Iterable(2, 1) should contain theSameElementsAs Iterable(1, 2)

要测试有序集，您必须将其转换为序列。

Set(1, 2).toSeq should contain theSameElementsInOrderAs List(1, 2)
于 2014-06-16T12:28:46.810 回答
28

您可以尝试.toSeq有序集合和.toSet无序集合，据我所知，它捕获了您想要的内容。

以下通过：

class Temp extends FunSuite with ShouldMatchers {
  test("1")  { Array(1, 2).toSeq should equal (List(1, 2).toSeq) }
  test("2")  { Array(2, 1).toSeq should not equal (List(1, 2).toSeq) }
  test("2b") { Array(2, 1) should not equal (List(1, 2)) }  
  test("3")  { Iterable(2, 1).toSet should equal (Iterable(1, 2).toSet) }
  test("4")  { Iterable(2, 1) should not equal (Iterable(1, 2)) }
}

顺便说一句Set，没有订购。

编辑：为避免删除重复元素，请尝试toSeq.sorted. 以下通行证：

  test("5")  { Iterable(2, 1).toSeq.sorted should equal (Iterable(1, 2).toSeq.sorted) }
  test("6")  { Iterable(2, 1).toSeq should not equal (Iterable(1, 2).toSeq) }

编辑2：对于元素无法排序的无序集合，可以使用这个方法：

  def sameAs[A](c: Traversable[A], d: Traversable[A]): Boolean = 
    if (c.isEmpty) d.isEmpty
    else {
      val (e, f) = d span (c.head !=)
      if (f.isEmpty) false else sameAs(c.tail, e ++ f.tail)
    }

'a 'b 'c例如（注意使用没有定义顺序的符号）

  test("7")  { assert( sameAs(Iterable(2, 1),    Iterable(1, 2)     )) }
  test("8")  { assert( sameAs(Array('a, 'c, 'b), List('c, 'a, 'b)   )) }
  test("9")  { assert( sameAs("cba",             Set('a', 'b', 'c') )) }

替代sameAs实现：

  def sameAs[A](c: Traversable[A], d: Traversable[A]) = {
    def counts(e: Traversable[A]) = e groupBy identity mapValues (_.size)
    counts(c) == counts(d)
  }
于 2011-09-15T17:48:24.203 回答