update gauge test

Author: xuanzhaogao

README.md

@@ -4,3 +4,11 @@
 <!-- [![Dev](https://img.shields.io/badge/docs-dev-blue.svg)](https://ArrogantGao.github.io/BPGauge.jl/dev/) -->
 <!-- [![Build Status](https://github.com/ArrogantGao/BPGauge.jl/actions/workflows/CI.yml/badge.svg?branch=main)](https://github.com/ArrogantGao/BPGauge.jl/actions/workflows/CI.yml?query=branch%3Amain) -->
 <!-- [![Coverage](https://codecov.io/gh/ArrogantGao/BPGauge.jl/branch/main/graph/badge.svg)](https://codecov.io/gh/ArrogantGao/BPGauge.jl) -->
+
+This package provides a gauge transformation for graph tensor network states.
+
+A simple example:
+```julia
+
+
+```

test/gauge.jl

@@ -66,41 +66,36 @@ end
     @test inner_product(tn, adjoint(tn)) ≈ 1.0
 end
 
-@testset "guage mps" begin
+@testset "Vidal gauge mps" begin
     g = chain(100)
     tn = random_state(g, d_virtual = 8)
     normalize_state!(tn)
 
     bp_state = BPState(tn)
     bp_path = BPPath(tn)
-    bp!(bp_state, bp_path, tn, verbose = true, err_bound = 1e-14)
+    bp!(bp_state, bp_path, tn, err_bound = 1e-14)
 
     gauge!(tn, bp_state)
+    @test inner_product(tn, adjoint(tn)) ≈ 1.0
 
-    for i in 1:nv(g) - 2
+    for i in 2:nv(g) - 2
         G = tn.gauge_tensors[tn.gauge_tensors_map[(i, i + 1)]]
         T = tn.site_tensors[i + 1]
         L = ein"ij, jk, jln, kmn -> lm"(G, G, T, conj(T))
-        @show i, maximum(abs.(L ./ L[1, 1] - I(size(L, 1))))
+        @test maximum(abs.(L ./ L[1, 1] - I(size(L, 1)))) < 1e-6
     end
 end
 
-
-@testset "Vidal gauge" begin
-    g = chain(100)
-    tn = random_state(g, d_virtual = 8)
+@testset "graph gauge" begin
+    g = random_regular_graph(30, 3)
+    tn = random_state(g, d_virtual = 4)
     normalize_state!(tn)
-    
+    @test inner_product(tn, adjoint(tn)) ≈ 1.0
+
     bp_state = BPState(tn)
     bp_path = BPPath(tn)
-    bp!(bp_state, bp_path, tn, err_bound = 1e-14)
+    bp!(bp_state, bp_path, tn, err_bound = 1e-8)
 
     gauge!(tn, bp_state)
-
-    for i in 2:nv(g) - 2
-        G = tn.gauge_tensors[tn.gauge_tensors_map[(i, i + 1)]]
-        T = tn.site_tensors[i + 1]
-        L = ein"ij, jk, jln, kmn -> lm"(G, G, T, conj(T))
-        @test maximum(abs.(L ./ L[1, 1] - I(size(L, 1)))) < 1e-6
-    end
+    @test inner_product(tn, adjoint(tn)) ≈ 1.0
 end