From 7db38e2e4f2c24c2a21d1c19c12bfdad4727ebad Mon Sep 17 00:00:00 2001
From: Carlo Zancanaro <carlo@zancanaro.id.au>
Date: Mon, 1 Jun 2015 16:41:07 +1000
Subject: Make listOf able to take a min/max length of list
---
src/main/java/au/id/zancanaro/javacheck/Generators.java | 13 +++++++++++++
1 file changed, 13 insertions(+)
(limited to 'src/main/java/au/id/zancanaro/javacheck')
diff --git a/src/main/java/au/id/zancanaro/javacheck/Generators.java b/src/main/java/au/id/zancanaro/javacheck/Generators.java
index bf1b1b3..eab3eed 100644
--- a/src/main/java/au/id/zancanaro/javacheck/Generators.java
+++ b/src/main/java/au/id/zancanaro/javacheck/Generators.java
@@ -80,6 +80,19 @@ public final class Generators {
};
}
+ public static <T> Generator<List<T>> listOf(Generator<T> gen, int minElements, int maxElements) {
+ return (random, size) -> {
+ Generator<Integer> countGen = sized(s -> integer(minElements, maxElements));
+ int count = countGen.generate(random, size).getValue();
+ return Generator.list(count, gen).generate(random, size)
+ .filter(list ->
+ minElements <= list.size()
+ && list.size() < maxElements);
+ };
+ }
+
+ ;
+
public static <T> Generator<List<T>> listOf(Generator<T> gen) {
return (random, size) -> {
Generator<Integer> countGen = sized(s -> integer(0, s));
--
cgit v1.2.3