Add rle support for BitArray (JuliaLang#511)

LilithHafner · Aug 7, 2019 · 416af55 · 416af55
1 parent 0faf210
commit 416af55
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/misc.jl b/src/misc.jl
@@ -18,7 +18,7 @@ julia> rle([1,1,1,2,2,3,3,3,3,2,2,2])
 ([1, 2, 3, 2], [3, 2, 4, 3])
 ```
 """
-function rle(v::Vector{T}) where T
+function rle(v::AbstractVector{T}) where T
     n = length(v)
     vals = T[]
     lens = Int[]

diff --git a/test/misc.jl b/test/misc.jl
@@ -15,6 +15,11 @@ vals, lens = rle(z)
 @test lens == [2, 2, 1, 1, 3]
 @test inverse_rle(vals, lens) == z
 
+z = BitArray([true, true, false, false, true])
+(vals, lens) = rle(z)
+@test vals == [true, false, true]
+@test lens == [2, 2, 1]
+
 # levelsmap
 a = [1, 1, 2, 2, 2, 3, 1, 2, 2, 3, 3, 3, 3, 2]
 b = [true, false, false, true, false, true, true, false]