add deleteat! (#176)

JuliaData · Dec 11, 2018 · 6a232bc · 6a232bc
1 parent eebecd2
commit 6a232bc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/array.jl b/src/array.jl
@@ -790,3 +790,5 @@ end
 
 refs(A::CategoricalArray) = A.refs
 pool(A::CategoricalArray) = A.pool
+
+Base.deleteat!(A::CategoricalArray, inds) = (deleteat!(A.refs, inds); A)
diff --git a/test/13_arraycommon.jl b/test/13_arraycommon.jl
@@ -1094,4 +1094,13 @@ end
     end
 end
 
+@testset "deleteat!" begin
+    x = ['a':'z';]
+    y = categorical(x)
+    deleteat!(x, [1, 3])
+    deleteat!(y, [1, 3])
+    @test x == y
+    @test levels(y) == 'a':'z'
+end
+
 end
Original file line number	Diff line number	Diff line change
Expand Up		@@ -790,3 +790,5 @@ end

		refs(A::CategoricalArray) = A.refs
		pool(A::CategoricalArray) = A.pool

		Base.deleteat!(A::CategoricalArray, inds) = (deleteat!(A.refs, inds); A)