Create derivative sparse arrays

Author: kshyatt

test/runtests.jl

@@ -6,6 +6,9 @@ include("testsuite.jl")
 const init_code = quote
     using Test, JLArrays, SparseArrays
 
+    sparse_types(::Type{<:JLArray}) = (JLSparseVector, JLSparseMatrixCSC, JLSparseMatrixCSR)
+    sparse_types(::Type{<:Array})   = (SparseVector, SparseMatrixCSC)
+
     include("testsuite.jl")
 
     # Disable Float16-related tests until JuliaGPU/KernelAbstractions#600 is resolved
@@ -16,11 +19,7 @@ const init_code = quote
 end
 
 custom_tests = Dict{String, Expr}()
-for AT in (:JLArray, :Array), name in filter(n->n != "sparse", keys(TestSuite.tests))
-    custom_tests["$(AT)/$name"] = :(TestSuite.tests[$name]($AT))
-end
-
-for AT in (:JLSparseMatrixCSR, :JLSparseMatrixCSC, :JLSparseVector, :SparseMatrixCSC, :SparseVector), name in ["sparse"]
+for AT in (:JLArray, :Array), name in keys(TestSuite.tests)
     custom_tests["$(AT)/$name"] = :(TestSuite.tests[$name]($AT))
 end
 

test/testsuite.jl

@@ -63,6 +63,10 @@ supported_eltypes() = (Int16, Int32, Int64,
                        ComplexF16, ComplexF32, ComplexF64,
                        Complex{Int16}, Complex{Int32}, Complex{Int64})
 
+# derived sparse types that are supported by the array type
+
+sparse_types(::Type{AT}) where {AT} = ()
+
 # some convenience predicates for filtering test eltypes
 isrealtype(T) = T <: Real
 iscomplextype(T) = T <: Complex

test/testsuite/sparse.jl

@@ -1,9 +1,12 @@
 @testsuite "sparse" (AT, eltypes)->begin
-    if AT <: AbstractSparseVector
-        broadcasting_vector(AT, eltypes)
-    elseif AT <: AbstractSparseMatrix
-        broadcasting_matrix(AT, eltypes)
-        mapreduce_matrix(AT, eltypes)
+    sparse_ATs = sparse_types(AT)
+    for sparse_AT in sparse_ATs
+        if sparse_AT <: AbstractSparseVector
+            broadcasting_vector(sparse_AT, eltypes)
+        elseif sparse_AT <: AbstractSparseMatrix
+            broadcasting_matrix(sparse_AT, eltypes)
+            mapreduce_matrix(sparse_AT, eltypes)
+        end
     end
 end