Define checkindex with AbstractFill only for AbstractInfUnitRange (#220)

* Check for flag before defining checkindex

* Add leading underscore

* Check `hasmethod` instead of a variable

* Define method only for AbstractInfUnitRange

* Tests for bounds checking

* Fix test

Author: jishnub

Project.toml

@@ -1,6 +1,6 @@
 name = "InfiniteArrays"
 uuid = "4858937d-0d70-526a-a4dd-2d5cb5dd786c"
-version = "0.15.8"
+version = "0.15.9"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/infarrays.jl

@@ -428,7 +428,8 @@ Base._unsafe_getindex(::IndexStyle, A::AbstractFill{<:Any,2}, kr::InfAxes, jr::U
 
 @inline getindex(A::ApplyMatrix{<:Any,typeof(hcat)}, kr::InfAxes, j::Integer) = layout_getindex(A, kr, j)
 
-Base.checkindex(::Type{Bool}, inds::AbstractUnitRange, I::AbstractFill) = Base.checkindex(Bool, inds, getindex_value(I))
+Base.checkindex(::Type{Bool}, inds::AbstractInfUnitRange, I::AbstractFill) = Base.checkindex(Bool, inds, getindex_value(I))
+Base.checkindex(::Type{Bool}, inds::AbstractInfUnitRange, I::AbstractFill{Bool}) = axes(I,1) == inds
 LazyArrays.cache_getindex(::InfiniteCardinal{0}, A::AbstractVector, I, J...) = layout_getindex(A, I, J...)
 LazyArrays.cache_getindex(::InfiniteCardinal{0}, A::CachedVector{<:Any,<:AbstractVector,<:AbstractFill{<:Any,1}}, I::AbstractVector) = LazyArrays.cache_getindex(nothing, A, I)
 

test/runtests.jl

@@ -1286,6 +1286,14 @@ end
     @test (a*Hcat(Zeros(1,2), permutedims(1:∞)))[1,1:5] == (a*Vcat(Hcat(Zeros(1,2), permutedims(1:∞))))[1,1:5]
 end
 
+@testset "checkindex" begin
+    r = 1:∞
+    for v in (4, true)
+        f = Fill(v, ∞)
+        @test checkbounds(Bool, r, f)
+    end
+end
+
 include("test_infconv.jl")
 include("test_infblock.jl")
 include("test_infbanded.jl")