More ∞ Toep mul support (#50)

* More ∞ Toep mul support

* update tests

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "InfiniteLinearAlgebra"
 uuid = "cde9dba0-b1de-11e9-2c62-0bab9446c55c"
-version = "0.4.3"
+version = "0.4.4"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/banded/infbanded.jl

@@ -401,7 +401,7 @@ similar(M::MulAdd{<:AbstractBandedLayout,<:AbstractBandedLayout}, ::Type{T}, axe
 # BandedFill * BandedFill
 ###
 
-copy(M::MulAdd{BandedColumns{FillLayout},BandedColumns{FillLayout},ZerosLayout}) =
+copy(M::MulAdd{<:BandedColumns{<:AbstractFillLayout},<:BandedColumns{<:AbstractFillLayout},ZerosLayout}) =
     _bandedfill_mul(M, axes(M.A), axes(M.B))
 
 _bandedfill_mul(M, _, _) = copyto!(similar(M), M)
@@ -417,6 +417,13 @@ function _bandedfill_mul(M::MulAdd, ::Tuple{InfAxes,InfAxes}, ::Tuple{InfAxes,In
     ret
 end
 
+_bandedfill_mul(M::MulAdd, ::Tuple{InfAxes,Any}, ::Tuple{Any,Any}) = ApplyArray(*, M.A, M.B)
+_bandedfill_mul(M::MulAdd, ::Tuple{InfAxes,Any}, ::Tuple{Any,InfAxes}) = ApplyArray(*, M.A, M.B)
+_bandedfill_mul(M::MulAdd, ::Tuple{Any,Any}, ::Tuple{Any,InfAxes}) = ApplyArray(*, M.A, M.B)
+_bandedfill_mul(M::MulAdd, ::Tuple{Any,InfAxes}, ::Tuple{InfAxes,InfAxes}) = ApplyArray(*, M.A, M.B)
+_bandedfill_mul(M::MulAdd, ::Tuple{InfAxes,InfAxes}, ::Tuple{InfAxes,Any}) = ApplyArray(*, M.A, M.B)
+_bandedfill_mul(M::MulAdd, ::Tuple{Any,InfAxes}, ::Tuple{InfAxes,Any}) = ApplyArray(*, M.A, M.B)
+
 mulreduce(M::Mul{<:BandedColumns{<:AbstractFillLayout}, <:PertToeplitzLayout}) = ApplyArray(M)
 mulreduce(M::Mul{<:PertToeplitzLayout, <:BandedColumns{<:AbstractFillLayout}}) = ApplyArray(M)
 mulreduce(M::Mul{<:BandedColumns{<:AbstractFillLayout}, <:BandedToeplitzLayout}) = ApplyArray(M)

test/runtests.jl

@@ -161,6 +161,20 @@ end
         @test A .* 2.0 isa BandedMatrix{Float64,<:Fill}
         @test Eye(∞)*A isa BandedMatrix{Float64,<:Ones}
         @test A*Eye(∞) isa BandedMatrix{Float64,<:Ones}
+
+        @test A*A isa BandedMatrix
+        @test (A*A)[1:10,1:10] == BandedMatrix(2 => Ones(8))
+
+        Ã = _BandedMatrix(Fill(1,1,∞), ∞, -1,1)
+        @test A*Ã isa BandedMatrix
+        @test Ã*A isa BandedMatrix
+        @test Ã*Ã isa BandedMatrix
+
+        B = _BandedMatrix(Ones(1,10),∞,-1,1)
+        C = _BandedMatrix(Ones(1,10),10,-1,1)
+        D = _BandedMatrix(Ones(1,∞),10,-1,1)
+
+        @test (A*B)[1:10,1:10] == (B*C)[1:10,1:10] == (D*A)[1:10,1:10] == D*B == (C*D)[1:10,1:10] == BandedMatrix(2 => Ones(8))
     end
 
     @testset "Banded Broadcast" begin