forked from bits-and-blooms/bitset
-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
popcnt_amd64_test.go
79 lines (71 loc) · 2.17 KB
/
popcnt_amd64_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
//go:build !go1.9 && amd64 && !appengine
// +build !go1.9,amd64,!appengine
// This file tests the popcnt funtions
package bitset
import (
"testing"
)
func TestPopcntSliceCond(t *testing.T) {
s := []uint64{2, 3, 5, 7, 11, 13, 17, 19, 23, 29}
oldUseAsm := useAsm
defer func() { useAsm = oldUseAsm }()
useAsm = false
resGo := popcntSlice(s)
useAsm = (true && oldUseAsm)
resAsm := popcntSlice(s)
if resGo != resAsm {
t.Errorf("The implementations are different: GO %d != ASM %d", resGo, resAsm)
}
}
func TestPopcntMaskSliceCond(t *testing.T) {
s := []uint64{2, 3, 5, 7, 11, 13, 17, 19, 23, 29}
m := []uint64{31, 37, 41, 43, 47, 53, 59, 61, 67, 71}
oldUseAsm := useAsm
defer func() { useAsm = oldUseAsm }()
useAsm = false
resGo := popcntMaskSlice(s, m)
useAsm = (true && oldUseAsm)
resAsm := popcntMaskSlice(s, m)
if resGo != resAsm {
t.Errorf("The implementations are different: GO %d != ASM %d", resGo, resAsm)
}
}
func TestPopcntAndSliceCond(t *testing.T) {
s := []uint64{2, 3, 5, 7, 11, 13, 17, 19, 23, 29}
m := []uint64{31, 37, 41, 43, 47, 53, 59, 61, 67, 71}
oldUseAsm := useAsm
defer func() { useAsm = oldUseAsm }()
useAsm = false
resGo := popcntAndSlice(s, m)
useAsm = (true && oldUseAsm)
resAsm := popcntAndSlice(s, m)
if resGo != resAsm {
t.Errorf("The implementations are different: GO %d != ASM %d", resGo, resAsm)
}
}
func TestPopcntOrSliceCond(t *testing.T) {
s := []uint64{2, 3, 5, 7, 11, 13, 17, 19, 23, 29}
m := []uint64{31, 37, 41, 43, 47, 53, 59, 61, 67, 71}
oldUseAsm := useAsm
defer func() { useAsm = oldUseAsm }()
useAsm = false
resGo := popcntOrSlice(s, m)
useAsm = (true && oldUseAsm)
resAsm := popcntOrSlice(s, m)
if resGo != resAsm {
t.Errorf("The implementations are different: GO %d != ASM %d", resGo, resAsm)
}
}
func TestPopcntXorSliceCond(t *testing.T) {
s := []uint64{2, 3, 5, 7, 11, 13, 17, 19, 23, 29}
m := []uint64{31, 37, 41, 43, 47, 53, 59, 61, 67, 71}
oldUseAsm := useAsm
defer func() { useAsm = oldUseAsm }()
useAsm = false
resGo := popcntXorSlice(s, m)
useAsm = (true && oldUseAsm)
resAsm := popcntXorSlice(s, m)
if resGo != resAsm {
t.Errorf("The implementations are different: GO %d != ASM %d", resGo, resAsm)
}
}