Merge pull request #145 from jonniediegelman/lazyarray-setindex

Make `setindex!` work for `LazyArray`s. Fixes #144

Author: jonniedie

Project.toml

@@ -1,7 +1,7 @@
 name = "ComponentArrays"
 uuid = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 authors = ["Jonnie Diegelman <47193959+jonniedie@users.noreply.github.com>"]
-version = "0.12.1"
+version = "0.12.2"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

src/lazyarray.jl

@@ -15,6 +15,10 @@ const LazyMatrix{T,G} = LazyArray{T,2,G}
 
 Base.getindex(a::LazyArray, i...) =  _un_iter(getfield(a, :gen), i)
 
+function Base.setindex!(a::LazyArray, val, i...)
+    a[i...] .= val
+end
+
 _un_iter(iter, idxs) = _un_iter(iter.f, iter.iter, idxs)
 _un_iter(f, iter::Base.Generator, idxs) = f(_un_iter(iter.f, iter.iter, idxs))
 _un_iter(f, iter::Base.Iterators.ProductIterator, idxs) = f(getindex.(iter.iterators, idxs))

test/runtests.jl

@@ -266,6 +266,10 @@ end
     @test tempmat["b", "b"][1, 1]["a", :a][:a, :a] == 100000
     @test tempmat[:b, :a][2].b == 1000
 
+    temp_b = deepcopy(temp.b)
+    temp.b .= temp.b .* 100
+    @test temp.b[1] == temp_b[1] .* 100
+
     temp2 = deepcopy(ca)
     temp3 = deepcopy(ca_MVector)
     @test (temp2 .= ca .* 1) isa ComponentArray