Add more tests

Author: mtfishman

src/algorithms.jl

@@ -72,17 +72,20 @@ can't be passed to `MatrixAlgebraKit.select_algorithm`.
 """
 function select_algorithm end
 
-Base.@constprop :aggressive function select_algorithm(f::F, A, alg::Alg=nothing; kwargs...) where {F,Alg}
+Base.@constprop :aggressive function select_algorithm(f::F, A, alg::Alg=nothing;
+                                                      kwargs...) where {F,Alg}
     return _select_algorithm(f, A, alg; kwargs...)
 end
 
 function _select_algorithm(f::F, A, alg::Nothing; kwargs...) where {F}
     return default_algorithm(f, A; kwargs...)
 end
-Base.@constprop :aggressive function _select_algorithm(f::F, A, alg::Symbol; kwargs...) where {F}
+Base.@constprop :aggressive function _select_algorithm(f::F, A, alg::Symbol;
+                                                       kwargs...) where {F}
     return Algorithm{alg}(; kwargs...)
 end
-Base.@constprop :aggressive function _select_algorithm(f::F, A, ::Type{Alg}; kwargs...) where {F,Alg}
+Base.@constprop :aggressive function _select_algorithm(f::F, A, ::Type{Alg};
+                                                       kwargs...) where {F,Alg}
     return Alg(; kwargs...)
 end
 function _select_algorithm(f::F, A, alg::NamedTuple; kwargs...) where {F}
@@ -174,7 +177,8 @@ macro functiondef(f)
 
     return esc(quote
                    # out of place to inplace
-                   Base.@constprop :aggressive $f(A; kwargs...) = $f!(copy_input($f, A); kwargs...)
+                   Base.@constprop :aggressive $f(A; kwargs...) = $f!(copy_input($f, A);
+                                                                      kwargs...)
                    $f(A, alg::AbstractAlgorithm) = $f!(copy_input($f, A), alg)
 
                    # fill in arguments

test/algorithms.jl

@@ -0,0 +1,62 @@
+using MatrixAlgebraKit
+using Test
+using TestExtras
+using MatrixAlgebraKit: LAPACK_SVDAlgorithm, NoTruncation, PolarViaSVD, TruncatedAlgorithm,
+                        default_algorithm, select_algorithm
+
+@testset "default_algorithm" begin
+    A = randn(3, 3)
+    for f in (svd_compact!, svd_compact, svd_full!, svd_full)
+        @test @constinferred(default_algorithm(f, A)) === LAPACK_DivideAndConquer()
+    end
+    for f in (eig_full!, eig_full, eig_vals!, eig_vals)
+        @test @constinferred(default_algorithm(f, A)) === LAPACK_Expert()
+    end
+    for f in (eigh_full!, eigh_full, eigh_vals!, eigh_vals)
+        @test @constinferred(default_algorithm(f, A)) ===
+              LAPACK_MultipleRelativelyRobustRepresentations()
+    end
+    for f in (lq_full!, lq_full, lq_compact!, lq_compact, lq_null!, lq_null)
+        @test @constinferred(default_algorithm(f, A)) == LAPACK_HouseholderLQ()
+    end
+    for f in (left_polar!, left_polar, right_polar!, right_polar)
+        @test @constinferred(default_algorithm(f, A)) ==
+              PolarViaSVD(LAPACK_DivideAndConquer())
+    end
+    for f in (qr_full!, qr_full, qr_compact!, qr_compact, qr_null!, qr_null)
+        @test @constinferred(default_algorithm(f, A)) == LAPACK_HouseholderQR()
+    end
+    for f in (schur_full!, schur_full, schur_vals!, schur_vals)
+        @test @constinferred(default_algorithm(f, A)) === LAPACK_Expert()
+    end
+
+    @test @constinferred(default_algorithm(qr_compact!, A; blocksize=2)) ===
+          LAPACK_HouseholderQR(; blocksize=2)
+end
+
+@testset "select_algorithm" begin
+    A = randn(3, 3)
+    for f in (svd_trunc!, svd_trunc)
+        @test @constinferred(select_algorithm(f, A)) ===
+              TruncatedAlgorithm(LAPACK_DivideAndConquer(), NoTruncation())
+    end
+    for f in (eig_trunc!, eig_trunc)
+        @test @constinferred(select_algorithm(f, A)) ===
+              TruncatedAlgorithm(LAPACK_Expert(), NoTruncation())
+    end
+    for f in (eigh_trunc!, eigh_trunc)
+        @test @constinferred(select_algorithm(f, A)) ===
+              TruncatedAlgorithm(LAPACK_MultipleRelativelyRobustRepresentations(),
+                                 NoTruncation())
+    end
+
+    @test @constinferred(select_algorithm(svd_compact!, A)) === LAPACK_DivideAndConquer()
+    @test @constinferred(select_algorithm(svd_compact!, A, nothing)) ===
+          LAPACK_DivideAndConquer()
+    @test @constinferred(select_algorithm(svd_compact!, A, :LAPACK_QRIteration)) ===
+          LAPACK_QRIteration()
+    @test @constinferred(select_algorithm(svd_compact!, A, LAPACK_QRIteration)) ===
+          LAPACK_QRIteration()
+    @test @constinferred(select_algorithm(svd_compact!, A, LAPACK_QRIteration())) ===
+          LAPACK_QRIteration()
+end

test/runtests.jl

@@ -1,5 +1,8 @@
 using SafeTestsets
 
+@safetestset "Algorithms" begin
+    include("algorithms.jl")
+end
 @safetestset "Truncate" begin
     include("truncate.jl")
 end