summaryrefslogtreecommitdiff
path: root/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java')
-rw-r--r--src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java b/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
index 6e21f27..ddef800 100644
--- a/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
+++ b/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
@@ -56,17 +56,17 @@ public class ShrinkTree<T> {
public static <T> Stream<ShrinkTree<T>[]> promoteChildren(ShrinkTree<? super T>[] trees) {
return IntStream.range(0, trees.length)
.mapToObj(index -> trees[index].getChildren().map(child ->
- IntStream.range(0, trees.length)
+ (ShrinkTree<T>[]) IntStream.range(0, trees.length)
.mapToObj(i -> (i == index ? child : trees[i]))
.toArray(ShrinkTree[]::new)))
- .flatMap(x -> x)
- .map(x -> (ShrinkTree<T>[]) x);
+ .flatMap(x -> x);
}
+ @SuppressWarnings("unchecked")
public static <T> Stream<ShrinkTree<T>[]> removeChildren(ShrinkTree<? super T>[] trees) {
return IntStream.range(0, trees.length)
.mapToObj(index ->
- IntStream.range(0, trees.length)
+ (ShrinkTree<T>[]) IntStream.range(0, trees.length)
.filter(i -> i != index)
.mapToObj(i -> trees[i])
.toArray(ShrinkTree[]::new));