pass basic tests

Author: ogauthe

Project.toml

@@ -1,7 +1,7 @@
 name = "FusionTensors"
 uuid = "e16ca583-1f51-4df0-8e12-57d32947d33e"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
@@ -12,26 +12,26 @@ HalfIntegers = "f0d1745a-41c9-11e9-1dd9-e5d34d218721"
 LRUCache = "8ac3fa9e-de4c-5943-b1dc-09c6b5f20637"
 LabelledNumbers = "f856a3a6-4152-4ec4-b2a7-02c1a55d7993"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
-SparseArraysBase = "0d5efcca-f356-4864-8770-e1ed8d78f208"
 Strided = "5e0ebb24-38b0-5f93-81fe-25c709ecae67"
 SymmetrySectors = "f8a8ad64-adbc-4fce-92f7-ffe2bb36a86e"
 TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
+TensorProducts = "decf83d6-1968-43f4-96dc-fdb3fe15fc6d"
 TypeParameterAccessors = "7e5a90cf-f82e-492e-a09b-e3e26432c138"
 WignerSymbols = "9f57e263-0b3d-5e2e-b1be-24f2bb48858b"
 
 [compat]
 Accessors = "0.1.39"
 BlockArrays = "1.2.0"
-BlockSparseArrays = "0.2.9"
-GradedUnitRanges = "0.1.1"
+BlockSparseArrays = "0.3.9"
+GradedUnitRanges = "0.2.2"
 HalfIntegers = "1.6.0"
 LRUCache = "1.6.1"
 LabelledNumbers = "0.1.0"
 LinearAlgebra = "1.10.0"
-SparseArraysBase = "0.2.0"
 Strided = "2.2.0"
-SymmetrySectors = "0.1.1"
-TensorAlgebra = "0.1.5"
+SymmetrySectors = "0.1.7"
+TensorAlgebra = "0.2.4"
+TensorProducts = "0.1.3"
 TypeParameterAccessors = "0.2.0"
 WignerSymbols = "2.0.0"
 julia = "1.10"

src/fusion_trees/clebsch_gordan_tensors.jl

@@ -5,9 +5,8 @@
 using HalfIntegers: half
 using WignerSymbols: clebschgordan
 
-using GradedUnitRanges: dual, fusion_product
+using GradedUnitRanges: dual
 using SymmetrySectors:
-  ⊗,
   AbelianStyle,
   AbstractSector,
   NotAbelianStyle,
@@ -20,6 +19,7 @@ using SymmetrySectors:
   trivial,
   zero_odd
 using TensorAlgebra: contract
+using TensorProducts: ⊗
 
 function symbol_1j(s::AbstractSector)
   cgt = clebsch_gordan_tensor(s, dual(s), trivial(s), 1)
@@ -66,7 +66,7 @@ function clebsch_gordan_tensor(s1::O2, s2::O2, s3::O2)
   d2 = quantum_dimension(s2)
   d3 = quantum_dimension(s3)
   cgt = zeros((d1, d2, d3))
-  s3 ∉ blocklabels(fusion_product(s1, s2)) && return cgt
+  s3 ∉ blocklabels(s1 ⊗ s2) && return cgt
 
   # adapted from TensorKit
   l1 = sector_label(s1)

src/fusion_trees/fusiontree.jl

@@ -3,10 +3,11 @@
 # TBD
 # compatibility with TensorKit conventions?
 
-using GradedUnitRanges:
-  AbstractGradedUnitRange, GradedUnitRanges, flip, fusion_product, isdual, sector_type
-using SymmetrySectors: ×, AbstractSector, SectorProduct, SymmetrySectors, arguments, trivial
+using GradedUnitRanges: AbstractGradedUnitRange, GradedUnitRanges, flip, isdual, sector_type
+using SymmetrySectors:
+  ×, AbstractSector, SectorProduct, SymmetrySectors, arguments, trivial, to_gradedrange
 using TensorAlgebra: flatten_tuples
+using TensorProducts: ⊗, tensor_product
 
 #
 # A fusion tree fuses N sectors sec1, secN  onto one sector fused_sec. A given set of
@@ -186,7 +187,7 @@ end
 
 # TODO move to GradedUnitRanges
 function nsymbol(s1::AbstractSector, s2::AbstractSector, s3::AbstractSector)
-  full_space = fusion_product(s1, s2)
+  full_space = tensor_product(s1, s2)
   x = findfirst(==(s3), blocklabels(full_space))
   isnothing(x) && return 0  # OR labelled(0, s3)?
   return Int(blocklengths(full_space)[x])
@@ -241,7 +242,7 @@ end
 function fuse_next_sector(
   parent_tree::SectorFusionTree, branch_sector::AbstractSector, level_arrow::Bool
 )
