Merge pull request #265 from jonniedie/another-lazy-array-thing

Make `haskey` work for `LazyArray`
jonniedie · Jul 26, 2024 · f5574a5 · f5574a5 · jonniediegelman · Jul 26, 2024
2 parents 664eb73 + 4fdfe36
commit f5574a5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "ComponentArrays"
 uuid = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 authors = ["Jonnie Diegelman <[email protected]>"]
-version = "0.15.15"
+version = "0.15.16"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

diff --git a/src/lazyarray.jl b/src/lazyarray.jl
@@ -30,6 +30,8 @@ Base.propertynames(a::LazyArray) = propertynames(first(a))
 
 Base.keys(a::LazyArray) = Base.OneTo(length(a))
 
+Base.haskey(a::LazyArray, i::Integer) = i in keys(a)
+
 Base.iterate(a::LazyArray) = iterate(getfield(a, :gen))
 Base.iterate(a::LazyArray, state...) = iterate(getfield(a, :gen), state...)
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -345,6 +345,9 @@ end
     @test propertynames(ca2) == (:a, :b, :c) # ComponentArray
     @test propertynames(ca2.b) == (:a, :b) # LazyArray
 
+    @test haskey(ca2, :a) # ComponentArray
+    @test haskey(ca2.b, 1) # LazyArray
+
     @test keys(ca2) == (:a, :b, :c)
     @test keys(ca2.b) == Base.OneTo(2)
 end