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
|
package mem
import (
"bytes"
"encoding/hex"
"fmt"
"io"
"math/rand"
"testing"
)
var (
testSizes = [...]int{
0, 1, 2, 3, 4, 6, 8, 10, 31, 32, 33, 64, 100, 1024, 4096,
}
benchmarkSizes = [...]int{
7, 10, 31, 32, 100, 1024, 4096,
}
)
func testCopy(t *testing.T, test, init func(dst, src []byte) int) {
for _, N := range testSizes {
t.Run(fmt.Sprintf("N=%d", N), func(t *testing.T) {
src := make([]byte, N)
dst := make([]byte, N)
exp := make([]byte, N)
prng := rand.New(rand.NewSource(0))
io.ReadFull(prng, src)
io.ReadFull(prng, dst)
copy(exp, dst)
init(exp, src)
n := test(dst, src)
if n != N {
t.Errorf("copying did not apply to enough bytes: %d != %d", n, N)
}
if !bytes.Equal(dst, exp) {
t.Error("copying produced the wrong output")
t.Logf("expected:\n%s", hex.Dump(exp))
t.Logf("found: \n%s", hex.Dump(dst))
t.Logf("source: \n%s", hex.Dump(src))
}
})
}
}
func benchmarkCopy(b *testing.B, test func(dst, src []byte) int) {
for _, N := range benchmarkSizes {
b.Run(fmt.Sprintf("N=%d", N), func(b *testing.B) {
dst := make([]byte, N)
src := make([]byte, N)
io.ReadFull(rand.New(rand.NewSource(0)), src)
b.SetBytes(int64(N))
b.ResetTimer()
for i := 0; i < b.N; i++ {
test(dst, src)
}
})
}
}
|
