Tests rework (#75)

* initial update of tests, adding cft tests

* fix comment

* LoopTNR cft methods test

* update tolerances

* test the models

* add free_energy function

* test using new free_energy function

* test fermions

* update test running

* update ATRG3D test

* fix deprecation warnings

Author: VictorVanthilt

.github/workflows/CI.yml

@@ -22,14 +22,13 @@ jobs:
           - '1'
         os:
           - ubuntu-latest
-        arch:
-          - x64
+          - macOS-latest
+          - windows-latest
     steps:
       - uses: actions/checkout@v4
       - uses: julia-actions/setup-julia@v2
         with:
           version: ${{ matrix.version }}
-          arch: ${{ matrix.arch }}
       - uses: julia-actions/cache@v2
       - uses: julia-actions/julia-buildpkg@latest
       - uses: julia-actions/julia-runtest@latest

src/TNRKit.jl

@@ -51,7 +51,7 @@ export run!
 
 # models
 include("models/ising.jl")
-export classical_ising, classical_ising_symmetric, ising_βc, f_onsager,
+export classical_ising, classical_ising_symmetric, ising_βc, f_onsager, ising_cft_exact,
     ising_βc_3D, classical_ising_symmetric_3D, classical_ising_3D
 
 include("models/gross-neveu.jl")
@@ -67,6 +67,9 @@ include("models/clock.jl")
 export classical_clock
 
 # utility functions
+include("utility/free_energy.jl")
+export free_energy
+
 include("utility/cft.jl")
 export cft_data, central_charge, cft_data!
 

src/utility/cft.jl

@@ -12,7 +12,7 @@ function cft_data(scheme::TNRScheme; v = 1, unitcell = 1, is_real = true)
     outinds = Tuple(collect(1:unitcell))
     ininds = Tuple(collect((unitcell + 1):(2unitcell)))
 
-    T = permute(T, outinds, ininds)
+    T = permute(T, (outinds, ininds))
     D, _ = eig(T)
 
     data = zeros(ComplexF64, dim(space(D, 1)))
@@ -48,7 +48,7 @@ function cft_data(scheme::BTRG; v = 1, unitcell = 1, is_real = true)
     outinds = Tuple(collect(1:unitcell))
     ininds = Tuple(collect((unitcell + 1):(2unitcell)))
 
-    T = permute(T, outinds, ininds)
+    T = permute(T, (outinds, ininds))
     D, _ = eig(T)
 
     data = zeros(ComplexF64, dim(space(D, 1)))

src/utility/free_energy.jl

@@ -0,0 +1,8 @@
+function free_energy(data, β; scalefactor = 2.0, unitcell = 1.0)
+    lnz = 0.0
+    x = 1.0 - log(unitcell) / log(scalefactor)
+    for (i, z) in enumerate(data)
+        lnz += log(z) * scalefactor^(x - i)
+    end
+    return -lnz / β
+end

test/fermions.jl

@@ -0,0 +1,37 @@
+f_bench = -1.4515448845652446
+T = gross_neveu_start(0, 0, 0)
+
+# === TRG ===
+@testset "TRG - Gross-Neveu Model" begin
+    scheme = TRG(T)
+    data = run!(scheme, truncdim(16), maxiter(25))
+    @test free_energy(data, 1.0) ≈ f_bench rtol = 1.0e-3
+end
+
+# === BTRG ===
+@testset "BTRG - Gross-Neveu Model" begin
+    scheme = BTRG(T)
+    data = run!(scheme, truncdim(16), maxiter(25))
+    @test free_energy(data, 1.0) ≈ f_bench rtol = 1.0e-4
+end
+
+# === HOTRG ===
+@testset "HOTRG - Gross-Neveu Model" begin
+    scheme = HOTRG(T)
+    data = run!(scheme, truncdim(16), maxiter(25))
+    @test free_energy(data, 1.0; scalefactor = 4.0) ≈ f_bench rtol = 1.0e-3
+end
+
+# === ATRG ===
+@testset "ATRG - Gross-Neveu Model" begin
+    scheme = ATRG(T)
+    data = run!(scheme, truncdim(16), maxiter(25))
+    @test free_energy(data, 1.0; scalefactor = 4.0) ≈ f_bench rtol = 1.0e-2
+end
+
+# === LoopTNR ===
+@testset "LoopTNR - Gross-Neveu Model" begin
+    scheme = LoopTNR(T)
+    data = run!(scheme, truncdim(8), maxiter(10))
+    @test free_energy(data, 1.0) ≈ f_bench rtol = 1.0e-3
+end

test/ising.jl

@@ -0,0 +1,47 @@
+println("--------------------")
+println(" Testing all models ")
+println("--------------------")
+
+models_2D = [
+    classical_ising(),
+    classical_ising_symmetric(),
+    gross_neveu_start(0, 0, 0),
+    # classical_clock(), # TODO: find out how to test this model
+    classical_potts(3),
+    classical_potts_symmetric(3),
+    sixvertex(Float64, Trivial),
+    sixvertex(Float64, U1Irrep),
+    sixvertex(Float64, CU1Irrep),
+]
+
+temperatures = [
+    ising_βc,
+    ising_βc,
+    1.0,
+    potts_βc(3),
+    potts_βc(3),
+    1.0,
+    1.0,
+    1.0,
+]
+
+answers = [
+    f_onsager, # Hack because classical_ising starts from larger lattice
+    f_onsager,
+    -1.4515448845652446,
+    -4.119552029995684, # This is an approximation!
+    -4.119552029995684, # This is an approximation!
+    3 / 2 * log(3 / 4),
+    3 / 2 * log(3 / 4),
+    3 / 2 * log(3 / 4),
+]
+
+unitcells = vcat([2.0], fill(1.0, length(models_2D) - 1))
+
+@testset "2D Models" begin
+    for (model, temp, answer, unitcell) in zip(models_2D, temperatures, answers, unitcells)
+        scheme = TRG(model)
+        data = run!(scheme, truncdim(16), maxiter(25))
+        @test free_energy(data, temp; unitcell = unitcell) ≈ answer rtol = 1.0e-3
+    end
+end

test/models.jl

@@ -5,5 +5,6 @@ using PEPSKit: InfinitePartitionFunction, CTMRGEnv, SequentialCTMRG, leading_bou
 using QuadGK
 
 include("spaces.jl") # do they give spacemismatches?
-include("ising.jl") # do they give the correct results (with the expected accuracy)?
-include("finalize.jl") # do they give the correct results (with the expected accuracy)?
+include("schemes.jl") # do they give the correct results (with the expected accuracy)?
+include("models.jl") # do they give the correct results (with the expected accuracy)?
+include("fermions.jl") # do they give the correct results (with the expected accuracy)?