fix tests

Author: ogauthe

NDTensors/src/lib/BlockSparseArrays/ext/BlockSparseArraysGradedAxesExt/test/runtests.jl

@@ -200,14 +200,14 @@ const elts = (Float32, Float64, Complex{Float32}, Complex{Float64})
       b = 2 * a
       @test block_nstored(b) == 2
       @test Array(b) == 2 * Array(a)
-      @test_broken a[:, :] isa BlockSparseArray
+      @test a[:, :] isa BlockSparseArray
       for ax in axes(b)
-        @test ax isa BlockedUnitRange
+        @test ax isa BlockedOneTo
       end
 
       I = [Block(1)[1:1]]
-      @test_broken a[I, :]
-      @test_broken a[:, I]
+      @test a[I, :] isa BlockSparseArray
+      @test a[:, I] isa BlockSparseArray
       @test size(a[I, I]) == (1, 1)
       @test !GradedAxes.isdual(axes(a[I, I], 1))
     end

NDTensors/src/lib/GradedAxes/src/gradedunitrange.jl

@@ -177,7 +177,7 @@ end
 function gradedunitrange_blockfirsts(a::AbstractGradedUnitRange)
   return labelled.(blockfirsts(unlabel_blocks(a)), blocklabels(a))
 end
-function BlockArrays.blockfirsts(a::AbstractGradedUnitRange)
+function BlockArrays.blockfirsts(a::GradedUnitRange)
   return gradedunitrange_blockfirsts(a)
 end
 function BlockArrays.blockfirsts(a::GradedOneTo)

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

@@ -106,7 +106,7 @@ end
   @test length(ax) == length(a)
   @test blocklengths(ax) == blocklengths(a)
   @test blocklabels(ax) == blocklabels(a)
-  @test_broken(blockfirsts(a)) == [2, 3]
+  @test blockfirsts(a) == [2, 3]
 
   # Regression test for ambiguity error.
   x = gradedrange(["x" => 2, "y" => 3])

NDTensors/src/lib/TensorAlgebra/ext/TensorAlgebraGradedAxesExt/test/test_contract.jl

@@ -2,13 +2,16 @@
 using BlockArrays: Block, blocksize
 using Compat: Returns
 using NDTensors.BlockSparseArrays: BlockSparseArray
-using NDTensors.GradedAxes: gradedrange
+using NDTensors.GradedAxes: GradedAxes, gradedrange
 using NDTensors.Sectors: U1
 using NDTensors.SparseArrayInterface: densearray
 using NDTensors.TensorAlgebra: contract
 using Random: randn!
 using Test: @test, @testset
 
+#TODO remove once fuse_labels is defined in Sectors
+GradedAxes.fuse_labels(m::U1, n::U1) = U1(m.n + n.n)
+
 function randn_blockdiagonal(elt::Type, axes::Tuple)
   a = BlockSparseArray{elt}(axes)
   blockdiaglength = minimum(blocksize(a))