test: introduce tests for SquareSymmetric unit cell

fix: missing files

fix: adjust load order

Author: jack-dunham

src/InfiniteTensorContractions.jl

@@ -6,6 +6,8 @@ using KrylovKit
 using LinearAlgebra
 using TensorKit
 using DocStringExtensions
+using Observers
+using DataFrames
 
 export ITC
 
@@ -48,6 +50,7 @@ include("convergenceinfo.jl")
 include("utils.jl")
 include("callback.jl")
 
+include("compositetensor.jl")
 include("abstractunitcell.jl")
 include("abstractproblem.jl")
 include("networks.jl")

test/networks.jl

@@ -23,4 +23,38 @@
     @test virtualspace(ct, 2) == s * s'
     @test virtualspace(ct, 3) == w' * w
     @test virtualspace(ct, 4) == n' * n
+
+    @testset "SquareSymmetric" verbose = true begin
+        data = [
+            1 2 3
+            2 3 1
+            3 1 2
+        ]
+
+        ucsym = UnitCell{SquareSymmetric}([1, 2, 3])
+
+        @test UnitCell{Square}(data) == ucsym
+
+        @test ucsym == UnitCell{SquareSymmetric}([1 2 3])
+        @test ucsym == UnitCell{SquareSymmetric}([1; 2; 3;;])
+
+        for i in 1:3
+            @test ucsym[i, :] == data[i, :]
+            @test ucsym[:, i] == data[:, i]
+        end
+
+        @test identity.(ucsym) == ucsym
+        @test exp2.(ucsym) == UnitCell{SquareSymmetric}([2, 4, 8])
+        @test ucsym .+ ucsym == 2 * ucsym
+        @test ucsym + ucsym == 2 * ucsym
+
+        let rv1 = ucsym + data, rv2 = ucsym .+ data
+            for rv in (rv1, rv2)
+                @test rv == 2 * data
+                @test isa(rv, UnitCell{Square})
+            end
+        end
+
+        @test ucsym == ucsym'
+    end
 end