Move packages, make lists shrink in size, generate lists instead of arrays as the 'primitive' operation (issues with generics)

1 files changed, 36 insertions, 0 deletions

diff --git a/src/main/java/au/id/zancanaro/javacheck/Generator.java b/src/main/java/au/id/zancanaro/javacheck/Generator.java
new file mode 100644
index 0000000..66ebf1b
--- /dev/null
+++ b/src/main/java/au/id/zancanaro/javacheck/Generator.java
@@ -0,0 +1,36 @@
+package au.id.zancanaro.javacheck;
+
+import java.util.List;
+import java.util.Random;
+import java.util.function.Function;
+
+public interface Generator<T> {
+    RoseTree<T> generate(Random random, int size);
+
+    static <T> Generator<T> pure(T value) {
+        return (random, size) -> RoseTree.pure(value);
+    }
+
+    @SafeVarargs
+    static <T> Generator<List<T>> tuple(Generator<T>... generators) {
+        return (random, size) -> {
+            @SuppressWarnings("unchecked")
+            RoseTree<T>[] result = (RoseTree<T>[]) new RoseTree[generators.length];
+            int index = 0;
+            for (Generator<T> generator : generators) {
+                result[index++] = generator.generate(random, size);
+            }
+            return RoseTree.zip(Function.identity(), result);
+        };
+    }
+
+    default <R> Generator<R> fmap(Function<T, R> f) {
+        return (random, size) -> this.generate(random, size).fmap(f);
+    }
+
+    default <R> Generator<R> flatMap(Function<T, Generator<R>> action) {
+        return (random, size) -> {
+            return RoseTree.join(this.generate(random, size).fmap(action).fmap(g -> g.generate(random, size)));
+        };
+    }
+}