Define setprime for Index (#34)

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "ITensorBase"
 uuid = "4795dd04-0d67-49bb-8f44-b89c448a1dc7"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.1.16"
+version = "0.1.17"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

src/index.jl

@@ -93,6 +93,7 @@ gettag(i::Index, tagname::String, default) = gettag(name(i), tagname, default)
 settag(i::Index, tagname::String, tag::String) = setname(i, settag(name(i), tagname, tag))
 unsettag(i::Index, tagname::String) = setname(i, unsettag(name(i), tagname))
 
+setprime(i::Index, plev) = setname(i, setprime(name(i), plev))
 prime(i::Index) = setname(i, prime(name(i)))
 noprime(i::Index) = setname(i, noprime(name(i)))
 sim(i::Index) = setname(i, sim(name(i)))

test/test_basics.jl

@@ -12,6 +12,7 @@ using ITensorBase:
   inds,
   plev,
   prime,
+  setprime,
   settag,
   tags,
   unsettag
@@ -38,6 +39,11 @@ using Test: @test, @test_broken, @test_throws, @testset
     @test_throws ErrorException ITensor(randn(elt, 2, 2), Index.((2, 3)))
     @test_throws ErrorException ITensor(randn(elt, 4), Index.((2, 2)))
 
+    i = Index(2)
+    @test plev(i) == 0
+    i = setprime(i, 2)
+    @test plev(i) == 2
+
     i = Index(2)
     i = settag(i, "X", "x")
     @test hastag(i, "X")