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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
/*
* SPDX-FileCopyrightText: © Hypermode Inc. <hello@hypermode.com>
* SPDX-License-Identifier: Apache-2.0
*/
package sim
import (
"bytes"
"compress/gzip"
"os"
"testing"
)
func TestZipfian(t *testing.T) {
s := NewZipfian(1.5, 1, 100)
m := make(map[uint64]uint64, 100)
for i := 0; i < 100; i++ {
k, err := s()
if err != nil {
t.Fatal(err)
}
m[k]++
}
if len(m) == 0 || len(m) == 100 {
t.Fatal("zipfian not skewed")
}
}
func TestUniform(t *testing.T) {
s := NewUniform(100)
for i := 0; i < 100; i++ {
if _, err := s(); err != nil {
t.Fatal(err)
}
}
}
func TestParseLIRS(t *testing.T) {
s := NewReader(ParseLIRS, bytes.NewReader([]byte{
'0', '\n',
'1', '\r', '\n',
'2', '\r', '\n',
}))
for i := uint64(0); i < 3; i++ {
v, err := s()
if err != nil {
t.Fatal(err)
}
if v != i {
t.Fatal("value mismatch")
}
}
}
func TestReadLIRS(t *testing.T) {
f, err := os.Open("./gli.lirs.gz")
if err != nil {
t.Fatal(err)
}
r, err := gzip.NewReader(f)
if err != nil {
t.Fatal(err)
}
s := NewReader(ParseLIRS, r)
for i := uint64(0); i < 100; i++ {
if _, err = s(); err != nil {
t.Fatal(err)
}
}
}
func TestParseARC(t *testing.T) {
s := NewReader(ParseARC, bytes.NewReader([]byte{
'1', '2', '7', ' ', '6', '4', ' ', '0', ' ', '0', '\r', '\n',
'1', '9', '1', ' ', '3', '6', ' ', '0', ' ', '0', '\r', '\n',
}))
for i := uint64(0); i < 100; i++ {
v, err := s()
if err != nil {
t.Fatal(err)
}
if v != 127+i {
t.Fatal("value mismatch")
}
}
}
func TestCollection(t *testing.T) {
s := NewUniform(100)
c := Collection(s, 100)
if len(c) != 100 {
t.Fatal("collection not full")
}
}
func TestStringCollection(t *testing.T) {
s := NewUniform(100)
c := StringCollection(s, 100)
if len(c) != 100 {
t.Fatal("string collection not full")
}
}
|
