Pass truncation kwargs to insert_local_tensor (#106)

Author: leburgel

src/treetensornetworks/solvers/update_step.jl

@@ -185,7 +185,16 @@ function local_update(
   #end
 
   psi, spec = insert_local_tensor(
-    psi, phi, region; eigen_perturbation=drho, ortho, normalize, kwargs...
+    psi,
+    phi,
+    region;
+    eigen_perturbation=drho,
+    ortho,
+    normalize,
+    cutoff,
+    maxdim,
+    mindim,
+    kwargs...,
   )
 
   update!(

test/test_treetensornetworks/test_solvers/test_dmrg.jl

@@ -112,7 +112,6 @@ end
   cutoff = 1e-12
 
   tooth_lengths = fill(2, 3)
-  root_vertex = (3, 2)
   c = named_comb_tree(tooth_lengths)
   s = siteinds("S=1/2", c)
 
@@ -140,4 +139,19 @@ end
   @test inner(psi', H, psi) ≈ inner(psi2', Hline, psi2) atol = 1e-5
 end
 
+@testset "Regression test: tree truncation" begin
+  maxdim = 4
+  nsite = 2
+  nsweeps = 10
+
+  c = named_comb_tree((3, 2))
+  s = siteinds("S=1/2", c)
+  os = ITensorNetworks.heisenberg(c)
+  H = TTN(os, s)
+  psi = random_ttn(s; link_space=5)
+  psi = dmrg(H, psi; nsweeps, maxdim, nsite)
+
+  @test all(edge_data(linkdims(psi)) .<= maxdim)
+end
+
 nothing

test/test_treetensornetworks/test_solvers/test_tdvp.jl

@@ -633,7 +633,7 @@ end
       F = abs(scalar(dag(psix) * contract(psi)))
     end
 
-    @test norm(Sz_tdvp - Sz_exact) < 1e-5 # broken when rebasing local draft on remote main, fix this
+    @test norm(Sz_tdvp - Sz_exact) < 1e-5
   end
 
   # TODO: apply gates in ITensorNetworks