Fix tests

Author: mtfishman

src/sitetype.jl

@@ -79,7 +79,6 @@ function Base.AbstractUnitRange(t::SiteType)
   end
   return Base.OneTo(length(t))
 end
-# kwargs are passed for fancier constructors, like `ITensors.Index`.
 function (rangetype::Type{<:AbstractUnitRange})(t::SiteType)
   return rangetype(AbstractUnitRange(t))
 end

test/test_itensorbaseext.jl

@@ -1,3 +1,4 @@
+using GradedUnitRanges: GradedUnitRanges
 using ITensorBase: ITensor, Index, gettag, hastag, prime, settag
 using NamedDimsArrays: dename
 using QuantumOperatorDefinitions: OpName, SiteType, StateName, op, site, sites, state

test/test_symmetrysectorsext.jl

@@ -1,6 +1,6 @@
 using BlockArrays: AbstractBlockArray, blocklengths
 using BlockSparseArrays: BlockSparseArray
-using GradedUnitRanges: blocklabels
+using GradedUnitRanges: blocklabels, dual, isdual
 using ITensorBase: ITensor, Index, gettag, prime, settag
 using QuantumOperatorDefinitions: OpName, SiteType, StateName, op, state
 using SymmetrySectors: SectorProduct, U1, Z
@@ -64,25 +64,19 @@ using Test: @test, @test_broken, @testset
   t = SiteType("S=1/2"; gradings=("Sz",))
   a = AbstractArray(2.0 * StateName("0"), t)
   @test a == [2, 0]
-  @test a isa AbstractBlockArray
-  # TODO: Currently slicing a dense array by graded ranges outputs a `BlockedArray`
-  # rather than a `BlockSparseArray`.
-  # See: https://github.com/ITensor/GradedUnitRanges.jl/issues/9
-  @test_broken a isa BlockSparseArray
+  @test a isa BlockSparseArray
   (r1,) = axes(a)
   @test blocklabels(r1) == [SectorProduct((; Sz=U1(0))), SectorProduct((; Sz=U1(1)))]
   @test blocklengths(r1) == [1, 1]
 
   t = SiteType("S=1/2"; gradings=("Sz",))
   a = op("σ⁺", t)
   @test a == [0 2; 0 0]
-  @test a isa AbstractBlockArray
-  @test_broken a isa BlockSparseArray
+  @test a isa BlockSparseArray
   (r1, r2) = axes(a)
   @test blocklabels(r1) == [SectorProduct((; Sz=U1(0))), SectorProduct((; Sz=U1(1)))]
   @test blocklengths(r1) == [1, 1]
-  # TODO: This is a bug in indexing with GradedUnitRangeDual, fix this.
-  @test_broken blocklabels(r2) ==
+  @test blocklabels(r2) ==
     [SectorProduct((; Sz=U1(0))), SectorProduct((; Sz=U1(-1)))]
   @test blocklengths(r2) == [1, 1]
 end
@@ -97,17 +91,15 @@ end
 
   i′ = prime(i)
   a = op("σ⁺", i)
-  # TODO: The indices should be `(i′, dual(i))`.
-  @test a == ITensor([0 2; 0 0], (i′, i))
+  @test a == ITensor([0 2; 0 0], (i′, dual(i)))
+  @test all(isdual.(axes(a)) .== (false, true))
   a′ = dename(a)
-  @test a′ isa AbstractBlockArray
-  @test_broken a′ isa BlockSparseArray
+  @test a′ isa BlockSparseArray
   # TODO: Test these without denaming `a`.
   (r1, r2) = axes(a′)
   @test blocklabels(r1) == [SectorProduct((; Sz=U1(0))), SectorProduct((; Sz=U1(1)))]
   @test blocklengths(r1) == [1, 1]
-  # TODO: This is a bug in indexing with GradedUnitRangeDual, fix this.
-  @test_broken blocklabels(r2) ==
+  @test blocklabels(r2) ==
     [SectorProduct((; Sz=U1(0))), SectorProduct((; Sz=U1(-1)))]
   @test blocklengths(r2) == [1, 1]
 end