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
|
package remembering
import (
"flag"
"fmt"
"io"
"os"
"path/filepath"
"strings"
)
var linesFlag = flag.Int(
"lines",
1000000,
"The number of menu lines to rank",
)
func MainTest() {
flag.Parse()
n := *linesFlag
tmp, err := os.MkdirTemp("", "remembering-bench-")
if err != nil {
panic(err)
}
defer os.RemoveAll(tmp)
err = os.Setenv("XDG_DATA_HOME", tmp)
if err != nil {
panic(err)
}
menu := strings.Builder{}
for i := 0; i < n; i++ {
fmt.Fprintf(&menu, "item-%07d\n", i)
}
// half the items are already known, with spread counts
profile := strings.Builder{}
for i := 0; i < n; i += 2 {
fmt.Fprintf(
&profile, "%d profile item-%07d\n",
i%97, i,
)
}
path, err := profilePath("bench")
if err != nil {
panic(err)
}
err = os.MkdirAll(filepath.Dir(path), 0755)
if err != nil {
panic(err)
}
err = os.WriteFile(
path, []byte(profile.String()), 0644,
)
if err != nil {
panic(err)
}
rc := run(envT{
allArgs: []string{
"remembering", "-p", "bench",
"--", "tail", "-n1",
},
in: strings.NewReader(menu.String()),
out: io.Discard,
err: os.Stderr,
})
if rc != 0 {
panic("ranking failed")
}
fmt.Printf("ranked %d lines\n", n)
}
|
