Merge pull request #355 from lxvm/patch-1

Update vector_of_array.jl

Author: ChrisRackauckas

src/vector_of_array.jl

@@ -304,6 +304,11 @@ Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
     return Adapt.adapt(__parameterless_type(T),
         reshape(reduce(hcat, vecs), size(A.u[1])..., length(A.u)))
 end
+Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
+    ::NotSymbolic, I::Colon...) where {T <: Number, N}
+    @assert length(I) == ndims(A.u)
+    return A.u[I...]
+end
 
 Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
         ::NotSymbolic, I::AbstractArray{Bool},

test/basic_indexing.jl

@@ -257,3 +257,6 @@ foo!(u_vector)
 # test efficiency 
 num_allocs = @allocations foo!(u_matrix)
 @test num_allocs == 0
+
+# issue 354
+@test VectorOfArray(ones(1))[:] == ones(1)