Add tensors test

bsc-quantic · Aug 1, 2023 · 18b9620 · 18b9620
1 parent 4dd2766
commit 18b9620
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/test/MatrixProductOperator_test.jl b/test/MatrixProductOperator_test.jl
@@ -76,6 +76,14 @@
  only(select(ψ, last(ψ.interlayer)[3])).meta[:alias][:l]
  end
  end
+
+ @testset "tensors" begin
+ arrays = [rand(1, 2, 2), rand(1, 1, 2, 2), rand(1, 2, 2)]
+ ψ = MatrixProduct{Operator,Open}(arrays, order = (:l, :r, :i, :o))
+
+ @test tensors(ψ, 1) isa Tensor
+ @test size(tensors(ψ)) |> first == length(ψ) == 3
+ end
  end
 
  @testset "`Periodic` boundary" begin
@@ -131,6 +139,14 @@
  @test only(select(ψ, first(ψ.interlayer)[3])).meta[:alias][:r] ===
  only(select(ψ, first(ψ.interlayer)[1])).meta[:alias][:l]
  end
+
+ @testset "tensors" begin
+ arrays = [rand(1, 1, 2, 2), rand(1, 1, 2, 2), rand(1, 1, 2, 2)]
+ ψ = MatrixProduct{Operator,Periodic}(arrays, order = (:l, :r, :i, :o))
+
+ @test tensors(ψ, 1) isa Tensor
+ @test size(tensors(ψ)) |> first == length(ψ) == 3
+ end
  end
  end
  end

diff --git a/test/MatrixProductState_test.jl b/test/MatrixProductState_test.jl
@@ -80,6 +80,14 @@
  @test tensors(ψ, 1).meta[:alias][:r] === tensors(ψ, 2).meta[:alias][:l]
  @test tensors(ψ, 2).meta[:alias][:r] === tensors(ψ, 3).meta[:alias][:l]
  end
+
+ @testset "tensors" begin
+ arrays = [rand(1, 2), rand(1, 1, 2), rand(1, 2)]
+ ψ = MatrixProduct{State,Open}(arrays, order = (:l, :r, :o))
+
+ @test tensors(ψ, 1) isa Tensor
+ @test size(tensors(ψ)) |> first == length(ψ) == 3
+ end
  end
  end
 
@@ -128,6 +136,14 @@
  @test tensors(ψ, 2).meta[:alias][:r] === tensors(ψ, 3).meta[:alias][:l]
  @test tensors(ψ, 3).meta[:alias][:r] === tensors(ψ, 1).meta[:alias][:l]
  end
+
+ @testset "tensors" begin
+ arrays = [rand(1, 1, 2), rand(1, 1, 2), rand(1, 1, 2)]
+ ψ = MatrixProduct{State,Periodic}(arrays, order = (:l, :r, :o))
+
+ @test tensors(ψ, 1) isa Tensor
+ @test size(tensors(ψ)) |> first == length(ψ) == 3
+ end
  end
  end
  end