separate tests into groups

Author: lkdvos

test/ad.jl test/autodiff/ad.jltest/ad.jl renamed to test/autodiff/ad.jl

@@ -1,3 +1,6 @@
+using Test, TestExtras
+using TensorKit
+using TensorOperations
 using ChainRulesCore
 using ChainRulesTestUtils
 using FiniteDifferences: FiniteDifferences, central_fdm, forward_fdm

test/braidingtensor.jl

@@ -1,4 +1,5 @@
 # TODO: Make into proper tests and integrate in testset
+# note: this is not part of the testsuite!
 
 import TensorKit: BraidingTensor
 

test/bugfixes.jl

@@ -0,0 +1,2 @@
+using Aqua
+Aqua.test_all(TensorKit)

test/other/aqua.jl

@@ -0,0 +1,87 @@
+using Test, TestExtras
+using TensorKit
+using TensorOperations
+using LinearAlgebra: LinearAlgebra
+using Zygote
+
+@testset "BugfixConvert" begin
+    v = randn(
+        ComplexF64,
+        (
+            Vect[(Irrep[U₁] ⊠ Irrep[SU₂] ⊠ FermionParity)](
+                (-3, 1 / 2, 1) => 3, (-5, 1 / 2, 1) => 10, (-7, 1 / 2, 1) => 13,
+                (-9, 1 / 2, 1) => 9, (-11, 1 / 2, 1) => 1, (-5, 3 / 2, 1) => 3,
+                (-7, 3 / 2, 1) => 3, (-9, 3 / 2, 1) => 1
+            ) ⊗ Vect[(Irrep[U₁] ⊠ Irrep[SU₂] ⊠ FermionParity)]((1, 1 / 2, 1) => 1)'
+        ) ← Vect[(Irrep[U₁] ⊠ Irrep[SU₂] ⊠ FermionParity)](
+            (-3, 1 / 2, 1) => 3, (-5, 1 / 2, 1) => 10, (-7, 1 / 2, 1) => 13,
+            (-9, 1 / 2, 1) => 9, (-11, 1 / 2, 1) => 1, (-5, 3 / 2, 1) => 3,
+            (-7, 3 / 2, 1) => 3, (-9, 3 / 2, 1) => 1
+        )
+    )
+    w = convert(typeof(real(v)), v)
+    @test w == v
+    @test scalartype(w) == Float64
+end
+
+# https://github.com/quantumkithub/TensorKit.jl/issues/178
+@testset "Issue #178" begin
+    t = rand(U1Space(1 => 1) ← U1Space(1 => 1)')
+    a = convert(Array, t)
+    @test a == zeros(size(a))
+end
+
+# https://github.com/quantumkithub/TensorKit.jl/issues/194
+@testset "Issue #194" begin
+    t1 = rand(ℂ^4 ← ℂ^4)
+    t2 = tensoralloc(typeof(t1), space(t1), Val(true), TensorOperations.ManualAllocator())
+    t3 = similar(t2, ComplexF64, space(t1))
+    @test storagetype(t3) == Vector{ComplexF64}
+    t4 = similar(t2, domain(t1))
+    @test storagetype(t4) == Vector{Float64}
+    t5 = similar(t2)
+    @test storagetype(t5) == Vector{Float64}
+    tensorfree!(t2)
+end
+
+# https://github.com/quantumkithub/TensorKit.jl/issues/201
+@testset "Issue #201" begin
+    function f(A::AbstractTensorMap)
+        U, S, V, = svd_compact(A)
+        return tr(S)
+    end
+    function f(A::AbstractMatrix)
+        S = LinearAlgebra.svdvals(A)
+        return sum(S)
+    end
+    A₀ = randn(Z2Space(4, 4) ← Z2Space(4, 4))
+    grad1, = Zygote.gradient(f, A₀)
+    grad2, = Zygote.gradient(f, convert(Array, A₀))
+    @test convert(Array, grad1) ≈ grad2
+
+    function g(A::AbstractTensorMap)
+        U, S, V, = svd_compact(A)
+        return tr(U * V)
+    end
+    function g(A::AbstractMatrix)
+        U, S, V, = LinearAlgebra.svd(A)
+        return tr(U * V')
+    end
+    B₀ = randn(ComplexSpace(4) ← ComplexSpace(4))
+    grad3, = Zygote.gradient(g, B₀)
+    grad4, = Zygote.gradient(g, convert(Array, B₀))
+    @test convert(Array, grad3) ≈ grad4
+end
+
+# https://github.com/quantumkithub/TensorKit.jl/issues/209
+@testset "Issue #209" begin
+    function f(T, D)
+        @tensor T[1, 4, 1, 3] * D[3, 4]
+    end
+    V = Z2Space(2, 2)
+    D = DiagonalTensorMap(randn(4), V)
+    T = randn(V ⊗ V ← V ⊗ V)
+    g1, = Zygote.gradient(f, T, D)
+    g2, = Zygote.gradient(f, T, TensorMap(D))
+    @test g1 ≈ g2
+end

test/other/bugfixes.jl

@@ -70,182 +70,3 @@ istestfile(fn) = endswith(fn, ".jl") && !contains(fn, "setup")
         end
     end
 end
-
-
-using Test
-using TestExtras
-using Random
-using TensorKit
-using Combinatorics
-using TensorKit: ProductSector, fusiontensor, pentagon_equation, hexagon_equation
-using TensorOperations
-using Base.Iterators: take, product
-# using SUNRepresentations: SUNIrrep
-# const SU3Irrep = SUNIrrep{3}
-using LinearAlgebra: LinearAlgebra
-using Zygote: Zygote
-
-const TK = TensorKit
-
-Random.seed!(1234)
-
-# don't run all tests on GPU, only the GPU
-# specific ones
-is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
-
-smallset(::Type{I}) where {I <: Sector} = take(values(I), 5)
-function smallset(::Type{ProductSector{Tuple{I1, I2}}}) where {I1, I2}
-    iter = product(smallset(I1), smallset(I2))
-    s = collect(i ⊠ j for (i, j) in iter if dim(i) * dim(j) <= 6)
-    return length(s) > 6 ? rand(s, 6) : s
-end
-function smallset(::Type{ProductSector{Tuple{I1, I2, I3}}}) where {I1, I2, I3}
-    iter = product(smallset(I1), smallset(I2), smallset(I3))
-    s = collect(i ⊠ j ⊠ k for (i, j, k) in iter if dim(i) * dim(j) * dim(k) <= 6)
-    return length(s) > 6 ? rand(s, 6) : s
-end
-function randsector(::Type{I}) where {I <: Sector}
-    s = collect(smallset(I))
-    a = rand(s)
-    while a == one(a) # don't use trivial label
-        a = rand(s)
-    end
-    return a
-end
-function hasfusiontensor(I::Type{<:Sector})
-    try
-        fusiontensor(one(I), one(I), one(I))
-        return true
-    catch e
-        if e isa MethodError
-            return false
-        else
-            rethrow(e)
-        end
-    end
-end
-
-sectorlist = (
-    Z2Irrep, Z3Irrep, Z4Irrep, Z3Irrep ⊠ Z4Irrep,
-    U1Irrep, CU1Irrep, SU2Irrep,
-    FermionParity, FermionParity ⊠ FermionParity,
-    FermionParity ⊠ U1Irrep ⊠ SU2Irrep, FermionParity ⊠ SU2Irrep ⊠ SU2Irrep, # Hubbard-like
-    FibonacciAnyon, IsingAnyon,
-    Z2Irrep ⊠ FibonacciAnyon ⊠ FibonacciAnyon,
-)
-
-# spaces
-Vtr = (ℂ^2, (ℂ^3)', ℂ^4, ℂ^3, (ℂ^2)')
-Vℤ₂ = (
-    Vect[Z2Irrep](0 => 1, 1 => 1),
-    Vect[Z2Irrep](0 => 1, 1 => 2)',
-    Vect[Z2Irrep](0 => 3, 1 => 2)',
-    Vect[Z2Irrep](0 => 2, 1 => 3),
-    Vect[Z2Irrep](0 => 2, 1 => 5),
-)
-Vfℤ₂ = (
-    Vect[FermionParity](0 => 1, 1 => 1),
-    Vect[FermionParity](0 => 1, 1 => 2)',
-    Vect[FermionParity](0 => 2, 1 => 1)',
-    Vect[FermionParity](0 => 2, 1 => 3),
-    Vect[FermionParity](0 => 2, 1 => 5),
-)
-Vℤ₃ = (
-    Vect[Z3Irrep](0 => 1, 1 => 2, 2 => 1),
-    Vect[Z3Irrep](0 => 2, 1 => 1, 2 => 1),
-    Vect[Z3Irrep](0 => 1, 1 => 2, 2 => 1)',
-    Vect[Z3Irrep](0 => 1, 1 => 2, 2 => 3),
-    Vect[Z3Irrep](0 => 1, 1 => 3, 2 => 3)',
-)
-VU₁ = (
-    Vect[U1Irrep](0 => 1, 1 => 2, -1 => 2),
-    Vect[U1Irrep](0 => 3, 1 => 1, -1 => 1),
-    Vect[U1Irrep](0 => 2, 1 => 2, -1 => 1)',
-    Vect[U1Irrep](0 => 1, 1 => 2, -1 => 3),
-    Vect[U1Irrep](0 => 1, 1 => 3, -1 => 3)',
-)
-VfU₁ = (
-    Vect[FermionNumber](0 => 1, 1 => 2, -1 => 2),
-    Vect[FermionNumber](0 => 3, 1 => 1, -1 => 1),
-    Vect[FermionNumber](0 => 2, 1 => 2, -1 => 1)',
-    Vect[FermionNumber](0 => 1, 1 => 2, -1 => 3),
-    Vect[FermionNumber](0 => 1, 1 => 3, -1 => 3)',
-)
-VCU₁ = (
-    Vect[CU1Irrep]((0, 0) => 1, (0, 1) => 2, 1 => 1),
-    Vect[CU1Irrep]((0, 0) => 3, (0, 1) => 0, 1 => 1),
-    Vect[CU1Irrep]((0, 0) => 1, (0, 1) => 0, 1 => 2)',
-    Vect[CU1Irrep]((0, 0) => 2, (0, 1) => 2, 1 => 1),
-    Vect[CU1Irrep]((0, 0) => 2, (0, 1) => 1, 1 => 2)',
-)
-VSU₂ = (
-    Vect[SU2Irrep](0 => 3, 1 // 2 => 1),
-    Vect[SU2Irrep](0 => 2, 1 => 1),
-    Vect[SU2Irrep](1 // 2 => 1, 1 => 1)',
-    Vect[SU2Irrep](0 => 2, 1 // 2 => 2),
-    Vect[SU2Irrep](0 => 1, 1 // 2 => 1, 3 // 2 => 1)',
-)
-VfSU₂ = (
-    Vect[FermionSpin](0 => 3, 1 // 2 => 1),
-    Vect[FermionSpin](0 => 2, 1 => 1),
-    Vect[FermionSpin](1 // 2 => 1, 1 => 1)',
-    Vect[FermionSpin](0 => 2, 1 // 2 => 2),
-    Vect[FermionSpin](0 => 1, 1 // 2 => 1, 3 // 2 => 1)',
-)
-VSU₂U₁ = (
-    Vect[SU2Irrep ⊠ U1Irrep]((0, 0) => 1, (1 // 2, -1) => 1),
-    Vect[SU2Irrep ⊠ U1Irrep](
-        (0, 0) => 2, (0, 2) => 1, (1, 0) => 1, (1, -2) => 1,
-        (1 // 2, -1) => 1
-    ),
-    Vect[SU2Irrep ⊠ U1Irrep]((1 // 2, 1) => 1, (1, -2) => 1)',
-    Vect[SU2Irrep ⊠ U1Irrep]((0, 0) => 2, (0, 2) => 1, (1 // 2, 1) => 1),
-    Vect[SU2Irrep ⊠ U1Irrep]((0, 0) => 1, (1 // 2, 1) => 1)',
-)
-Vfib = (
-    Vect[FibonacciAnyon](:I => 1, :τ => 1),
-    Vect[FibonacciAnyon](:I => 1, :τ => 2)',
-    Vect[FibonacciAnyon](:I => 3, :τ => 2)',
-    Vect[FibonacciAnyon](:I => 2, :τ => 3),
-    Vect[FibonacciAnyon](:I => 2, :τ => 2),
-)
-
-if !is_buildkite
-    Ti = time()
-    @time include("fusiontrees.jl")
-    @time include("spaces.jl")
-    @time include("tensors.jl")
-    @time include("factorizations.jl")
-    @time include("diagonal.jl")
-    @time include("planar.jl")
-    if !(Sys.isapple() && get(ENV, "CI", "false") == "true") && isempty(VERSION.prerelease)
-        @time include("ad.jl")
-    end
-    @time include("bugfixes.jl")
-    Tf = time()
-    printstyled(
-        "Finished all tests in ",
-        string(round((Tf - Ti) / 60; sigdigits = 3)),
-        " minutes."; bold = true, color = Base.info_color()
-    )
-    println()
-    @testset "Aqua" verbose = true begin
-        using Aqua
-        Aqua.test_all(TensorKit)
-    end
-else
-    Ti = time()
-    #=using CUDA
-    if CUDA.functional()
-    end
-    using AMDGPU
-    if AMDGPU.functional()
-    end=#
-    Tf = time()
-    printstyled(
-        "Finished all GPU tests in ",
-        string(round((Tf - Ti) / 60; sigdigits = 3)),
-        " minutes."; bold = true, color = Base.info_color()
-    )
-    println()
-end