more careful test of ByRow for PooledArray (#2837)

Author: bkamins

Project.toml

@@ -29,7 +29,7 @@ DataAPI = "1.8"
 InvertedIndices = "1"
 IteratorInterfaceExtensions = "0.1.1, 1"
 Missings = "0.4.2, 1"
-PooledArrays = "1.2.1"
+PooledArrays = "1.3.0"
 PrettyTables = "0.12, 1"
 Reexport = "0.1, 0.2, 1"
 SortingAlgorithms = "0.1, 0.2, 0.3, 1"

test/select.jl

@@ -1,6 +1,6 @@
 module TestSelect
 
-using DataFrames, Test, Random, Statistics, CategoricalArrays
+using DataFrames, Test, Random, Statistics, CategoricalArrays, PooledArrays
 
 const ≅ = isequal
 
@@ -1668,4 +1668,14 @@ end
     @test_throws ArgumentError select(gdf, [:c1, :c2] => AsTable)
 end
 
+@testset "ByRow on PooledArray calls function on each entry" begin
+    id = 0
+    df = DataFrame(a=PooledArray([1, 1, 1]))
+    function f(x)
+        id += 1
+        return id
+    end
+    @test select(df, :a => ByRow(f) => :a) == DataFrame(a=1:3)
+end
+
 end # module