test normalize_eachmatrix

stecrotti · May 27, 2024 · eff4741 · eff4741
1 parent 18ffe03
commit eff4741
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/test/periodic_tensor_train.jl b/test/periodic_tensor_train.jl
@@ -75,6 +75,21 @@
         end
     end
 
+
+    @testset "Normalize eachmatrix" begin
+        rng = MersenneTwister(0)
+        qs = (2, 4)
+        L = 4
+        A = rand_periodic_tt( [1; rand(rng, 10:15, L-1); 1], qs... )
+        A.z = -100 * rand(rng)
+        x, p = sample(rng, A)
+        e = evaluate(A, x)
+        z = normalization(A)
+        normalize_eachmatrix!(A)
+        @test evaluate(A, x) ≈ e
+        @test float(normalization(A)) ≈ float(z)
+    end
+
     @testset "Flat" begin
         L = 5
         bondsizes = rand(1:4, L)

diff --git a/test/tensor_train.jl b/test/tensor_train.jl
@@ -163,6 +163,20 @@ end
         end
     end
 
+    @testset "Normalize eachmatrix" begin
+        rng = MersenneTwister(0)
+        qs = (2, 4)
+        L = 4
+        A = rand_tt( [1; rand(rng, 10:15, L-1); 1], qs... )
+        A.z = -100 * rand(rng)
+        x, p = sample(rng, A)
+        e = evaluate(A, x)
+        z = normalization(A)
+        normalize_eachmatrix!(A)
+        @test evaluate(A, x) ≈ e
+        @test float(normalization(A)) ≈ float(z)
+    end
+
     @testset "Sum of TTs" begin
         for N in 1:3
             for q in 1:3