Merge pull request #127 from SciML/map

add a map overload

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "RecursiveArrayTools"
 uuid = "731186ca-8d62-57ce-b412-fbd966d074cd"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "2.8.2"
+version = "2.9.0"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
@@ -16,7 +16,7 @@ ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
 ArrayInterface = "2.7"
 RecipesBase = "0.7, 0.8, 1.0"
 Requires = "0.5, 1.0"
-StaticArrays = "1.0"
+StaticArrays = "0.12, 1.0"
 ZygoteRules = "0.2"
 julia = "1.3"
 

src/vector_of_array.jl

@@ -151,6 +151,7 @@ end
   VA.t,VA.u
 end
 
+Base.map(f,A::RecursiveArrayTools.AbstractVectorOfArray) = map(f,A.u)
 Base.mapreduce(f,op,A::AbstractVectorOfArray) = mapreduce(f,op,(mapreduce(f,op,x) for x in A.u))
 
 ## broadcasting

test/interface_tests.jl

@@ -74,3 +74,6 @@ emptyva = VectorOfArray(Array{Vector{Float64}}([]))
 @test isempty(emptyva)
 emptyda = DiffEqArray(Array{Vector{Float64}}([]), Vector{Float64}())
 @test isempty(emptyda)
+
+A = VectorOfArray(map(i->rand(2,4),1:7))
+@test map(x->maximum(x),A) isa Vector