Fix MTK.build_function on "arrays of sparse matrices" and "arrays of arrays of sparse matrices" (#423)

* Update build_function.jl

* update related test

Author: wupeifan

src/build_function.jl

@@ -173,10 +173,10 @@ function _build_function(target::JuliaTarget, rhss, args...;
 
 	if eltype(eltype(rhss)) <: AbstractArray # Array of arrays of arrays
 		ip_sys_exprs = reduce(vcat,[vec(reduce(vcat,[vec([:($X[$i][$j][$k] = $(conv(rhs))) for (k, rhs) ∈ enumerate(rhsel2)]) for (j, rhsel2) ∈ enumerate(rhsel)],init=Expr[])) for (i,rhsel) ∈ enumerate(_rhss)],init=Expr[])
-	elseif eltype(eltype(rhss)) <: SparseMatrixCSC # Array of arrays of arrays
-		ip_sys_exprs = reduce(vcat,[vec(reduce(vcat,[vec([:($X[$i][$j].nzval[$k] = $(conv(rhs))) for (k, rhs) ∈ enumerate(rhsel2)]) for (j, rhsel2) ∈ enumerate(rhsel)])) for (i,rhsel) ∈ enumerate(_rhss)])
+	elseif eltype(eltype(rhss)) <: SparseMatrixCSC # Array of arrays of sparse matrices
+		ip_sys_exprs = reduce(vcat,[vec(reduce(vcat,[vec([:($X[$i][$j].nzval[$k] = $(conv(rhs))) for (k, rhs) ∈ enumerate(rhsel2.nzval)]) for (j, rhsel2) ∈ enumerate(rhsel)])) for (i,rhsel) ∈ enumerate(_rhss)])
 	elseif eltype(rhss) <: SparseMatrixCSC # Array of sparse matrices
-		ip_sys_exprs = reduce(vcat,[vec([:($X[$i].nzval[$j] = $(conv(rhs))) for (j, rhs) ∈ enumerate(rhsel)]) for (i,rhsel) ∈ enumerate(_rhss)])
+		ip_sys_exprs = reduce(vcat,[vec([:($X[$i].nzval[$j] = $(conv(rhs))) for (j, rhs) ∈ enumerate(rhsel.nzval)]) for (i,rhsel) ∈ enumerate(_rhss)])
     elseif eltype(rhss) <: AbstractArray # Array of arrays
 		ip_sys_exprs = reduce(vcat,[vec([:($X[$i][$j] = $(conv(rhs))) for (j, rhs) ∈ enumerate(rhsel)]) for (i,rhsel) ∈ enumerate(_rhss)], init = Expr[])
     elseif rhss isa SparseMatrixCSC

test/build_function_arrayofarray.jl

@@ -114,8 +114,8 @@ julia_sparse_arraymat = h_sparse_arraymat_julia(input)
 mtk_sparse_arraymat = h_sparse_arraymat_oop(input)
 @test_broken julia_sparse_arraymat == mtk_sparse_arraymat
 h_sparse_arraymat_julia!(out_1_arraymat, input)
-@test_broken h_sparse_arraymat_ip!(out_2_arraymat, input)
-@test_broken out_1_arraymat == out_2_arraymat
+h_sparse_arraymat_ip!(out_2_arraymat, input)
+@test out_1_arraymat == out_2_arraymat
 
 # Array of 1D Vectors
 h_sparse_arrayvec = sparse.([[a, 0, c], [0, 0, 0], [1, a, b]])