Add tests

Author: mtfishman

src/sparsearrays.jl

@@ -5,7 +5,7 @@ function eachstoredindex(m::AbstractSparseMatrixCSC)
     # TODO: This loses the compile time element type, is there a better lazy way?
     return Iterators.map(CartesianIndex, zip(I, J))
 end
-function eachstoredindex(a::Base.ReshapedArray{<:Any, <:Any, <: AbstractSparseMatrixCSC})
+function eachstoredindex(a::Base.ReshapedArray{<:Any, <:Any, <:AbstractSparseMatrixCSC})
     return @interface SparseArrayInterface() eachstoredindex(a)
 end
 
@@ -17,16 +17,16 @@ function SparseArrays.SparseMatrixCSC{Tv, Ti}(m::AnyAbstractSparseMatrix) where
     return m′
 end
 
-function SparseArrayDOK(a::Base.ReshapedArray{<:Any, <:Any, <: AbstractSparseMatrixCSC})
+function SparseArrayDOK(a::Base.ReshapedArray{<:Any, <:Any, <:AbstractSparseMatrixCSC})
     return SparseArrayDOK{eltype(a), ndims(a)}(a)
 end
 function SparseArrayDOK{T}(
-        a::Base.ReshapedArray{<:Any, <:Any, <: AbstractSparseMatrixCSC}
+        a::Base.ReshapedArray{<:Any, <:Any, <:AbstractSparseMatrixCSC}
     ) where {T}
     return SparseArrayDOK{T, ndims(a)}(a)
 end
 function SparseArrayDOK{T, N}(
-        a::Base.ReshapedArray{<:Any, N, <: AbstractSparseMatrixCSC}
+        a::Base.ReshapedArray{<:Any, N, <:AbstractSparseMatrixCSC}
     ) where {T, N}
     a′ = SparseArrayDOK{T, N}(undef, size(a))
     for I in eachstoredindex(a)

test/Project.toml

@@ -11,6 +11,7 @@ SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 SparseArraysBase = "0d5efcca-f356-4864-8770-e1ed8d78f208"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
+TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [sources]
@@ -29,4 +30,5 @@ SafeTestsets = "0.1.0"
 SparseArraysBase = "0.7.0"
 StableRNGs = "1.0.2"
 Suppressor = "0.2.8"
+TensorAlgebra = "0.6"
 Test = "<0.0.1, 1"

test/test_tensoralgebraext.jl

@@ -0,0 +1,30 @@
+using SparseArrays: SparseMatrixCSC, findnz, nnz
+using SparseArraysBase: SparseMatrixDOK, sparsezeros
+using TensorAlgebra: contract, matricize
+using Test: @testset, @test
+
+@testset "TensorAlgebraExt (eltype = $elt)" for elt in (Float32, ComplexF64)
+    a = sparsezeros(elt, (2, 2, 2))
+    a[1, 1, 1] = 1
+    a[2, 1, 2] = 2
+
+    # matricize
+    m = matricize(a, (1, 3), (2,))
+    @test m isa SparseMatrixCSC{elt}
+    @test nnz(m) == 2
+    @test isstored(m, 1, 1)
+    @test m[1, 1] ≡ elt(1)
+    @test isstored(m, 4, 1)
+    @test m[4, 1] ≡ elt(2)
+    @test issetequal(eachstoredindex(m), [CartesianIndex(1, 1), CartesianIndex(4, 1)])
+    for I in setdiff(CartesianIndices(m), [CartesianIndex(1, 1), CartesianIndex(4, 1)])
+        @test m[I] ≡ zero(elt)
+    end
+
+    # contract
+    b, l = contract(a, ("i", "j", "k"), a, ("j", "k", "l"))
+    @test b isa SparseMatrixDOK{elt}
+    @test storedlength(b) == 1
+    @test only(eachstoredindex(b)) == CartesianIndex(1, 1)
+    @test b[1, 1] ≡ elt(1)
+end