Use Testsuite for Schur

Author: kshyatt

test/genericschur/eig.jl

@@ -22,9 +22,6 @@ if !is_buildkite
     @safetestset "Generalized Eigenvalue Decomposition" begin
         include("gen_eig.jl")
     end
-    @safetestset "Schur Decomposition" begin
-        include("schur.jl")
-    end
     @safetestset "Image and Null Space" begin
         include("orthnull.jl")
     end
@@ -55,10 +52,6 @@ if !is_buildkite
         include("genericlinearalgebra/eigh.jl")
     end
 
-    using GenericSchur
-    @safetestset "General Eigenvalue Decomposition" begin
-        include("genericschur/eig.jl")
-    end
 end
 
 @safetestset "QR / LQ Decomposition" begin
@@ -71,6 +64,9 @@ end
 @safetestset "Projections" begin
     include("projections.jl")
 end
+@safetestset "Schur Decomposition" begin
+    include("schur.jl")
+end
 
 using CUDA
 if CUDA.functional()

test/runtests.jl

@@ -2,30 +2,32 @@ using MatrixAlgebraKit
 using Test
 using TestExtras
 using StableRNGs
-using LinearAlgebra: I
+using LinearAlgebra: I, Diagonal
 
-@testset "schur_full! for T = $T" for T in (Float32, Float64, ComplexF32, ComplexF64)
-    rng = StableRNG(123)
-    m = 54
-    for alg in (LAPACK_Simple(), LAPACK_Expert())
-        A = randn(rng, T, m, m)
-        Tc = complex(T)
+BLASFloats = (Float32, Float64, ComplexF32, ComplexF64)
+GenericFloats = (Float16, BigFloat, Complex{BigFloat})
 
-        TA, Z, vals = @constinferred schur_full(A; alg)
-        @test eltype(TA) == eltype(Z) == T
-        @test eltype(vals) == Tc
-        @test isisometric(Z)
-        @test A * Z ≈ Z * TA
+@isdefined(TestSuite) || include("testsuite/TestSuite.jl")
+using .TestSuite
 
-        Ac = similar(A)
-        TA2, Z2, vals2 = @constinferred schur_full!(copy!(Ac, A), (TA, Z, vals), alg)
-        @test TA2 === TA
-        @test Z2 === Z
-        @test vals2 === vals
-        @test A * Z ≈ Z * TA
+is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
 
-        valsc = @constinferred schur_vals(A, alg)
-        @test eltype(valsc) == Tc
-        @test valsc ≈ eig_vals(A, alg)
+m = 54
+for T in (BLASFloats..., GenericFloats...)
+    TestSuite.seed_rng!(123)
+    if T ∈ BLASFloats
+        #=if CUDA.functional()
+            TestSuite.test_schur(CuMatrix{T}, (m, m); test_blocksize = false)
+            TestSuite.test_schur(Diagonal{T, CuVector{T}}, m; test_blocksize = false)
+        end
+        if AMDGPU.functional()
+            TestSuite.test_schur(ROCMatrix{T}, (m, m); test_blocksize = false)
+            TestSuite.test_schur(Diagonal{T, ROCVector{T}}, m; test_blocksize = false)
+        end=# # not yet supported
+    end
+    if !is_buildkite
+        TestSuite.test_schur(T, (m, m))
+        AT = Diagonal{T, Vector{T}}
+        TestSuite.test_schur(AT, m)
     end
 end

test/schur.jl

@@ -73,5 +73,6 @@ include("qr.jl")
 include("lq.jl")
 include("polar.jl")
 include("projections.jl")
+include("schur.jl")
 
 end

test/testsuite/TestSuite.jl

@@ -0,0 +1,38 @@
+using TestExtras
+using GenericSchur
+
+function test_schur(T::Type, sz; kwargs...)
+    summary_str = testargs_summary(T, sz)
+    return @testset "schur $summary_str" begin
+        test_schur_full(T, sz; kwargs...)
+    end
+end
+
+function test_schur_full(
+        T::Type, sz;
+        atol::Real = 0, rtol::Real = precision(T),
+        kwargs...
+    )
+    summary_str = testargs_summary(T, sz)
+    return @testset "eig_full! $summary_str" begin
+        A = instantiate_matrix(T, sz)
+        Ac = deepcopy(A)
+        Tc = isa(A, Diagonal) ? eltype(T) : complex(eltype(T))
+
+        TA, Z, vals = @testinferred schur_full(A)
+        @test eltype(TA) == eltype(Z) == eltype(T)
+        @test eltype(vals) == Tc
+        @test isisometric(Z)
+        @test A * Z ≈ Z * TA
+
+        TA2, Z2, vals2 = @testinferred schur_full!(Ac, (TA, Z, vals))
+        @test TA2 === TA
+        @test Z2 === Z
+        @test vals2 === vals
+        @test A * Z ≈ Z * TA
+
+        valsc = @testinferred schur_vals(A)
+        @test eltype(valsc) == Tc
+        @test valsc ≈ eig_vals(A)
+    end
+end