add filter method for ArrayPartition

Author: aplavin

src/array_partition.jl

@@ -114,6 +114,7 @@ Base.:(==)(A::ArrayPartition,B::ArrayPartition) = A.x == B.x
 
 Base.map(f,A::ArrayPartition) = ArrayPartition(map(x->map(f,x), A.x))
 Base.mapreduce(f,op,A::ArrayPartition) = mapreduce(f,op,(mapreduce(f,op,x) for x in A.x))
+Base.filter(f,A::ArrayPartition) = ArrayPartition(map(x->filter(f,x), A.x))
 Base.any(f,A::ArrayPartition) = any(f,(any(f,x) for x in A.x))
 Base.any(f::Function,A::ArrayPartition) = any(f,(any(f,x) for x in A.x))
 Base.any(A::ArrayPartition) = any(identity, A)

test/partitions_test.jl

@@ -106,6 +106,9 @@ _broadcast_wrapper(y) = _scalar_op.(y)
 # Testing map
 @test map(x->x^2, x) == ArrayPartition(x.x[1].^2, x.x[2].^2)
 
+# Testing filter
+@test filter(x->iseven(round(Int, x)), x) == ArrayPartition([2], [4.0])
+
 #### testing copyto!
 S = [
      ((1,),(2,)) => ((1,),(2,)),