diff --git a/slowCheck/src/slowCheck/Generator.java b/slowCheck/src/slowCheck/Generator.java index decec98e0e95..48671d7808f2 100644 --- a/slowCheck/src/slowCheck/Generator.java +++ b/slowCheck/src/slowCheck/Generator.java @@ -39,7 +39,12 @@ public final class Generator { } public Generator flatMap(@NotNull Function> fun) { - return from(data -> fun.apply(generateValue(data)).generateValue(data)); + return from(data -> { + T value = generateValue(data); + Generator result = fun.apply(value); + if (result == null) throw new NullPointerException(fun + " returned null on " + value); + return result.generateValue(data); + }); } public Generator noShrink() {