95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package rlepluslazy
|
|
|
|
import (
|
|
"math/rand"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSumRuns(t *testing.T) {
|
|
{
|
|
a, err := RunsFromSlice([]uint64{0, 1, 2, 3, 4, 5, 6, 7, 8, 11, 12, 13, 14})
|
|
assert.NoError(t, err)
|
|
b, err := RunsFromSlice([]uint64{0, 1, 2, 3, 9, 10, 16, 17, 18, 50, 51, 70})
|
|
assert.NoError(t, err)
|
|
|
|
s, err := Sum(a, b)
|
|
assert.NoError(t, err)
|
|
bis, err := SliceFromRuns(s)
|
|
assert.Equal(t, []uint64{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 16, 17, 18, 50, 51, 70}, bis)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
{
|
|
a, err := RunsFromSlice([]uint64{0, 1, 2, 3, 4, 5, 6, 7, 8, 11, 12, 13, 14})
|
|
assert.NoError(t, err)
|
|
b, err := RunsFromSlice([]uint64{0, 1, 2, 3, 9, 10, 16, 17, 18, 50, 51, 70})
|
|
assert.NoError(t, err)
|
|
|
|
s, err := Sum(b, a)
|
|
assert.NoError(t, err)
|
|
bis, err := SliceFromRuns(s)
|
|
assert.Equal(t, []uint64{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 16, 17, 18, 50, 51, 70}, bis)
|
|
assert.NoError(t, err)
|
|
}
|
|
}
|
|
|
|
func randomBits(N int, max uint64) []uint64 {
|
|
all := make(map[uint64]struct{})
|
|
for len(all) <= N {
|
|
x := rand.Uint64() % max
|
|
if _, has := all[x]; has {
|
|
continue
|
|
}
|
|
all[x] = struct{}{}
|
|
}
|
|
|
|
res := make([]uint64, 0, N)
|
|
for x := range all {
|
|
res = append(res, x)
|
|
}
|
|
sort.Slice(res, func(i, j int) bool { return res[i] < res[j] })
|
|
|
|
return res
|
|
}
|
|
|
|
func sum(a, b []uint64) []uint64 {
|
|
all := make(map[uint64]struct{})
|
|
for _, x := range a {
|
|
all[x] = struct{}{}
|
|
}
|
|
for _, x := range b {
|
|
all[x] = struct{}{}
|
|
}
|
|
res := make([]uint64, 0, len(all))
|
|
for x := range all {
|
|
res = append(res, x)
|
|
}
|
|
sort.Slice(res, func(i, j int) bool { return res[i] < res[j] })
|
|
|
|
return res
|
|
}
|
|
|
|
func TestSumRandom(t *testing.T) {
|
|
N := 100
|
|
|
|
for i := 0; i < N; i++ {
|
|
abits := randomBits(1000, 2000)
|
|
bbits := randomBits(1000, 2000)
|
|
sumbits := sum(abits, bbits)
|
|
|
|
a, err := RunsFromSlice(abits)
|
|
assert.NoError(t, err)
|
|
b, err := RunsFromSlice(bbits)
|
|
assert.NoError(t, err)
|
|
|
|
s, err := Sum(b, a)
|
|
assert.NoError(t, err)
|
|
bis, err := SliceFromRuns(s)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, sumbits, bis)
|
|
}
|
|
}
|