Make setindex! work for LazyArrays. Fixes #144

Author: jonniediegelman

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