Test blockrange

Author: mtfishman

test/test_basics.jl

@@ -22,12 +22,14 @@ using BlockSparseArrays:
   BlockSparseMatrix,
   BlockSparseVector,
   BlockView,
-  blockstoredlength,
+  blockrange,
   blockreshape,
-  eachblockstoredindex,
-  eachstoredblock,
+  blockstoredlength,
   blockstype,
   blocktype,
+  eachblockaxis,
+  eachblockstoredindex,
+  eachstoredblock,
   sparsemortar,
   view!
 using GPUArraysCore: @allowscalar

test/test_blockrange.jl

@@ -0,0 +1,16 @@
+using BlockArrays: Block, blocklength
+using BlockSparseArrays: blockrange, eachblockaxis
+using Test: @test, @testset
+
+@testset "blockrange" begin
+  r = blockrange(AbstractUnitRange{Int}[Base.OneTo(3), 1:4])
+  @test eachblockaxis(r) == [Base.OneTo(3), 1:4]
+  @test eachblockaxis(r)[1] === Base.OneTo(3)
+  @test eachblockaxis(r)[2] === 1:4
+  @test r[Block(1)] == 1:3
+  @test r[Block(2)] == 4:7
+  @test first(r) == 1
+  @test last(r) == 7
+  @test blocklength(r) == 2
+  @test r == 1:7
+end