Add @DataSource, allow printing of RoseTrees, other small changes to generators

1 files changed, 8 insertions, 4 deletions

diff --git a/src/main/java/au/id/zancanaro/PropertyError.java b/src/main/java/au/id/zancanaro/PropertyError.java
index b2fb589..a0ee6df 100644
--- a/src/main/java/au/id/zancanaro/PropertyError.java
+++ b/src/main/java/au/id/zancanaro/PropertyError.java
@@ -6,10 +6,14 @@ import java.util.Iterator;
 
 public class PropertyError extends AssertionError {
     public PropertyError(String methodName, long seed, ShrinkResult shrunk) {
-        super(String.format("%s(%s)\n\tSeed: %s\n%s",
-                methodName, join(", ", shrunk.args),
-                seed,
-                shrunk.thrown.getMessage()));
+        super(shrunk.thrown.getMessage() == null ?
+                String.format("%s(%s)\n\tSeed: %s",
+                        methodName, join(", ", shrunk.args),
+                        seed):
+                String.format("%s(%s)\n\tSeed: %s\n%s",
+                        methodName, join(", ", shrunk.args),
+                        seed,
+                        shrunk.thrown.getMessage()));
         initCause(shrunk.thrown);
     }