在ScalaCheck中生成选项[T]

Question

我试图在ScalaCheck中生成可选参数,但没有成功.

似乎没有直接机制.Gen.containerOf[Option, Thing](thingGenerator)失败,因为它无法找到隐含的Buildable[Thing, Option].

我试过了

for {
  thing <- Gen.listOfN[Thing](1, thingGenerator)
} yield thing.headOption

但这不起作用,因为listOfN产生一个总长度为N的列表.结果我总是得到一个Some[Thing].类似地,listOf1不起作用,因为(a)它不产生空列表,而且(b)它效率低,因为我不能设置元素数量的最大限制.

我怎样才能生成Option[Thing]包含Nones的内容？

编辑:我找到了解决方案,但它并不简洁.有比这更好的方法吗？

for {
  thing <- for {
    qty <- Gen.choose(0,1)
    things <- Gen.listOfN[Thing](qty, thingGenerator)
  } yield things.headOption
} yield thing

编辑2:我将此概括为

def optional[T](g: Gen[T]) = 
  for (qty <- Gen.choose(0, 1); xs <- Gen.listOfN[T](qty, g)) yield xs.headOption

所以我不必多次写它.但肯定这已经在图书馆了,我只是错过了它？

Answer 1

现在你可以使用:

Gen.option(yourGen)

Answer 2

您可以使用pick在Some和None生成器之间随机选择:

val someThing = thingGenerator.map( Some.apply )
val noThing = Gen.value( None:Option[Thing] )
val optThing = Gen.oneOf( someThing, noThing )