add propertynames

Author: KristofferC

src/structarray.jl

@@ -44,6 +44,8 @@ setindex!(s::StructArray, val, I::Int...) = set_ith!(s, val, I...)
 fields(::Type{<:NamedTuple{K}}) where {K} = K
 fields(::Type{<:StructArray{T}}) where {T} = fields(T)
 
+Base.propertynames(s::StructArray) = fieldnames(typeof(columns(s)))
+
 @generated function fields(t::Type{T}) where {T}
    return :($(Expr(:tuple, [QuoteNode(f) for f in fieldnames(T)]...)))
 end

test/runtests.jl

@@ -36,3 +36,8 @@ f_infer() = StructArray{ComplexF64}(rand(2,2), rand(2,2))
 @testset "inferrability" begin
     @inferred f_infer()
 end
+
+@testset "propertynames" begin
+    a = StructArray{ComplexF64}(Float64[], Float64[])
+    @test sort(collect(propertynames(a))) == [:im, :re]
+end