add vector support for @pivot_longer

Author: cnrrobertson

src/parsing.jl

@@ -32,6 +32,10 @@ function parse_tidy(tidy_expr::Union{Expr,Symbol,Number}; # Can be symbol or exp
       endindex = QuoteNode(endindex)
     end
     return :(Between($startindex, $endindex))
+  elseif @capture(tidy_expr, names_vect)
+    return Symbol.(names.args)
+  elseif @capture(tidy_expr, -names_vect)
+    return Not(Symbol.(names.args))
   elseif @capture(tidy_expr, (lhs_ = fn_(args__)) | (lhs_ = fn_.(args__)))
     if length(args) == 0
       lhs = QuoteNode(lhs)
@@ -499,4 +503,4 @@ function parse_blocks(exprs...)
     return (MacroTools.rmlines(exprs[1]).args...,)
   end
   return exprs
-end
+end

test/test_pivots.jl

@@ -30,9 +30,27 @@ end
     )
     test_long3 = @pivot_longer(test_df, -(name:num))
     test_long4 = @pivot_longer(test_df, label)
+
+    true_long5 = DataFrame(
+        name = ["A","B","A","B","A","B","A","B"],
+        variable = ["label","label","label","label","num","num","num","num"],
+        value = [1,1,2,2,1,2,3,4],
+    )
+    test_long5 = @pivot_longer(test_df, [label,num])
+    
+    true_long6 = DataFrame(
+        label = [1,1,2,2],
+        num = [1,2,3,4],
+        variable = ["name","name","name","name"],
+        value = ["A","B","A","B"],
+    )
+    test_long6 = @pivot_longer(test_df, -[label,num])
+
     @test all(Array(true_long1 .== test_long1))
     @test all(Array(true_long1 .== test_long2))
     @test all(Array(true_long3 .== test_long3))
     @test all(Array(true_long3 .== test_long4))
+    @test all(Array(true_long5 .== test_long5))
+    @test all(Array(true_long6 .== test_long6))
 end
 end