Use axes in making shift operator to support banded-block-banded matr… (#82)

* Use axes in making shift operator to support banded-block-banded matrices

* add test

* Update test_basisconcat.jl

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "ContinuumArrays"
 uuid = "7ae1f121-cc2c-504b-ac30-9b923412ae5c"
-version = "0.6.2"
+version = "0.6.3"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/bases/bases.jl

@@ -344,7 +344,7 @@ function arguments(::ApplyLayout{typeof(*)}, V::SubQuasiArray{<:Any,2,<:Any,<:Tu
     A = parent(V)
     _,jr = parentindices(V)
     first(jr) ≥ 1 || throw(BoundsError())
-    P = _BandedMatrix(Ones{Int}(1,length(jr)), axes(A,2), first(jr)-1,1-first(jr))
+    P = _BandedMatrix(Ones{Int}((Base.OneTo(1),axes(jr,1))), axes(A,2), first(jr)-1,1-first(jr))
     A,P
 end
 

test/test_basisconcat.jl

@@ -1,5 +1,5 @@
 using ContinuumArrays, BlockArrays, Test
-import ContinuumArrays: PiecewiseBasis, VcatBasis, HvcatBasis
+import ContinuumArrays: PiecewiseBasis, VcatBasis, HvcatBasis, arguments, ApplyLayout
 
 @testset "ConcatBasis" begin
     @testset "PiecewiseBasis" begin
@@ -39,6 +39,8 @@ import ContinuumArrays: PiecewiseBasis, VcatBasis, HvcatBasis
 
         @test size(S,2) == 5
         @test axes(S,1) == axes(S1,1) == axes(S2,1)
+        @test blockaxes(S) == (Block.(1:1), Block.(1:2))
+
         @test S == S
 
         @test S[0.1,1:5] == [vcat.(S1[0.1,:],0); vcat.(0, S2[0.1,:])]
@@ -71,5 +73,9 @@ import ContinuumArrays: PiecewiseBasis, VcatBasis, HvcatBasis
         @test S[0.1,getindex.(Block(1),1:2)] == [[S1[0.1,1] 0; 0 0], [S1[0.1,2] 0; 0 0]]
         D = Derivative(axes(S,1))
         @test_broken (D*S)[0.1,1] # throws error
+
+        v = view(S, :, Block.(2:3))
+        @test v[0.1,1] == S[0.1,3]
+        @test blockisequal(axes(arguments(ApplyLayout{typeof(*)}(), v)[2],1), axes(S,2))
     end
 end