package bits

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestWriteBits(t *testing.T) {
	buf := make([]byte, 6)
	pos := 0
	WriteBits(buf, &pos, uint64(0x2a), 6)
	WriteBits(buf, &pos, uint64(0x0c), 6)
	WriteBits(buf, &pos, uint64(0x1f), 6)
	WriteBits(buf, &pos, uint64(0x5a), 8)
	WriteBits(buf, &pos, uint64(0xaaec4), 20)
	require.Equal(t, []byte{0xA8, 0xC7, 0xD6, 0xAA, 0xBB, 0x10}, buf)
}