Fix tests

Author: mtfishman

src/fillarrays/kroneckerarray.jl

@@ -21,6 +21,8 @@ const SquareEyeKronecker{T,A<:SquareEye{T},B<:AbstractMatrix{T}} = KroneckerMatr
 const KroneckerSquareEye{T,A<:AbstractMatrix{T},B<:SquareEye{T}} = KroneckerMatrix{T,A,B}
 const SquareEyeSquareEye{T,A<:SquareEye{T},B<:SquareEye{T}} = KroneckerMatrix{T,A,B}
 
+_getindex(a::Eye, I1::Colon, I2::Colon) = a
+
 # Like `adapt` but preserves `Eye`.
 _adapt(to, a::Eye) = a
 

src/kroneckerarray.jl

@@ -182,14 +182,13 @@ function Base.getindex(a::KroneckerVector, i::Integer)
   return a.a[cld(i, k)] * a.b[(i - 1) % k + 1]
 end
 
-## function Base.getindex(a::KroneckerVector, i::CartesianProduct)
-##   return a.a[i.a] ⊗ a.b[i.b]
-## end
+# Allow customizing for `FillArrays.Eye`.
+_getindex(a::AbstractArray, I...) = a[I...]
 function Base.getindex(a::KroneckerArray{<:Any,N}, I::Vararg{CartesianProduct,N}) where {N}
-  return a.a[map(Base.Fix2(getfield, :a), I)...] ⊗ a.b[map(Base.Fix2(getfield, :b), I)...]
+  return _getindex(arg1(a), arg1.(I)...) ⊗ _getindex(arg2(a), arg2.(I)...)
 end
 function Base.getindex(a::KroneckerArray{<:Any,N}, I::Vararg{CartesianPair,N}) where {N}
-  return a.a[map(Base.Fix2(getfield, :a), I)...] ⊗ a.b[map(Base.Fix2(getfield, :b), I)...]
+  return _getindex(arg1(a), arg1.(I)...) ⊗ _getindex(arg2(a), arg2.(I)...)
 end
 # Fix ambigiuity error.
 Base.getindex(a::KroneckerArray{<:Any,0}) = a.a[] * a.b[]