-  new_space = fusion_product(root_sector(parent_tree), branch_sector)
+  new_space = to_gradedrange(root_sector(parent_tree) ⊗ branch_sector)
   return mapreduce(vcat, zip(blocklabels(new_space), blocklengths(new_space))) do (la, n)
     return [
       append_tree_leave(parent_tree, branch_sector, level_arrow, la, outer_mult) for

src/fusiontensor/fusiontensor.jl

@@ -17,6 +17,7 @@ using GradedUnitRanges:
   space_isequal
 using SymmetrySectors: SectorProduct, TrivialSector
 using TensorAlgebra: BlockedTuple, tuplemortar
+using TensorProducts: tensor_product
 
 struct FusionTensor{T,N,Axes,Mat,Mapping} <: AbstractArray{T,N}
   data_matrix::Mat
@@ -108,13 +109,13 @@ function promote_sectors(
   # fuse with trivial to insert all missing arguments inside each GradedAxis
   # avoid depending on SymmetrySectors internals
   s0 = trivial(T)
-  return map_blocklabels.(s -> only(blocklabels(fusion_product(s0, s))), legs)
+  return map_blocklabels.(s -> only(blocklabels(tensor_product(s0, s))), legs)
 end
 
 function promote_sector_type(legs)
   # fuse trivial sectors to produce unified type
   # avoid depending on SymmetrySectors internals
-  return sector_type(fusion_product(trivial.(legs)...))
+  return sector_type(tensor_product(trivial.(legs)...))
 end
 
 # initialize with already computed data_matrix
@@ -229,7 +230,7 @@ function initialize_data_matrix(
 )
   # non-abelian fusion trees have float eltype: need compatible type
   promoted = promote_type(elt, fusiontree_eltype(sector_type(mat_row_axis)))
-  mat = BlockSparseArray{promoted}(mat_row_axis, mat_col_axis)
+  mat = BlockSparseArray{promoted}(undef, mat_row_axis, mat_col_axis)
   initialize_allowed_sectors!(mat)
   return mat
 end

test/Project.toml

@@ -9,6 +9,7 @@ SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
 SymmetrySectors = "f8a8ad64-adbc-4fce-92f7-ffe2bb36a86e"
 TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
+TensorProducts = "decf83d6-1968-43f4-96dc-fdb3fe15fc6d"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]

test/basics/test_array_cast.jl

@@ -5,8 +5,9 @@ using Test: @test, @test_throws, @testset
 using BlockArrays: Block, BlockedArray, blocksize
 
 using FusionTensors: FusionTensor, data_matrix, to_fusiontensor
-using GradedUnitRanges: dual, fusion_product, gradedrange
+using GradedUnitRanges: dual, gradedrange
 using SymmetrySectors: O2, SectorProduct, SU2, TrivialSector, U1
+using TensorProducts: tensor_product
 
 include("setup.jl")
 
@@ -365,7 +366,7 @@ end
   end
 
   @testset "large identity" begin
-    g = reduce(fusion_product, (SU2(1 / 2), SU2(1 / 2), SU2(1 / 2)))
+    g = reduce(tensor_product, (SU2(1 / 2), SU2(1 / 2), SU2(1 / 2)))
     N = 3
     codomain_legs = ntuple(_ -> g, N)
     domain_legs = dual.(codomain_legs)

test/basics/test_basics.jl

@@ -14,10 +14,10 @@ using FusionTensors:
   matrix_size,
   ndims_domain,
   ndims_codomain
-using GradedUnitRanges:
-  blockmergesort, dual, flip, fusion_product, gradedrange, sector_type, space_isequal
+using GradedUnitRanges: blockmergesort, dual, flip, gradedrange, sector_type, space_isequal
 using SymmetrySectors: U1, SU2, SectorProduct, TrivialSector, Z
 using TensorAlgebra: tuplemortar
+using TensorProducts: tensor_product
 
 include("setup.jl")
 
@@ -27,10 +27,11 @@ include("setup.jl")
 
   # check dual convention when initializing data_matrix
   ft0 = FusionTensor(Float64, (g1,), (g2,))
+  @test ft0 isa FusionTensor
   @test space_isequal(matrix_row_axis(ft0), g1)
   @test space_isequal(matrix_column_axis(ft0), g2)
 
-  m = BlockSparseArray{Float64}(g1, g2)
+  m = BlockSparseArray{Float64}(undef, g1, g2)
   ft1 = FusionTensor(m, (g1,), (g2,))
 
   # getters
@@ -104,9 +105,9 @@ end
   g2 = gradedrange([U1(0) => 2, U1(1) => 2, U1(3) => 1])
   g3 = dual(gradedrange([U1(-1) => 1, U1(0) => 2, U1(1) => 1]))
   g4 = dual(gradedrange([U1(-1) => 1, U1(0) => 1, U1(1) => 1]))
-  gr = fusion_product(g1, g2)
-  gc = dual(fusion_product(dual(g3), dual(g4)))
-  m2 = BlockSparseArray{Float64}(gr, gc)
+  gr = tensor_product(g1, g2)
+  gc = dual(tensor_product(dual(g3), dual(g4)))
+  m2 = BlockSparseArray{Float64}(undef, gr, gc)
   ft = FusionTensor(m2, (g1, g2), (g3, g4))
 
   @test data_matrix(ft) == m2
@@ -249,7 +250,7 @@ end
   @test_throws DimensionMismatch ft7 * ft3
 end
 
-@testset "mising SectorProduct" begin
+@testset "missing SectorProduct" begin
   g1 = gradedrange([SectorProduct(U1(1)) => 1])
   g2 = gradedrange([SectorProduct(U1(1), SU2(1//2)) => 1])
   g3 = gradedrange([SectorProduct(U1(1), SU2(1//2), Z{2}(1)) => 1])
@@ -267,7 +268,7 @@ end
   gA = gradedrange([SectorProduct(; A=U1(1)) => 1])
   gB = gradedrange([SectorProduct(; B=SU2(1//2)) => 1])
   gC = gradedrange([SectorProduct(; C=Z{2}(0)) => 1])
-  gABC = fusion_product(fusion_product(gA, gB), gC)
+  gABC = tensor_product(gA, gB, gC)
   S = sector_type(gABC)
 
   ft = FusionTensor(Float64, (gA, gB), (dual(gA), dual(gB), gC))

test/basics/test_fusion_trees.jl

@@ -13,7 +13,7 @@ using FusionTensors:
   leaves,
   outer_multiplicity_indices,
   root_sector
-using GradedUnitRanges: blocklabels, fusion_product, sector_type
+using GradedUnitRanges: blocklabels, sector_type
 using SymmetrySectors: ×, SectorProduct, SU, SU2, TrivialSector, arguments
 
 @testset "Trivial fusion trees" begin