fix test

Author: ogauthe

NDTensors/src/lib/GradedAxes/src/gradedunitrangedual.jl

@@ -57,23 +57,23 @@ end
 function blockedunitrange_getindices(
   a::GradedUnitRangeDual, indices::Vector{<:BlockIndexRange{1}}
 )
-  # flip v to stay consistent with other cases where axes(v) are used
+  # dual v axes to stay consistent with other cases where axes(v) are used
   return dual_axes(blockedunitrange_getindices(nondual(a), indices))
 end
 
 function blockedunitrange_getindices(
   a::GradedUnitRangeDual,
   indices::BlockVector{<:BlockIndex{1},<:Vector{<:BlockIndexRange{1}}},
 )
-  # flip v axis to preserve dual information
+  # dual v axis to preserve dual information
   # axes(v) will appear in axes(view(::BlockSparseArray, [Block(1)[1:1]]))
   return dual_axes(blockedunitrange_getindices(nondual(a), indices))
 end
 
 function blockedunitrange_getindices(
   a::GradedUnitRangeDual, indices::AbstractVector{<:Union{Block{1},BlockIndexRange{1}}}
 )
-  # flip v axis to preserve dual information
+  # dual v axis to preserve dual information
   # axes(v) will appear in axes(view(::BlockSparseArray, [Block(1)]))
   return dual_axes(blockedunitrange_getindices(nondual(a), indices))
 end
@@ -82,7 +82,10 @@ end
 function blockedunitrange_getindices(
   a::GradedUnitRangeDual, indices::AbstractBlockVector{<:Block{1}}
 )
-  return dual_axes(blockedunitrange_getindices(nondual(a), indices))
+  v = blockedunitrange_getindices(nondual(a), indices)
+  # v elements are not dualled by dual_axes due to different structure.
+  # take element dual here.
+  return dual_axes(dual.(v))
 end
 
 function dual_axes(v::BlockVector)

NDTensors/src/lib/GradedAxes/test/test_dual.jl

@@ -219,15 +219,35 @@ end
     @test label(ad[Block(2)]) == U1(-1)
     @test label(ad[Block(2)[1:1]]) == U1(-1)
 
-    I = mortar([Block(2)[1:1]])
-    g = ad[I]
-    @test length(g) == 1
-    @test label(first(g)) == U1(-1)
-    @test isdual(g[Block(1)])
+    v = ad[mortar([Block(2)[1:1]])]
+    @test v isa AbstractVector{LabelledInteger{Int64,U1}}
+    @test length(v) == 1
+    @test label(first(v)) == U1(-1)
+    @test unlabel(first(v)) == 3
+    @test isdual(v[Block(1)])
+    @test isdual(axes(v, 1))
+    @test blocklabels(axes(v, 1)) == [U1(-1)]
 
-    @test isdual(axes(ad[[Block(1)]], 1))  # used in view(::BlockSparseVector, [Block(1)])
-    @test isdual(axes(ad[mortar([Block(1)[1:1]])], 1))  # used in view(::BlockSparseVector, [Block(1)[1:1]])
-    @test isdual(axes(ad[mortar([[Block(1)], [Block(2)]])]))
+    v = ad[[Block(2)]]
+    @test v isa AbstractVector{LabelledInteger{Int64,U1}}
+    @test isdual(axes(v, 1))  # used in view(::BlockSparseVector, [Block(1)])
+    @test label(first(v)) == U1(-1)
+    @test unlabel(first(v)) == 3
+    @test blocklabels(axes(v, 1)) == [U1(-1)]
+
+    v = ad[mortar([Block(2)[1:1]])]
+    @test v isa AbstractVector{LabelledInteger{Int64,U1}}
+    @test isdual(axes(v, 1))  # used in view(::BlockSparseVector, [Block(1)[1:1]])
+    @test label(first(v)) == U1(-1)
+    @test unlabel(first(v)) == 3
+    @test blocklabels(axes(v, 1)) == [U1(-1)]
+
+    v = ad[mortar([[Block(2)], [Block(1)]])]
+    @test v isa AbstractVector{LabelledInteger{Int64,U1}}
+    @test isdual(axes(v, 1))
+    @test label(first(v)) == U1(-1)
+    @test unlabel(first(v)) == 3
+    @test blocklabels(axes(v, 1)) == [U1(-1), U1(0)]
   end
 end