生成具有 int 字段且不重复的案例类列表答案

【问题标题】：generate list of case class with int field without repeat生成具有 int 字段且不重复的案例类列表
【发布时间】：2021-01-22 12:31:10
【问题描述】：

我想生成一个包含多个字段的类的列表。其中之一是 Int 类型，不必重复。你能帮我写代码吗？

接下来我尝试了：

  case class Person(name: String, age: Int)

  implicit val genPerson: Gen[Person] = 
    for {
    name <- arbitrary[String]
    age <- Gen.posNum[Int]
    } yield Person(name, age)

  implicit val genListOfPerson: Gen[scala.List[Person]] = Gen.listOfN(3, genPerson)

问题是我得到了一个年龄相同的人的实例。

【问题讨论】：

当您说“不必重复”时，您的意思是“没有两个生成的Persons 的年龄相同”？

标签： scala unit-testing scalacheck

【解决方案1】：

如果您要求生成的列表中没有两个Persons 具有相同的年龄，您可以

implicit def IntsArb: Arbitrary[Int] = Arbitrary(Gen.choose[Int](0, Int.MaxValue))
implicit val StringArb: Arbitrary[String] = Arbitrary(Gen.listOfN(5, Gen.alphaChar).map(_.mkString))
implicit val PersonGen = Arbitrary(Gen.resultOf(Person.apply _))
implicit val PersonsGen: Arbitrary[List[Person]] =
  Arbitrary(Gen.listOfN(3, PersonGen.arbitrary).map { persons =>
    val grouped: Map[Int, List[Person]] = persons.groupBy(_.age)
    grouped.values.map(_.head) // safe because groupBy
  })

请注意，这将返回一个没有重复年龄的List，但不能保证列表的大小为 3（保证列表不为空，大小最多为 3）。

如果有一个大小为 3 的列表很重要，如果“骰子对你不利”，则冒着生成失败的风险，你可以有类似的东西：

def uniqueAges(persons: List[Person], target: Int): Gen[List[Person]] = {
  val grouped: Map[Int, List[Person]] = persons.groupBy(_.age)
  val uniquelyAged = grouped.values.map(_.head)
  val n = uniquelyAged.size
  if (n == target) Gen.const(uniquelyAged)
  else {
    val existingAges = grouped.keySet
    val genPerson = PersonGen.arbitrary.retryUntil { p => !existingAges(p.age) }
    Gen.listOf(target - n, genPerson)
      .flatMap(l => uniqueAges(l, target - n))
      .map(_ ++ uniquelyAged)
  }
}

implicit val PersonsGen: Arbitrary[List[Person]] =
  Arbitrary(Gen.listOfN(3, PersonGen.arbitrary).flatMap(l => uniqueAges(l, 3)))

【讨论】：

【解决方案2】：

你可以这样做：

implicit def IntsArb: Arbitrary[Int] = Arbitrary(Gen.choose[Int](0, Int.MaxValue))
implicit val StringArb: Arbitrary[String] = Arbitrary(Gen.listOfN(5, Gen.alphaChar).map(_.mkString))
implicit val PersonGen = Arbitrary(Gen.resultOf(Person.apply _))
implicit val PersonsGen: Arbitrary[List[Person]] = Arbitrary(Gen.listOfN(3, PersonGen.arbitrary))

【讨论】：