Test Kronecker matricize

Author: mtfishman

ext/KroneckerArraysTensorAlgebraExt/KroneckerArraysTensorAlgebraExt.jl

@@ -1,7 +1,8 @@
 module KroneckerArraysTensorAlgebraExt
 
-using KroneckerArrays: KroneckerArrays, KroneckerArray
-using TensorAlgebra: TensorAlgebra, AbstractBlockPermutation, FusionStyle, matricize
+using KroneckerArrays: KroneckerArrays, KroneckerArray, ⊗, arg1, arg2
+using TensorAlgebra:
+  TensorAlgebra, AbstractBlockPermutation, FusionStyle, matricize, unmatricize
 
 struct KroneckerFusion{A<:FusionStyle,B<:FusionStyle} <: FusionStyle
   a::A

test/test_tensoralgebra.jl

@@ -0,0 +1,10 @@
+using TensorAlgebra: matricize, unmatricize
+using KroneckerArrays: ⊗, arg1, arg2
+using Test: @test, @testset
+
+@testset "TensorAlgebraExt" begin
+  a = randn(2, 2, 2) ⊗ randn(3, 3, 3)
+  m = matricize(a, (1, 2), (3,))
+  @test m == matricize(arg1(a), (1, 2), (3,)) ⊗ matricize(arg2(a), (1, 2), (3,))
+  @test unmatricize(m, (axes(a, 1), axes(a, 2)), (axes(a, 3),)) == a
+end