mt schema

PharmCat · Aug 10, 2021 · 266367d · 266367d · PharmCat · Aug 11, 2021
1 parent 8d205d7
commit 266367d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/src/MetidaBase.jl b/src/MetidaBase.jl
@@ -9,7 +9,7 @@ module MetidaBase
     #@reexport using StatsModels
 
     import StatsModels: StatisticalModel
-    import Tables: istable, columnaccess, columns, getcolumn, columnnames
+    import Tables: istable, columnaccess, columns, getcolumn, columnnames, schema, rowaccess, rows
 
     import Base: getindex, length, ht_keyindex, show, pushfirst!, iterate, size
 

diff --git a/src/types.jl b/src/types.jl
@@ -64,6 +64,8 @@ Tables.getcolumn(t::MetidaTableRow, nm::Symbol) = getfield(t, :source)[getfield(
 
 Tables.columnnames(t::MetidaTableRow) = names(getfield(t, :source))
 
+Tables.schema(t::MetidaTable) = Tables.Schema(names(t), eltype(y) for y in t.table)
+
 ################################################################################
 # BASE
 ################################################################################

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -94,4 +94,6 @@ using Test, DataFrames
     MetidaBase.varfromcv(0.4) ≈ 0.1484200051182734
     MetidaBase.cvfromvar(0.4) ≈ 0.7013021443295824
     MetidaBase.cvfromsd(0.4) ≈ 0.41654636115540644
+
+    smt = MetidaBase.Tables.schema(mt)
 end