如何使用ScalaCheck生成混洗序列?

时间:2015-06-11 09:58:52

标签: scala scalacheck

我一直试图用scalacheck生成改组序列。 Scalacheck并没有直接提供任何发电机,我也无法在网上找到任何简单的答案。经过一番思考,下面就是我做的。我希望其他人会觉得有用。

3 个答案:

答案 0 :(得分:2)

org.scalacheck.Gen.pick(n: Int, l: Iterable[T]): Gen[Seq[T]]以随机顺序从n中选择l个不同的元素。使用pick致电n = l.length,您将随机生成

答案 1 :(得分:0)

def shuffle[T](s: Seq[T], prefix: Seq[T] = Vector()): Gen[Seq[T]] =
    if (s.length == 0) Gen.const(prefix.toSeq)
    else
      Gen.choose(0, s.length - 1)
        .flatMap { i => 
          shuffle(
            s.take(i) ++ s.takeRight(s.length - i - 1), 
            prefix :+ s(i)) }

答案 2 :(得分:0)

您可以使用集合混洗算法(例如scala.util.Random中的算法)来获得更清晰的解决方案:

/**
 * This represents a potential list shuffling. It
 * acts as a pure function - when applied to the
 * same list it always returns the same result
 */
class Shuffling(seed: Int) {
  def apply[T](xs: Seq[T]): Seq[T] = {
    val r = new scala.util.Random(seed)
    r.shuffle(xs)
  }
}

import org.scalacheck.Arbitrary._
import org.scalacheck.Gen

val shufflingGen: Gen[Shuffling] = arbitrary[Int].map(new Shuffling(_))

上面的代码段定义了shuffling - 列表的可能重新排序。它还定义了一个提供任意混洗实例的生成器。以下示例显示了如何使用shuffling重新排序整数范围:

def shuffledRange(n: Int): Gen[Seq[Int]] = {
  for {
    shuffling <- shufflingGen
  } yield shuffling(Range(0, n))
}

Shuffling类的形式添加一个间接级别似乎很奇怪 - 我们也可以直接在列表中应用Random.shuffle。这是因为除了Scalacheck已经在后台进行的隐式随机化之外,我们还想避免额外的随机源 - 混洗中随机性的来源来自scalacheck生成器。

这样做的一个好处是生成案例的可重复性(每当Scalacheck will support it时)。