add IsingBimodule spaces and help functions for fusiontree tests

Author: borisdevos

test/setup.jl

@@ -1,8 +1,9 @@
 module TestSetup
 
 export smallset, randsector, hasfusiontensor, force_planar
+export random_fusiontree, safe_tensor_product
 export sectorlist
-export Vtr, Vℤ₂, Vfℤ₂, Vℤ₃, VU₁, VfU₁, VCU₁, VSU₂, VfSU₂, VSU₂U₁, Vfib
+export Vtr, Vℤ₂, Vfℤ₂, Vℤ₃, VU₁, VfU₁, VCU₁, VSU₂, VfSU₂, VSU₂U₁, Vfib, VIB_diag, VIB_M
 
 using Random
 using TensorKit
@@ -75,6 +76,34 @@ function force_planar(tsrc::TensorMap{<:Any, <:GradedSpace})
     return tdst
 end
 
+function random_fusiontree(I::Type{<:Sector}, N::Int) # for fusion tree tests
+    in = nothing
+    out = nothing
+    while in === nothing
+        out = ntuple(n -> randsector(I), N)
+        try
+            in = rand(collect(⊗(out...)))
+        catch e
+            if isa(e, ArgumentError)
+                in = nothing
+            else
+                rethrow(e)
+            end 
+        end
+    end
+    return out
+end
+
+# for fusion tree merge test
+function safe_tensor_product(x::I, y::I) where {I<:Sector}
+    tp = x ⊗ y
+    if isempty(tp)
+        return nothing
+    else
+        return tp
+    end
+end
+
 sectorlist = (
     Z2Irrep, Z3Irrep, Z4Irrep, Z3Irrep ⊠ Z4Irrep,
     U1Irrep, CU1Irrep, SU2Irrep,
@@ -160,6 +189,26 @@ Vfib = (
     Vect[FibonacciAnyon](:I => 2, :τ => 3),
     Vect[FibonacciAnyon](:I => 2, :τ => 2),
 )
-#TODO: add IsingBimodule spaces
+
+C0, C1 = IsingBimodule(1, 1, 0), IsingBimodule(1, 1, 1)
+D0, D1 = IsingBimodule(2, 2, 0), IsingBimodule(2, 2, 1)
+M, Mop = IsingBimodule(1, 2, 0), IsingBimodule(2, 1, 0)
+VIB_diag = (
+        Vect[IsingBimodule](C0 => 1, C1 => 2),
+        Vect[IsingBimodule](C0 => 2, C1 => 1),
+        Vect[IsingBimodule](C0 => 3, C1 => 1),
+        Vect[IsingBimodule](C0 => 2, C1 => 3),
+        Vect[IsingBimodule](C0 => 3, C1 => 2)
+)
+
+# not a random ordering! designed to make V1 ⊗ V2 ← V3 ⊗ V4 ⊗ V5 work (tensors)
+# while V1 ⊗ V2 ← V4 isn't empty (factorizations)
+VIB_M = (
+    Vect[IsingBimodule](C0 => 1, C1 => 2),
+         Vect[IsingBimodule](M => 3),
+         Vect[IsingBimodule](C0 => 2, C1 => 3),
+         Vect[IsingBimodule](M => 4),
+         Vect[IsingBimodule](D0 => 3, D1 => 4)
+)
 
 end