Merge pull request #194 from JuliaStats/alyst/repeat_methods

repeat() methods for DataArray and PooledDataArray

src/extras.jl

@@ -62,3 +62,17 @@ function cut{S, T}(x::AbstractVector{S}, breaks::Vector{T})
 end
 
 cut(x::AbstractVector, ngroups::Integer) = cut(x, quantile(x, [1 : ngroups - 1] / ngroups))
+
+function Base.repeat{T,N}(A::DataArray{T,N};
+                          inner::Array{Int} = ones(Int, ndims(A)),
+                          outer::Array{Int} = ones(Int, ndims(A)))
+    DataArray{T,N}(Compat.repeat(A.data; inner=inner, outer=outer),
+                   bitpack(Compat.repeat(A.na; inner=inner, outer=outer)))
+end
+
+function Base.repeat{T,R,N}(A::PooledDataArray{T,R,N};
+                            inner::Array{Int} = ones(Int, ndims(A)),
+                            outer::Array{Int} = ones(Int, ndims(A)))
+    PooledDataArray(RefArray{R,N}(Compat.repeat(A.refs; inner=inner, outer=outer)),
+                    A.pool)
+end

test/extras.jl

@@ -35,4 +35,18 @@ module TestExtras
                            "(35,60]", "(25,35]", "(60,100]",
                            "(35,60]", "(35,60]", "(25,35]"])
     @assert isequal(cats, pdv)
+
+    ##########
+    ## repeat
+    ##########
+
+    @test isequal(repeat(@data [3.0, 2.0, NA]; inner = [2], outer = [1]),
+                  @data [3.0, 3.0, 2.0, 2.0, NA, NA])
+    @test isequal(repeat(@pdata ["a", "b", NA]; inner = [2], outer = [1]),
+                  @pdata ["a", "a", "b", "b", NA, NA])
+    @test isequal(repeat(@data [1 2; 3 NA]; inner = [1, 2], outer = [2, 1]),
+                  @data [1 1 2 2; 3 3 NA NA; 1 1 2 2; 3 3 NA NA])
+    @test isequal(repeat(@pdata [:a :b NA]; inner = [2,1], outer = [1,3]),
+                  @pdata [:a :b NA :a :b NA :a :b NA;
+                          :a :b NA :a :b NA :a :b NA])
 end