Add some checks on type stability

Author: lkdvos

Project.toml

@@ -43,6 +43,7 @@ Random = "1"
 Statistics = "1"
 TensorKit = "0.14.6"
 TensorOperations = "5"
+TestExtras = "0.3"
 VectorInterface = "0.4, 0.5"
 Zygote = "0.6, 0.7"
 julia = "1.10"
@@ -53,6 +54,7 @@ FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
 QuadGK = "1fd47b50-473d-5c70-9696-f719f8f3bcdc"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
 
 [targets]
-test = ["Test", "SafeTestsets", "ChainRulesTestUtils", "FiniteDifferences", "QuadGK"]
+test = ["Test", "TestExtras", "SafeTestsets", "ChainRulesTestUtils", "FiniteDifferences", "QuadGK"]

test/ctmrg/fixed_iterscheme.jl

@@ -1,4 +1,4 @@
-using Test
+using Test, TestExtras
 using Accessors
 using Random
 using LinearAlgebra
@@ -37,7 +37,7 @@ atol = 1e-5
     env_conv1, = leading_boundary(CTMRGEnv(psi, ComplexSpace(χenv)), psi, ctm_alg)
 
     # do extra iteration to get SVD
-    env_conv2, info = ctmrg_iteration(n, env_conv1, ctm_alg)
+    env_conv2, info = @constinferred ctmrg_iteration(n, env_conv1, ctm_alg)
     env_fix, signs = gauge_fix(env_conv1, env_conv2)
     @test calc_elementwise_convergence(env_conv1, env_fix) ≈ 0 atol = atol
 
@@ -47,7 +47,7 @@ atol = 1e-5
     ctm_alg_fix = @set ctm_alg_fix.projector_alg.trscheme = notrunc()
 
     # do iteration with FixedSVD
-    env_fixedsvd, = ctmrg_iteration(n, env_conv1, ctm_alg_fix)
+    env_fixedsvd, = @constinferred ctmrg_iteration(n, env_conv1, ctm_alg_fix)
     env_fixedsvd = fix_global_phases(env_conv1, env_fixedsvd)
     @test calc_elementwise_convergence(env_conv1, env_fixedsvd) ≈ 0 atol = atol
 end

test/ctmrg/flavors.jl

@@ -1,4 +1,4 @@
-using Test
+using Test, TestExtras
 using Random
 using TensorKit
 using MPSKit
@@ -17,10 +17,10 @@ projector_algs = [:halfinfinite, :fullinfinite]
     # compute environments
     Random.seed!(32350283290358)
     psi = InfinitePEPS(2, χbond; unitcell)
-    env_sequential, = leading_boundary(
+    env_sequential, = @constinferred leading_boundary(
         CTMRGEnv(psi, ComplexSpace(χenv)), psi; alg=:sequential, projector_alg
     )
-    env_simultaneous, = leading_boundary(
+    env_simultaneous, = @constinferred leading_boundary(
         CTMRGEnv(psi, ComplexSpace(χenv)), psi; alg=:simultaneous, projector_alg
     )