Skip to content

Commit 8514785

Browse files
authored
Merge pull request #24 from ITensor/ksh/testindex
More tests for Index, tighten up dir minus method
2 parents d537091 + 8b6a46d commit 8514785

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

src/index.jl

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,8 @@ associated with an index, i.e. the index leg is directed into or out of a given
3434
Reverse direction of a directed `Arrow`. Will throw an error for an input of `Neither`.
3535
"""
3636
function -(dir::Arrow)
37-
if dir==Neither
38-
error("Cannot reverse direction of Arrow direction 'Neither'")
39-
else
40-
return dir==In ? Out : In
41-
end
37+
dir==Neither && throw(ArgumentError("Cannot reverse direction of Arrow direction 'Neither'"))
38+
return dir==In ? Out : In
4239
end
4340

4441
"""

test/test_index.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ import ITensors: In,Out,Neither
99
@test id(i) == 0
1010
@test dim(i) == 1
1111
@test dir(i) == Neither
12+
@test_throws ArgumentError -dir(i)
1213
@test plev(i) == 0
1314
@test tags(i) == TagSet("0")
1415
end
@@ -46,6 +47,8 @@ import ITensors: In,Out,Neither
4647
@test plev(i2) == 2
4748
i3 = i'''
4849
@test plev(i3) == 3
50+
i0 = noprime(i)
51+
@test plev(i0) == 0
4952
end
5053
@testset "IndexVal" begin
5154
i = Index(2)
@@ -59,5 +62,7 @@ import ITensors: In,Out,Neither
5962
@test plev(i(2)') == 1
6063
@test val(i(2)') == 2
6164
@test plev(prime(i(2),4)) == 4
65+
@test i[:] == [i(1); i(2)]
66+
@test sprint(show, i(2)) == sprint(show, i)*"=2"
6267
end
6368
end

0 commit comments

Comments
 (0)