Make haskey work for LazyArray

SciML · Jul 26, 2024 · 4fdfe36 · 4fdfe36
1 parent 664eb73
commit 4fdfe36
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