Add tests

Author: mtfishman

src/lib/ITensorsNamedDimsArraysExt/examples/example_dmrg.jl

@@ -1,6 +1,5 @@
 using Adapt: adapt
-using ITensors: MPO, dmrg, random_mps, siteinds
-using ITensors.Ops: OpSum
+using ITensorMPS: MPO, OpSum, dmrg, random_mps, siteinds
 using ITensors.ITensorsNamedDimsArraysExt: to_nameddimsarray
 
 function main(; n, conserve_qns=false, nsweeps=3, cutoff=1e-4, arraytype=Array)

src/lib/ITensorsNamedDimsArraysExt/src/to_nameddimsarray.jl

@@ -63,6 +63,10 @@ using ITensors: QN
 using ..NDTensors.GradedAxes: GradedAxes
 GradedAxes.fuse_labels(l1::QN, l2::QN) = l1 + l2
 
+using ITensors: QN
+using ..NDTensors.SymmetrySectors: SymmetrySectors
+SymmetrySectors.dual(l::QN) = -l
+
 ## TODO: Add this back, define `CombinerArrays` library in NDTensors!
 ## using ..NDTensors: CombinerTensor, CombinerArray, storage
 ## # TODO: Delete when we directly use `CombinerArray` as storage.

src/lib/ITensorsNamedDimsArraysExt/test/Project.toml

@@ -0,0 +1,5 @@
+[deps]
+BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
+ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
+NDTensors = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
+Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"

src/lib/ITensorsNamedDimsArraysExt/test/test_basics.jl

@@ -0,0 +1,33 @@
+@eval module $(gensym())
+using BlockArrays: blocklengths
+using ITensors: ITensor, Index, QN, dag, inds, plev, random_itensor
+using ITensors.ITensorsNamedDimsArraysExt: to_nameddimsarray
+using NDTensors: tensor
+using NDTensors.BlockSparseArrays: BlockSparseArray, block_nstored
+using NDTensors.GradedAxes: isdual
+using NDTensors.LabelledNumbers: label
+using NDTensors.NamedDimsArrays: NamedDimsArray, unname
+using Test: @test, @testset
+@testset "to_nameddimsarray" begin
+  i = Index([QN(0) => 2, QN(1) => 3])
+  a = random_itensor(i', dag(i))
+  b = to_nameddimsarray(a)
+  @test b isa ITensor
+  @test plev(inds(b)[1]) == 1
+  @test plev(inds(b)[2]) == 0
+  @test inds(b)[1] == i'
+  @test inds(b)[2] == dag(i)
+  nb = tensor(b)
+  @test nb isa NamedDimsArray{Float64}
+  bb = unname(nb)
+  @test bb isa BlockSparseArray{Float64}
+  @test !isdual(axes(bb, 1))
+  @test isdual(axes(bb, 2))
+  @test blocklengths(axes(bb, 1)) == [2, 3]
+  @test blocklengths(axes(bb, 2)) == [2, 3]
+  @test label.(blocklengths(axes(bb, 1))) == [QN(0), QN(1)]
+  @test label.(blocklengths(axes(bb, 2))) == [QN(0), QN(-1)]
+  @test block_nstored(bb) == 2
+  @test b' * b ≈ to_nameddimsarray(a' * a)
+end
+end

src/lib/ITensorsNamedDimsArraysExt/test/test_examples.jl

@@ -1,5 +1,17 @@
+@eval module $(gensym())
+using ITensors: ITensors
+using Suppressor: @suppress
 using Test: @testset
-
 @testset "examples" begin
-  include("../examples/example_readme.jl")
+  @suppress include(
+    joinpath(
+      pkgdir(ITensors),
+      "src",
+      "lib",
+      "ITensorsNamedDimsArraysExt",
+      "examples",
+      "example_readme.jl",
+    ),
+  )
+end
 end

test/lib/ITensorsNamedDimsArraysExt/Project.toml

@@ -0,0 +1,4 @@
+[deps]
+ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
+NDTensors = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
+Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"

test/lib/ITensorsNamedDimsArraysExt/runtests.jl

@@ -0,0 +1,8 @@
+@eval module $(gensym())
+using ITensors: ITensors
+include(
+  joinpath(
+    pkgdir(ITensors), "src", "lib", "ITensorsNamedDimsArraysExt", "test", "runtests.jl"
+  ),
+)
+end

test/runtests.jl

@@ -19,6 +19,7 @@ ITensors.disable_threaded_blocksparse()
       "base",
       "threading",
       "lib/ContractionSequenceOptimization",
+      "lib/ITensorsNamedDimsArraysExt",
       "ext/ITensorsChainRulesCoreExt",
       "ext/ITensorsVectorInterfaceExt",
       "ext/NDTensorsMappedArraysExt",