Skip to content

Commit c206de8

Browse files
committed
add tests for styles of types
1 parent 7226788 commit c206de8

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

test/operators.jl

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,11 @@ module TestOperators
3434
mpo₂ = FiniteMPO(O₂)
3535
mpo₃ = FiniteMPO(O₃)
3636

37+
@test isfinite(mpo₁)
38+
@test isfinite(typeof(mpo₁))
39+
@test GeometryStyle(typeof(mpo₁)) == FiniteStyle()
3740
@test GeometryStyle(mpo₁) == FiniteStyle()
41+
@test OperatorStyle(typeof(mpo₁)) == MPOStyle()
3842
@test OperatorStyle(mpo₁) == MPOStyle()
3943

4044

@@ -95,7 +99,11 @@ module TestOperators
9599
H1 += H1'
96100
H = InfiniteMPO([H1])
97101

102+
@test !isfinite(H)
103+
@test !isfinite(typeof(H))
104+
@test GeometryStyle(typeof(H)) == InfiniteStyle()
98105
@test GeometryStyle(H) == InfiniteStyle()
106+
@test OperatorStyle(typeof(H)) == MPOStyle()
99107
@test OperatorStyle(H) == MPOStyle()
100108
end
101109

@@ -127,7 +135,11 @@ module TestOperators
127135
H′ = FiniteMPOHamiltonian(map(Base.Fix1(collect, Any), Ws)) # without type info
128136
@test H H′
129137

138+
@test isfinite(H)
139+
@test isfinite(typeof(H))
140+
@test GeometryStyle(typeof(H)) == FiniteStyle()
130141
@test GeometryStyle(H) == FiniteStyle()
142+
@test OperatorStyle(typeof(H)) == HamiltonianStyle()
131143
@test OperatorStyle(H) == HamiltonianStyle()
132144

133145
# Infinite
@@ -141,7 +153,11 @@ module TestOperators
141153
H′ = InfiniteMPOHamiltonian(map(Base.Fix1(collect, Any), Ws)) # without type info
142154
@test all(parent(H) .≈ parent(H′))
143155

156+
@test !isfinite(H)
157+
@test !isfinite(typeof(H))
158+
@test GeometryStyle(typeof(H)) == InfiniteStyle()
144159
@test GeometryStyle(H) == InfiniteStyle()
160+
@test OperatorStyle(typeof(H)) == HamiltonianStyle()
145161
@test OperatorStyle(H) == HamiltonianStyle()
146162
end
147163

test/states.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,9 @@ module TestStates
2525
L = rand(3:20)
2626
ψ = FiniteMPS(rand, elt, L, d, D)
2727

28+
@test isfinite(ψ)
2829
@test isfinite(typeof(ψ))
30+
@test GeometryStyle(typeof(ψ)) == FiniteStyle()
2931
@test GeometryStyle(ψ) == FiniteStyle()
3032
@test @constinferred physicalspace(ψ) == fill(d, L)
3133
@test all(x -> x D, @constinferred left_virtualspace(ψ))
@@ -102,7 +104,9 @@ module TestStates
102104

103105
ψ = InfiniteMPS([rand(elt, D * d, D), rand(elt, D * d, D)]; tol)
104106

107+
@test !isfinite(typeof(ψ))
105108
@test !isfinite(ψ)
109+
@test GeometryStyle(typeof(ψ)) == InfiniteStyle()
106110
@test GeometryStyle(ψ) == InfiniteStyle()
107111

108112
@test physicalspace(ψ) == fill(d, 2)
@@ -236,6 +240,7 @@ module TestStates
236240
ϕ₂ = LeftGaugedQP(rand, ψ)
237241

238242
@test GeometryStyle(ϕ₁) == FiniteStyle()
243+
@test GeometryStyle(typeof(ϕ₂)) == FiniteStyle()
239244

240245
@test @constinferred physicalspace(ϕ₁) == physicalspace(ψ)
241246
@test @constinferred left_virtualspace(ϕ₁) == left_virtualspace(ψ)
@@ -273,6 +278,7 @@ module TestStates
273278
ϕ₂ = LeftGaugedQP(rand, ψ)
274279

275280
@test GeometryStyle(ϕ₁) == InfiniteStyle()
281+
@test GeometryStyle(typeof(ϕ₂)) == InfiniteStyle()
276282

277283
@test @constinferred physicalspace(ϕ₁) == physicalspace(ψ)
278284
@test @constinferred left_virtualspace(ϕ₁) == left_virtualspace(ψ)

0 commit comments

Comments
 (0)