Fix externalinds, add tests for externalinds and internalinds (#…

…123)
ITensor · Jan 12, 2024 · 9a75be2 · 9a75be2
1 parent 0b72b7c
commit 9a75be2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/src/abstractitensornetwork.jl b/src/abstractitensornetwork.jl
@@ -263,7 +263,7 @@ function internalinds(tn::AbstractITensorNetwork)
 end
 
 function externalinds(tn::AbstractITensorNetwork)
-  return unique(flatten([uniqueinds(tn, e) for e in edges(tn)]))
+  return unique(flatten([uniqueinds(tn, v) for v in vertices(tn)]))
 end
 
 # Priming and tagging (changing Index identifiers)

diff --git a/src/exports.jl b/src/exports.jl
@@ -58,6 +58,8 @@ export AbstractITensorNetwork,
   norm_sqr_network,
   linkinds_combiners,
   combine_linkinds,
+  externalinds,
+  internalinds,
   subgraphs,
   reverse_bfs_edges,
   randomITensorNetwork,

diff --git a/test/test_itensornetwork.jl b/test/test_itensornetwork.jl
@@ -217,6 +217,9 @@ using Test
     @test isempty(commoninds(uie, cie))
 
     @test linkinds(ψ, e) == commoninds(ψ[1, 1], ψ[2, 1])
+
+    @test length(externalinds(ψ)) == length(vertices(g))
+    @test length(internalinds(ψ)) == length(edges(g))
   end
 
   @testset "ElType conversion, $new_eltype" for new_eltype in (Float32, ComplexF64)