Add Select/Reject constructors with vector

Author: juliohm

src/transforms/select.jl

@@ -15,6 +15,8 @@ end
 Select(cols::NTuple{N,AbstractString}) where {N} =
   Select(Symbol.(cols))
 
+Select(cols::AbstractVector) = Select(Tuple(cols))
+
 Select(cols...) = Select(cols)
 
 isrevertible(::Type{<:Select}) = true
@@ -84,6 +86,8 @@ end
 Reject(cols::NTuple{N,AbstractString}) where {N} =
   Reject(Symbol.(cols))
 
+Reject(cols::AbstractVector) = Reject(Tuple(cols))
+
 Reject(cols...) = Reject(cols)
 
 isrevertible(::Type{<:Reject}) = true

test/transforms.jl

@@ -31,6 +31,32 @@
     @test Tables.columnnames(n) == (:e, :c, :b, :a)
     tₒ = revert(T, n, c)
     @test t == tₒ
+
+    # selection with tuples
+    T = Select((:e, :c, :b, :a))
+    n, c = apply(T, t)
+    @test Tables.columnnames(n) == (:e, :c, :b, :a)
+    tₒ = revert(T, n, c)
+    @test t == tₒ
+
+    # selection with vectors
+    T = Select([:e, :c, :b, :a])
+    n, c = apply(T, t)
+    @test Tables.columnnames(n) == (:e, :c, :b, :a)
+    tₒ = revert(T, n, c)
+    @test t == tₒ
+
+    # selection with strings
+    T = Select("d", "c", "b")
+    n, c = apply(T, t)
+    @test Tables.columnnames(n) == (:d, :c, :b)
+    tₒ = revert(T, n, c)
+    @test t == tₒ
+
+    # selection with single column
+    @test (Select(:a) == Select("a") ==
+           Select((:a,)) == Select(("a",)) ==
+           Select([:a]) == Select(["a"]))
   end
 
   @testset "Reject" begin
@@ -65,6 +91,32 @@
     @test Tables.columnnames(n) == (:d, :f)
     tₒ = revert(T, n, c)
     @test t == tₒ
+
+    # rejection with tuples
+    T = Reject((:e, :c, :b, :a))
+    n, c = apply(T, t)
+    @test Tables.columnnames(n) == (:d, :f)
+    tₒ = revert(T, n, c)
+    @test t == tₒ
+
+    # rejection with vectors
+    T = Reject([:e, :c, :b, :a])
+    n, c = apply(T, t)
+    @test Tables.columnnames(n) == (:d, :f)
+    tₒ = revert(T, n, c)
+    @test t == tₒ
+
+    # rejection with strings
+    T = Reject("d", "c", "b")
+    n, c = apply(T, t)
+    @test Tables.columnnames(n) == (:a, :e, :f)
+    tₒ = revert(T, n, c)
+    @test t == tₒ
+
+    # rejection with single column
+    @test (Reject(:a) == Reject("a") ==
+           Reject((:a,)) == Reject(("a",)) ==
+           Reject([:a]) == Reject(["a"]))
   end
 
   @testset "Identity" begin