github.com/benz9527/toy-box/algo@v0.0.0-20240221120937-66c0c6bd5abd/bitmap/x_bitmap_test.go (about)

     1  package bitmap
     2  
     3  import (
     4  	"github.com/stretchr/testify/assert"
     5  	"testing"
     6  )
     7  
     8  func TestNewX32Bitmap(t *testing.T) {
     9  	bm := NewX32Bitmap(10)
    10  	bm2 := NewX32Bitmap(10)
    11  	originalOffsets := []uint64{9, 5, 7, 3, 2, 8, 1}
    12  	expectedOffsets := []uint64{1, 2, 3, 5, 7, 8, 9}
    13  	for _, offset := range originalOffsets {
    14  		bm.SetBit(offset, true)
    15  		bm2.SetBit(offset, true)
    16  	}
    17  	for _, offset := range expectedOffsets {
    18  		assert.True(t, bm.GetBit(offset))
    19  	}
    20  	assert.True(t, bm.EqualTo(bm2))
    21  }