Avoid ambiguity warning in getindex for LanczosJacobiBand (#15)

* Avoid ambiguity warning in getindex for LanczosJacobiBand

* Update test_lanczos.jl

Author: dlfivefifty

Project.toml

@@ -1,7 +1,7 @@
 name = "ClassicalOrthogonalPolynomials"
 uuid = "b30e2e7b-c4ee-47da-9d5f-2c5c27239acd"
 authors = ["Sheehan Olver <[email protected]>"]
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/lanczos.jl

@@ -140,8 +140,8 @@ end
 
 getindex(A::LanczosJacobiBand, I::Integer) = _lanczos_getindex(A, I)
 getindex(A::LanczosJacobiBand, I::AbstractVector) = _lanczos_getindex(A, I)
-getindex(K::LanczosJacobiBand, k::AbstractInfUnitRange) = view(K, k)
-getindex(K::SubArray{<:Any,1,<:LanczosJacobiBand}, k::AbstractInfUnitRange) = view(K, k)
+getindex(K::LanczosJacobiBand, k::AbstractInfUnitRange{<:Integer}) = view(K, k)
+getindex(K::SubArray{<:Any,1,<:LanczosJacobiBand}, k::AbstractInfUnitRange{<:Integer}) = view(K, k)
 
 copy(A::LanczosJacobiBand) = A # immutable
 

test/test_lanczos.jl

@@ -100,7 +100,7 @@ import ClassicalOrthogonalPolynomials: recurrencecoefficients, PaddedLayout
     @testset "Singularity" begin
         T = Chebyshev(); wT = WeightedChebyshev()
         x = axes(T,1)
-        
+
         w = wT * [1; zeros(∞)];
         Q = LanczosPolynomial(w)
         @test Q[0.1,1:10] ≈ Normalized(T)[0.1,1:10]
@@ -134,7 +134,7 @@ import ClassicalOrthogonalPolynomials: recurrencecoefficients, PaddedLayout
     @testset "Mixed Jacobi" begin
         P = Jacobi(1/2,0)
         x = axes(P,1)
-        
+
         w = @. sqrt(1-x)
         Q = LanczosPolynomial(w, P)
         @test Q[0.1,1:10] ≈ Normalized(P)[0.1,1:10]
@@ -199,4 +199,13 @@ import ClassicalOrthogonalPolynomials: recurrencecoefficients, PaddedLayout
         R = LanczosPolynomial((t .- x).^(-1/2) .* ChebyshevWeight())
         @test R[0.1,2] ≈ -0.03269577983003056
     end
+
+    @testset "LanczosJacobiBand" begin
+        x = Inclusion(ChebyshevInterval())
+        Q = LanczosPolynomial(  1 ./ (2 .+ x))
+        X = jacobimatrix(Q)
+        @test X.dv[3:10] ≈ [X[k,k] for k in 3:10]
+        @test X.dv[3:∞][1:5] ≈ X.dv[3:7]
+        @test X.dv[3:∞][2:∞][1:5] ≈ X.dv[4:8]
+    end
 end