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
83
84
85
86
87
88
89
|
// Copyright 2020 The Inet.Af AUTHORS. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package netipx
import (
"testing"
)
func TestUint128AddSub(t *testing.T) {
const add1 = 1
const sub1 = -1
tests := []struct {
in uint128
op int // +1 or -1 to add vs subtract
want uint128
}{
{uint128{0, 0}, add1, uint128{0, 1}},
{uint128{0, 1}, add1, uint128{0, 2}},
{uint128{1, 0}, add1, uint128{1, 1}},
{uint128{0, ^uint64(0)}, add1, uint128{1, 0}},
{uint128{^uint64(0), ^uint64(0)}, add1, uint128{0, 0}},
{uint128{0, 0}, sub1, uint128{^uint64(0), ^uint64(0)}},
{uint128{0, 1}, sub1, uint128{0, 0}},
{uint128{0, 2}, sub1, uint128{0, 1}},
{uint128{1, 0}, sub1, uint128{0, ^uint64(0)}},
{uint128{1, 1}, sub1, uint128{1, 0}},
}
for _, tt := range tests {
var got uint128
switch tt.op {
case add1:
got = tt.in.addOne()
case sub1:
got = tt.in.subOne()
default:
panic("bogus op")
}
if got != tt.want {
t.Errorf("%v add %d = %v; want %v", tt.in, tt.op, got, tt.want)
}
}
}
func TestBitsSetFrom(t *testing.T) {
tests := []struct {
bit uint8
want uint128
}{
{0, uint128{^uint64(0), ^uint64(0)}},
{1, uint128{^uint64(0) >> 1, ^uint64(0)}},
{63, uint128{1, ^uint64(0)}},
{64, uint128{0, ^uint64(0)}},
{65, uint128{0, ^uint64(0) >> 1}},
{127, uint128{0, 1}},
{128, uint128{0, 0}},
}
for _, tt := range tests {
var zero uint128
got := zero.bitsSetFrom(tt.bit)
if got != tt.want {
t.Errorf("0.bitsSetFrom(%d) = %064b want %064b", tt.bit, got, tt.want)
}
}
}
func TestBitsClearedFrom(t *testing.T) {
tests := []struct {
bit uint8
want uint128
}{
{0, uint128{0, 0}},
{1, uint128{1 << 63, 0}},
{63, uint128{^uint64(0) &^ 1, 0}},
{64, uint128{^uint64(0), 0}},
{65, uint128{^uint64(0), 1 << 63}},
{127, uint128{^uint64(0), ^uint64(0) &^ 1}},
{128, uint128{^uint64(0), ^uint64(0)}},
}
for _, tt := range tests {
ones := uint128{^uint64(0), ^uint64(0)}
got := ones.bitsClearedFrom(tt.bit)
if got != tt.want {
t.Errorf("ones.bitsClearedFrom(%d) = %064b want %064b", tt.bit, got, tt.want)
}
}
}
|
