Define dual(::AbstractNamedUnitRange) (#33)

Author: mtfishman

Project.toml

@@ -17,16 +17,19 @@ TypeParameterAccessors = "7e5a90cf-f82e-492e-a09b-e3e26432c138"
 
 [weakdeps]
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
+GradedUnitRanges = "e2de450a-8a67-46c7-b59c-01d5a3d041c5"
 
 [extensions]
 NamedDimsArraysBlockArraysExt = "BlockArrays"
+NamedDimsArraysGradedUnitRangesExt = "GradedUnitRanges"
 
 [compat]
 Adapt = "4.1.1"
 ArrayLayouts = "1.11.0"
 BlockArrays = "1.3.0"
 DerivableInterfaces = "0.3.7"
 FillArrays = "1.13.0"
+GradedUnitRanges = "0.1.3"
 LinearAlgebra = "1.10"
 MapBroadcast = "0.1.6"
 Random = "1.10"

ext/NamedDimsArraysGradedUnitRangesExt/NamedDimsArraysGradedUnitRangesExt.jl

@@ -0,0 +1,14 @@
+module NamedDimsArraysGradedUnitRangesExt
+
+using GradedUnitRanges: GradedUnitRanges, dual, isdual
+using NamedDimsArrays: AbstractNamedUnitRange, dename, name, named
+
+function GradedUnitRanges.dual(r::AbstractNamedUnitRange)
+  return named(dual(dename(r)), dual(name(r)))
+end
+
+function GradedUnitRanges.isdual(r::AbstractNamedUnitRange)
+  return isdual(dename(r))
+end
+
+end

src/abstractnameddimsarray.jl

@@ -277,6 +277,10 @@ function Base.similar(
   return similar_nameddimsarray(a, elt, inds)
 end
 
+function Base.similar(a::AbstractArray, inds::NaiveOrderedSet)
+  return similar_nameddimsarray(a, eltype(a), inds)
+end
+
 function Base.similar(a::AbstractArray, elt::Type, inds::NaiveOrderedSet)
   return similar_nameddimsarray(a, elt, inds)
 end

test/Project.toml

@@ -4,10 +4,12 @@ Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
 BlockSparseArrays = "2c9a651f-6452-4ace-a6ac-809f4280fbb4"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
+GradedUnitRanges = "e2de450a-8a67-46c7-b59c-01d5a3d041c5"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 NamedDimsArrays = "60cbd0c0-df58-4cb7-918c-6f5607b73fde"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
+SymmetrySectors = "f8a8ad64-adbc-4fce-92f7-ffe2bb36a86e"
 TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 

test/test_basics.jl

@@ -70,12 +70,31 @@ using Test: @test, @test_throws, @testset
 
     for na′ in (
       similar(na, Float32, (j, i)),
+      similar(na, Float32, NaiveOrderedSet((j, i))),
       similar(na, Float32, (aj, ai)),
+      similar(na, Float32, NaiveOrderedSet((aj, ai))),
       similar(a, Float32, (j, i)),
+      similar(a, Float32, NaiveOrderedSet((j, i))),
       similar(a, Float32, (aj, ai)),
+      similar(a, Float32, NaiveOrderedSet((aj, ai))),
     )
       @test eltype(na′) === Float32
-      @test nameddimsindices(na′) == (j, i)
+      @test all(nameddimsindices(na′) .== (j, i))
+      @test na′ ≠ na
+    end
+
+    for na′ in (
+      similar(na, (j, i)),
+      similar(na, NaiveOrderedSet((j, i))),
+      similar(na, (aj, ai)),
+      similar(na, NaiveOrderedSet((aj, ai))),
+      similar(a, (j, i)),
+      similar(a, NaiveOrderedSet((j, i))),
+      similar(a, (aj, ai)),
+      similar(a, NaiveOrderedSet((aj, ai))),
+    )
+      @test eltype(na′) === eltype(na)
+      @test all(nameddimsindices(na′) .== (j, i))
       @test na′ ≠ na
     end
 

test/test_gradedunitrangesext.jl

@@ -0,0 +1,12 @@
+using GradedUnitRanges: dual, gradedrange, isdual
+using NamedDimsArrays: dename, named
+using SymmetrySectors: U1
+using Test: @test, @testset
+
+@testset "NamedDimsArraysGradedUnitRangesExt" begin
+  r = gradedrange([U1(0) => 2, U1(1) => 2])
+  nr = named(r, "i")
+  nr_dual = dual(nr)
+  @test isdual(nr_dual)
+  @test isdual(dename(nr_dual))
+end