Fix replace for NaiveOrderedSet (#28)

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.11"
+version = "0.3.12"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/abstractnameddimsarray.jl

@@ -207,6 +207,12 @@ end
 function Base.map(f::Function, s::NaiveOrderedSet)
   return NaiveOrderedSet(map(f, values(s)))
 end
+function Base.replace(f::Union{Function,Type}, s::NaiveOrderedSet; kwargs...)
+  return NaiveOrderedSet(replace(f, values(s); kwargs...))
+end
+function Base.replace(s::NaiveOrderedSet, replacements::Pair...; kwargs...)
+  return NaiveOrderedSet(replace(values(s), replacements...; kwargs...))
+end
 
 function Base.axes(a::AbstractNamedDimsArray)
   return NaiveOrderedSet(map(named, axes(dename(a)), nameddimsindices(a)))

test/basics/test_adapt.jl renamed to test/test_adapt.jl

@@ -3,6 +3,7 @@ using NamedDimsArrays:
   NamedDimsArrays,
   AbstractNamedDimsArray,
   AbstractNamedDimsMatrix,
+  NaiveOrderedSet,
   Name,
   NameMismatch,
   NamedDimsCartesianIndex,
@@ -271,4 +272,23 @@ using Test: @test, @test_throws, @testset
       @test !iszero(na)
     end
   end
+  @testset "NaiveOrderedSet" begin
+    s = NaiveOrderedSet(("a", "b", "c"))
+    @test values(s) == ("a", "b", "c")
+    @test Tuple(s) == ("a", "b", "c")
+    @test s[1] == "a"
+    @test s[2] == "b"
+    @test s[3] == "c"
+    for s′ in (
+      replace(x -> x == "b" ? "x" : x, s),
+      replace(s, "b" => "x"),
+      map(x -> x == "b" ? "x" : x, s),
+    )
+      @test s′ isa NaiveOrderedSet
+      @test Tuple(s′) == ("a", "x", "c")
+      @test s′[1] == "a"
+      @test s′[2] == "x"
+      @test s′[3] == "c"
+    end
+  end
 end