-
Notifications
You must be signed in to change notification settings - Fork 2
/
source_test.go
84 lines (65 loc) · 1.55 KB
/
source_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
83
84
package fastrand_test
import (
"math/rand"
"testing"
"github.com/segmentio/fastrand"
)
func TestFastrand(t *testing.T) {
const N = 1000000
values := make(map[uint64]struct{}, N)
source := fastrand.NewSource(0)
for i := 0; i < N; i++ {
x := source.Uint64()
if _, exists := values[x]; exists {
t.Errorf("%d was generated twice", x)
break
}
values[x] = struct{}{}
}
}
func BenchmarkNewSource(b *testing.B) {
b.Run("math/rand", benchmarkMathRandNewSource)
b.Run("fastrand", benchmarkFastrandNewSource)
}
func benchmarkMathRandNewSource(b *testing.B) {
for i := 0; i < b.N; i++ {
_ = rand.NewSource(1234567890)
}
}
func benchmarkFastrandNewSource(b *testing.B) {
for i := 0; i < b.N; i++ {
_ = fastrand.NewSource(1234567890)
}
}
func BenchmarkRandSource(b *testing.B) {
b.Run("math/rand", benchmarkMathRandSource)
b.Run("fastrand", benchmarkFastrandSource)
}
func benchmarkMathRandSource(b *testing.B) {
benchmarkSource(b, rand.NewSource(0))
}
func benchmarkFastrandSource(b *testing.B) {
benchmarkSource(b, fastrand.NewSource(0))
}
func benchmarkSource(b *testing.B, source rand.Source) {
prng := rand.New(source)
for i := 0; i < b.N; i++ {
_ = prng.NormFloat64()
}
}
func BenchmarkSeed(b *testing.B) {
b.Run("math/rand", benchmarkMathRandSeed)
b.Run("fastrand", benchmarkFastrandSeed)
}
func benchmarkMathRandSeed(b *testing.B) {
r := rand.NewSource(0)
for i := 0; i < b.N; i++ {
r.Seed(1234567890)
}
}
func benchmarkFastrandSeed(b *testing.B) {
r := fastrand.NewSource(0)
for i := 0; i < b.N; i++ {
r.Seed(int64(i))
}
}