Fix union!(s::BitSet, r::AbstractUnitRange{<:Integer}) when two range…

…s do not overlap. Resizing of BitSet should be filled with 0 by default.
JuliaLang · Jun 4, 2022 · fd7c29c · fd7c29c
1 parent bd8dbc3
commit fd7c29c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/base/bitset.jl b/base/bitset.jl
@@ -137,7 +137,7 @@ function union!(s::BitSet, r::AbstractUnitRange{<:Integer})
 
     # grow s.bits as necessary
     if diffb >= len
-        _growend!(s.bits, diffb - len + 1)
+        _growend0!(s.bits, diffb - len + 1)
         # we set only some values to CHK0, those which will not be
         # fully overwritten (i.e. only or'ed with `|`)
         s.bits[end] = CHK0 # end == diffb + 1
@@ -146,7 +146,7 @@ function union!(s::BitSet, r::AbstractUnitRange{<:Integer})
         end
     end
     if diffa < 0
-        _growbeg!(s.bits, -diffa)
+        _growbeg0!(s.bits, -diffa)
         s.bits[1] = CHK0
         if diffb < 0
             s.bits[diffb - diffa + 1] = CHK0

diff --git a/test/bitset.jl b/test/bitset.jl
@@ -351,3 +351,12 @@ end
     # union! with an empty range doesn't modify the BitSet
     @test union!(x, b:a) == y
 end
+
+@testset "union!(::BitSet, ::AbstractUnitRange) when two ranges do not overlap" begin
+    # see #45574
+    a, b = minmax(rand(-1000:1000, 2)...)
+    c, d = minmax(rand(2000:3000, 2)...)
+    @test length(union!(BitSet(a:b), c:d)) == length(a:b) + length(c:d)
+    c, d = minmax(rand(-3000:-2000, 2)...)
+    @test length(union!(BitSet(a:b), c:d)) == length(a:b) + length(c:d)
+end