special-case AsTable

JuliaData · Oct 31, 2023 · a12716b · a12716b
1 parent 51c2792
commit a12716b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/parsing.jl b/src/parsing.jl
@@ -21,7 +21,7 @@ returns `nothing`.
 get_column_expr(x) = nothing
 function get_column_expr(e::Expr)
     e.head == :$ && return e.args[1]
-    onearg(e, :AsTable) && return e
+    onearg(e, :AsTable) && return :($AsTable($(e.args[2])))
     if onearg(e, :cols)
         Base.depwarn("cols is deprecated use $DOLLAR to escape column names instead", :cols)
         return e.args[2]

diff --git a/test/import.jl b/test/import.jl
@@ -20,8 +20,8 @@ import DataFramesMeta
     @test t == DataFrames.DataFrame(a = [1, 2, 3], b = [1, 1, 1], c  = [2, 2, 2])
 
     # AsTable on the RHS relies on the literal "AsTable" appearing
-    # so we test that this throws an error
-    @test_throws UndefVarError t = DataFramesMeta.@transform df :c = sum(AsTable([:a]))
+    t = DataFramesMeta.@rtransform df :c = sum(AsTable([:a]))
+    @test t == DataFrames.DataFrame(a = [1, 2, 3], c = [1, 2, 3])
 
     # And confusingly, if you use DataFrames.AsTable on the RHS, none of the
     # special escaping happens.