Define mapnameddimsindices (#25)

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "NamedDimsArrays"
 uuid = "60cbd0c0-df58-4cb7-918c-6f5607b73fde"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.3.9"
+version = "0.3.10"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/abstractnameddimsarray.jl

@@ -290,6 +290,9 @@ function replacenameddimsindices(a::AbstractNamedDimsArray, replacements::Pair..
   new_nameddimsindices = named.(dename.(old_nameddimsindices), last.(replacements))
   return replacenameddimsindices(a, (old_nameddimsindices .=> new_nameddimsindices)...)
 end
+function mapnameddimsindices(f, a::AbstractNamedDimsArray)
+  return setnameddimsindices(a, map(f, nameddimsindices(a)))
+end
 
 # `Base.isempty(a::AbstractArray)` is defined as `length(a) == 0`,
 # which involves comparing a named integer to an unnamed integer

test/basics/test_basics.jl

@@ -17,6 +17,7 @@ using NamedDimsArrays:
   dims,
   fusednames,
   isnamed,
+  mapnameddimsindices,
   name,
   named,
   nameddims,
@@ -138,6 +139,9 @@ using Test: @test, @test_throws, @testset
     nb = replacenameddimsindices(n -> n == named(1:3, "i") ? named(1:3, "k") : n, na)
     @test nameddimsindices(nb) == (named(1:3, "k"), named(1:4, "j"))
     @test dename(nb) == a
+    nb = mapnameddimsindices(n -> n == named(1:3, "i") ? named(1:3, "k") : n, na)
+    @test nameddimsindices(nb) == (named(1:3, "k"), named(1:4, "j"))
+    @test dename(nb) == a
     nb = setnameddimsindices(na, named(3, "i") => named(3, "k"))
     na[1, 1] = 11
     @test na[1, 1] == 11