From 6b41a02e287eaed8d0e1c956497e2948cf85f501 Mon Sep 17 00:00:00 2001 From: Jonathan Malmaud Date: Fri, 27 Aug 2021 19:44:10 -0700 Subject: [PATCH] Enable Tables accessors --- src/tables.jl | 4 ++++ test/test_tables.jl | 5 +++++ 2 files changed, 9 insertions(+) diff --git a/src/tables.jl b/src/tables.jl index a8f30ab..876275a 100644 --- a/src/tables.jl +++ b/src/tables.jl @@ -8,3 +8,7 @@ function _construct_pandas_from_tables(source) source_as_dict = Dict(column => Tables.getcolumn(source_columns, column) for column in Tables.columnnames(source_columns)) return invoke(DataFrame, Tuple{Vararg{Any}}, source_as_dict) end + +Tables.columnaccess(::DataFrame) = true +Tables.rowaccess(::DataFrame) = true +Tables.istable(::DataFrame) = true diff --git a/test/test_tables.jl b/test/test_tables.jl index ddcfdd8..7eae9ad 100644 --- a/test/test_tables.jl +++ b/test/test_tables.jl @@ -19,4 +19,9 @@ df_cols = Tables.columns(df) @test Tables.getcolumn(df_cols, :Gr) == [1, 1, 1] @test Tables.getcolumn(df_cols, :Val) == [7863.0, 7834.0, 7803.0] @test Tables.getcolumn(df_cols, :Temp) == [20.0, 100.0, 200.0] + +@test Tables.rowaccess(df) +@test Tables.columnaccess(df) +@test Tables.istable(df) + end