fix copyat_or_push! for StaticArrays

Author: ranocha

src/utils.jl

@@ -52,7 +52,9 @@ end
 
 @inline function copyat_or_push!{T,perform_copy}(a::AbstractVector{T},i::Int,x,nc::Type{Val{perform_copy}}=Val{true})
   @inbounds if length(a) >= i
-    if T <: Number || !perform_copy
+    if T <: Number || T <: StaticArray || !perform_copy
+      # TODO: Check for `setindex!`` if T <: StaticArray and use `copy!(b[i],a[i])`
+      #       or `b[i] = a[i]`, see #19
       a[i] = x
     else
       recursivecopy!(a[i],x)

test/copy_static_array_test.jl

@@ -16,6 +16,11 @@ a = [vec]
 b = zeros(a)
 recursivecopy!(b, a)
 @test a[1] == b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]
+b[1] = 2*b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]
 
 # Mutable FieldVector
 vec = MutableFV(1.,2.)
@@ -25,13 +30,23 @@ recursivecopy!(b, a)
 @test a[1] == b[1]
 a[1][1] *= 5
 @test a[1] != b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]
+b[1] = 2*b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]
 
 # SArray
 vec = @SArray [1., 2.]
 a = [vec]
 b = zeros(a)
 recursivecopy!(b, a)
 @test a[1] == b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]
+b[1] = 2*b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]
 
 # MArray
 vec = @MArray [1., 2.]
@@ -40,3 +55,8 @@ b = zeros(a)
 recursivecopy!(b, a)
 a[1][1] *= 5
 @test a[1] != b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]
+b[1] = 2*b[1]
+copyat_or_push!(a, 2, b[1])
+@test a[2] == b[1]

test/runtests.jl

@@ -7,6 +7,6 @@ tic()
 @time @testset "Partitions Tests" begin include("partitions_test.jl") end
 @time @testset "VecOfArr Indexing Tests" begin include("basic_indexing.jl") end
 @time @testset "VecOfArr Interface Tests" begin include("interface_tests.jl") end
-@time @testset "StaticArrays (recursivecopy!) Tests" begin include("copy_static_array_test.jl") end
+@time @testset "StaticArrays Tests" begin include("copy_static_array_test.jl") end
 toc()
 # Test the VectorOfArray code