Fix all with ArrayPartition and a functor

Author: mateuszbaran

src/array_partition.jl

@@ -226,7 +226,7 @@ Base.filter(f, A::ArrayPartition) = ArrayPartition(map(x -> filter(f, x), A.x))
 Base.any(f, A::ArrayPartition) = any((any(f, x) for x in A.x))
 Base.any(f::Function, A::ArrayPartition) = any((any(f, x) for x in A.x))
 Base.any(A::ArrayPartition) = any(identity, A)
-Base.all(f, A::ArrayPartition) = all(f, (all(f, x) for x in A.x))
+Base.all(f, A::ArrayPartition) = all((all(f, x) for x in A.x))
 Base.all(f::Function, A::ArrayPartition) = all((all(f, x) for x in A.x))
 Base.all(A::ArrayPartition) = all(identity, A)
 

test/partitions_test.jl

@@ -370,3 +370,8 @@ for i in 1:length(part_a.x)
     @test sub_a == sub_b # Test for value equality
     @test typeof(sub_a) === typeof(sub_b) # Test type equality
 end
+
+# ArrayPartition `all` with a functor
+struct TestIsnanFunctor end
+(::TestIsnanFunctor)(x) = isnan(x)
+@test all(TestIsnanFunctor(), ArrayPartition([NaN], [NaN]))