Update tests

Author: lkdvos

Project.toml

@@ -16,6 +16,7 @@ MatrixAlgebraKitChainRulesCoreExt = "ChainRulesCore"
 Aqua = "0.6, 0.7, 0.8"
 ChainRulesCore = "1"
 ChainRulesTestUtils = "1"
+CUDA = "0.12, 0.13"
 JET = "0.9"
 LinearAlgebra = "1"
 SafeTestsets = "0.1"
@@ -27,6 +28,7 @@ julia = "1.10"
 
 [extras]
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
+CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
 JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
@@ -36,4 +38,5 @@ TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [targets]
-test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras","ChainRulesCore", "ChainRulesTestUtils", "StableRNGs", "Zygote"]
+test = ["Aqua", "JET", "SafeTestsets", "CUDA", "Test", "TestExtras", "ChainRulesCore",
+        "ChainRulesTestUtils", "StableRNGs", "Zygote"]

test/cuda.jl

@@ -0,0 +1,3 @@
+println("hello")
+@test true
+using CUDA

test/runtests.jl

@@ -1,43 +1,64 @@
 using SafeTestsets
 
-@safetestset "Truncate" begin
-    include("truncate.jl")
-end
-@safetestset "QR / LQ Decomposition" begin
-    include("qr.jl")
-    include("lq.jl")
-end
-@safetestset "Singular Value Decomposition" begin
-    include("svd.jl")
-end
-@safetestset "Hermitian Eigenvalue Decomposition" begin
-    include("eigh.jl")
-end
-@safetestset "General Eigenvalue Decomposition" begin
-    include("eig.jl")
-end
-@safetestset "Schur Decomposition" begin
-    include("schur.jl")
-end
-@safetestset "Polar Decomposition" begin
-    include("polar.jl")
-end
-@safetestset "Image and Null Space" begin
-    include("orthnull.jl")
-end
-@safetestset "ChainRules" begin
-    include("chainrules.jl")
+pat = r"(?:--group=)(\w+)"
+arg_id = findfirst(pat, ARGS)
+const GROUP = if isnothing(arg_id)
+    uppercase(get(ENV, "GROUP", "ALL"))
+else
+    uppercase(only(match(pat, ARGS[arg_id]).captures))
 end
 
-@safetestset "MatrixAlgebraKit.jl" begin
-    @safetestset "Code quality (Aqua.jl)" begin
-        using MatrixAlgebraKit
-        using Aqua
-        Aqua.test_all(MatrixAlgebraKit)
+@time begin
+    if GROUP == "ALL" || GROUP == "DECOMPOSITIONS"
+        @safetestset "Truncate" begin
+            include("truncate.jl")
+        end
+        @safetestset "QR / LQ Decomposition" begin
+            include("qr.jl")
+            include("lq.jl")
+        end
+        @safetestset "Singular Value Decomposition" begin
+            include("svd.jl")
+        end
+        @safetestset "Hermitian Eigenvalue Decomposition" begin
+            include("eigh.jl")
+        end
+        @safetestset "General Eigenvalue Decomposition" begin
+            include("eig.jl")
+        end
+        @safetestset "Schur Decomposition" begin
+            include("schur.jl")
+        end
+        @safetestset "Polar Decomposition" begin
+            include("polar.jl")
+        end
+        @safetestset "Image and Null Space" begin
+            include("orthnull.jl")
+        end
+    end
+
+    if GROUP == "ALL" || GROUP == "ChainRules"
+        @safetestset "ChainRules" begin
+            include("chainrules.jl")
+        end
     end
-    @safetestset "Code linting (JET.jl)" begin
-        using MatrixAlgebraKit
-        using JET
-        JET.test_package(MatrixAlgebraKit; target_defined_modules=true)
+
+    if GROUP == "ALL" || GROUP == "CUDA"
+        @safetestset "CUDA" begin
+            include("cuda.jl")
+        end
+    end
+
+    if GROUP == "ALL" || GROUP == "UTILITY"
+        @safetestset "Code quality (Aqua.jl)" begin
+            using MatrixAlgebraKit
+            using Aqua
+            Aqua.test_all(MatrixAlgebraKit)
+        end
+        @safetestset "Code linting (JET.jl)" begin
+            using MatrixAlgebraKit
+            using JET
+            JET.test_package(MatrixAlgebraKit; target_defined_modules=true)
+        end
     end
 end