improve tables support

JuliaArrays · Jun 17, 2023 · 6b23237 · 6b23237
1 parent e2f0667
commit 6b23237
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/tables.jl b/src/tables.jl
@@ -12,6 +12,9 @@ function _schema(::Type{T}) where {T<:NTuple{N, Any}} where N
     return Tables.Schema{ntuple(identity, N), T}
 end
 
+StructArray(cols::Tables.AbstractColumns) = StructArray(Tables.columntable(cols))
+StructArray{T}(cols::Tables.AbstractColumns) where {T} = StructArray{T}(Tables.columntable(cols))
+
 function try_compatible_columns(rows::R, s::StructArray) where {R}
     Tables.isrowtable(rows) && Tables.columnaccess(rows) || return nothing
     T = eltype(rows)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -714,6 +714,11 @@ end
     # Testing integer column "names":
     @test invoke(append!, Tuple{StructVector,Any}, StructArray(([0],)), StructArray(([1],))) ==
         StructArray(([0, 1],))
+
+    dtab = (a=[1,2],) |> Tables.dictcolumntable
+    @test StructArray(dtab) == [(a=1,), (a=2,)]
+    @test StructArray{NamedTuple{(:a,), Tuple{Float64}}}(dtab) == [(a=1.,), (a=2.,)]
+    @test StructVector{NamedTuple{(:a,), Tuple{Float64}}}(dtab) == [(a=1,), (a=2,)]
 end
 
 struct S