Add checkindex for Blocks (#111)

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "BlockArrays"
 uuid = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
-version = "0.12.4"
+version = "0.12.5"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/blockaxis.jl

@@ -187,6 +187,9 @@ Base.dataids(b::BlockedUnitRange) = Base.dataids(blocklasts(b))
 ###
 # BlockedUnitRange interface
 ###
+Base.checkindex(::Type{Bool}, b::BlockRange, K::Int) = checkindex(Bool, Int.(b), K)
+Base.checkindex(::Type{Bool}, b::AbstractUnitRange{Int}, K::Block{1}) = checkindex(Bool, blockaxes(b,1), Int(K))
+
 function getindex(b::AbstractUnitRange{Int}, K::Block{1})
     @boundscheck K == Block(1) || throw(BlockBoundsError(b, K))
     b

test/test_blockindices.jl

@@ -239,6 +239,14 @@ end
     @testset "BlockIndex type piracy (#108)" begin
         @test zeros()[] == 0.0
     end
+
+    @testset "checkindex" begin
+        b = blockedrange([1,2,3])
+        @test !checkindex(Bool, b, Block(0))
+        @test checkindex(Bool, b, Block(1))
+        @test checkindex(Bool, b, Block(3))
+        @test !checkindex(Bool, b, Block(4))
+    end
 end
 
 @testset "BlockSlice" begin