blocklabels to return Vector

ITensor · Nov 7, 2024 · e44876a · e44876a
1 parent b66c22f
commit e44876a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/NDTensors/src/lib/GradedAxes/src/gradedunitrange.jl b/NDTensors/src/lib/GradedAxes/src/gradedunitrange.jl
@@ -68,7 +68,7 @@ end
 # == is just a range comparison that ignores labels. Need dedicated function to check equality.
 struct NoLabel end
 blocklabels(r::AbstractUnitRange) = Fill(NoLabel(), blocklength(r))
-blocklabels(la::LabelledUnitRange) = label(la)
+blocklabels(la::LabelledUnitRange) = [label(la)]
 
 function LabelledNumbers.labelled_isequal(a1::AbstractUnitRange, a2::AbstractUnitRange)
   return blockisequal(a1, a2) && (blocklabels(a1) == blocklabels(a2))

diff --git a/NDTensors/src/lib/GradedAxes/src/labelledunitrangedual.jl b/NDTensors/src/lib/GradedAxes/src/labelledunitrangedual.jl
@@ -12,7 +12,7 @@ nondual(a::LabelledUnitRangeDual) = a.nondual_unitrange
 dual(a::LabelledUnitRangeDual) = nondual(a)
 flip(a::LabelledUnitRangeDual) = dual(flip(nondual(a)))
 isdual(::LabelledUnitRangeDual) = true
-blocklabels(la::LabelledUnitRangeDual) = label(la)
+blocklabels(la::LabelledUnitRangeDual) = [label(la)]
 
 LabelledNumbers.label(a::LabelledUnitRangeDual) = dual(label(nondual(a)))
 LabelledNumbers.unlabel(a::LabelledUnitRangeDual) = unlabel(nondual(a))

diff --git a/NDTensors/src/lib/GradedAxes/test/test_dual.jl b/NDTensors/src/lib/GradedAxes/test/test_dual.jl
@@ -64,7 +64,7 @@ end
   la = labelled(1:2, U1(1))
   @test la isa LabelledUnitRange
   @test label(la) == U1(1)
-  @test blocklabels(la) == U1(1)
+  @test blocklabels(la) == [U1(1)]
   @test unlabel(la) == 1:2
   @test la == 1:2
   @test !isdual(la)
@@ -74,7 +74,7 @@ end
   lad = dual(la)
   @test lad isa LabelledUnitRangeDual
   @test label(lad) == U1(-1)
-  @test blocklabels(lad) == U1(-1)
+  @test blocklabels(lad) == [U1(-1)]
   @test unlabel(lad) == 1:2
   @test lad == 1:2
   @test labelled_isequal(lad, lad)
@@ -90,7 +90,7 @@ end
   lad = dual(la)
   @test lad isa LabelledUnitRangeDual
   @test label(lad) == 'x'
-  @test blocklabels(la) == 'x'
+  @test blocklabels(lad) == ['x']
   @test unlabel(lad) == 1:2
   @test lad == 1:2
   @test labelled_isequal(lad, lad)