From 7eabc41ad41e0da58fff2a6bb04212ea0cd3baa2 Mon Sep 17 00:00:00 2001
From: Carlo Zancanaro <carlo@zancanaro.id.au>
Date: Wed, 3 Jun 2015 12:39:14 +1000
Subject: Add some more interesting tests

---
 src/main/java/au/id/zancanaro/javacheck/Generator.java | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/main/java/au/id/zancanaro/javacheck/Generator.java')

diff --git a/src/main/java/au/id/zancanaro/javacheck/Generator.java b/src/main/java/au/id/zancanaro/javacheck/Generator.java
index c28d967..557cf97 100644
--- a/src/main/java/au/id/zancanaro/javacheck/Generator.java
+++ b/src/main/java/au/id/zancanaro/javacheck/Generator.java
@@ -1,5 +1,6 @@
 package au.id.zancanaro.javacheck;
 
+import java.util.Iterator;
 import java.util.List;
 import java.util.Random;
 import java.util.function.Function;
@@ -128,4 +129,8 @@ public interface Generator<T> {
     default <R> Generator<R> flatMap(Function<T, Generator<R>> action) {
         return (random, size) -> RoseTree.join(this.generate(random, size).map(action).map(g -> g.generate(random, size)));
     }
+
+    default Iterator<T> sample(Random random, int maxSize) {
+        return new GeneratorSampleIterator<>(this, random, maxSize);
+    }
 }
-- 
cgit v1.2.3