Add positive range generator

chipsalliance · Jul 15, 2020 · 8fab4d6 · 8fab4d6
1 parent 9dda7f5
commit 8fab4d6
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/test/scala/chiselTests/ChiselSpec.scala b/src/test/scala/chiselTests/ChiselSpec.scala
@@ -147,6 +147,20 @@ class ChiselPropSpec extends PropSpec with ChiselRunners with ScalaCheckProperty
   // Generator for small positive integers.
   val smallPosInts = Gen.choose(1, 4)
 
+  // Generator for positive (ascending or descending) ranges.
+  def posRange: Gen[Range] = for {
+    dir <- Gen.oneOf(true, false)
+    step <- Gen.choose(1, 3)
+    m <- Gen.choose(1, 10)
+    n <- Gen.choose(1, 10)
+  } yield {
+    if (dir) {
+      Range(m, (m+n)*step, step)
+    } else {
+      Range((m+n)*step, m, -step)
+    }
+  }
+
   // Generator for widths considered "safe".
   val safeUIntWidth = Gen.choose(1, 30)