more tests

Author: dlfivefifty

src/infqr.jl

@@ -7,8 +7,9 @@ end
 
 function AdaptiveQRData(::Union{SymmetricLayout{<:AbstractBandedLayout},AbstractBandedLayout}, A::AbstractMatrix{T}) where T
     l,u = bandwidths(A)
-    data = BandedMatrix{T}(undef,(2l+u+1,0),(l,l+u)) # pad super
-    AdaptiveQRData(CachedArray(data,A), Vector{T}(), 0)
+    FT = float(T)
+    data = BandedMatrix{FT}(undef,(2l+u+1,0),(l,l+u)) # pad super
+    AdaptiveQRData(CachedArray(data,A), Vector{FT}(), 0)
 end
 
 function AdaptiveQRData(::AbstractAlmostBandedLayout, A::AbstractMatrix{T}) where T

test/test_infql.jl

@@ -181,11 +181,17 @@ import BandedMatrices: _BandedMatrix
     end
 
     @testset "Tridiagonal QL" begin
-        for A in (SymTridiagonal([[1,2]; Fill(3,∞)], [[1, 2]; Fill(1,∞)]),
-                Tridiagonal([[1, 2]; Fill(1,∞)], [[1,2]; Fill(3,∞)], [[1, 2]; Fill(1,∞)]),
+        for A in (LinearAlgebra.SymTridiagonal([[1,2]; Fill(3,∞)], [[1, 2]; Fill(1,∞)]),
+                LinearAlgebra.Tridiagonal([[1, 2]; Fill(1,∞)], [[1,2]; Fill(3,∞)], [[1, 2]; Fill(1,∞)]),
                 LazyBandedMatrices.SymTridiagonal([[1,2]; Fill(3,∞)], [[1, 2]; Zeros(∞)]),
                 LazyBandedMatrices.SymTridiagonal([[1,2]; Fill(3,∞)], [[1, 2]; zeros(∞)]))
             @test abs.(ql(A).L[1:10,1:10]) ≈ abs.(ql(A[1:1000,1:1000]).L[1:10,1:10])
         end
+
+        A = LazyBandedMatrices.SymTridiagonal([[1,2]; Fill(3,∞)], [[1, 2]; Fill(1,∞)])
+        Q,L = ql(A)
+        @test (Q*L)[1:10,1:10] ≈ A[1:10,1:10]
+
+        @test (L*Q)[1:10,1:10] ≈ LazyBandedMatrices.SymTridiagonal(L*Q)[1:10,1:10]
     end
 end

test/test_infqr.jl

@@ -296,4 +296,11 @@ import SemiseparableMatrices: AlmostBandedLayout, VcatAlmostBandedLayout
         b = [1; 2; 3; zeros(∞)]
         @test (qr(A) \ b) ≈ (ul(A) \ b)
     end
+
+    @tstset "SymTridiagonal QR" begin
+        A = LazyBandedMatrices.SymTridiagonal([[1,2]; Fill(3,∞)], [[1, 2]; Fill(1,∞)])
+        Q,R = qr(A)
+        @test (Q*R)[1:10,1:10] ≈ A[1:10,1:10]
+        @test (R*Q)[1:10,1:10] ≈ LazyBandedMatrices.SymTridiagonal((R*Q)[1:10,1:10])
+    end
 end