-
Notifications
You must be signed in to change notification settings - Fork 1
/
registers_test.go
82 lines (70 loc) · 1.62 KB
/
registers_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package hyperloglog
import (
"math/rand"
"testing"
)
func TestRegistersGetSetSum(t *testing.T) {
length := uint32(16777216)
data := make([]uint8, length)
r := newRegisters(length)
for i := range data {
val := uint8(rand.Intn(16))
r.set(uint32(i), val)
data[i] = val
}
for i, exp := range data {
if got := r.get(uint32(i)); exp != got {
t.Errorf("expected %d, got %d", exp, got)
}
}
}
func TestRegistersZeros(t *testing.T) {
m := uint32(8)
rs := newRegisters(m)
for i := uint32(0); i < m; i++ {
rs.set(i, (uint8(i)%15)+1)
}
for i := uint32(0); i < m; i++ {
rs.set(i, (uint8(i)%15)+1)
}
for i := uint32(0); i < m; i++ {
exp := uint8(i%15) + 1
if got := rs.get(i); got != exp {
t.Errorf("expected %d, got %d", exp, got)
}
}
rs.rebase(1)
for i := uint32(0); i < m; i++ {
exp := uint8(i % 15)
if got := rs.get(i); got != exp {
t.Errorf("expected %d, got %d", exp, got)
}
}
if got := rs.nz; got != 1 {
t.Errorf("expected 1, got %d", got)
}
}
func assertRegistersNz(t *testing.T, rs *registers, nz uint32) {
t.Helper()
if rs.nz != nz {
t.Fatalf("registers.nz is not %d: actual=%d", nz, rs.nz)
}
}
func TestRegistersSetRepeatedly(t *testing.T) {
rs := newRegisters(16)
// count down nz when set non-zero values to each registers.
assertRegistersNz(t, rs, 16)
for i := uint32(0); i < 16; i++ {
rs.set(i, 1)
assertRegistersNz(t, rs, 15 - i)
}
// keep nz:0 when set non-zero values.
for i := uint8(1); i <= 15; i++ {
for j := uint32(0); j < 16; j++ {
rs.set(j, i)
if rs.nz != 0 {
t.Fatalf("registers.nz is not zero: actual=%d (i=%d, j=%d)", rs.nz, i, j)
}
}
}
}