93 lines
1.8 KiB
Go
93 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path"
|
|
"testing"
|
|
)
|
|
|
|
func exampleFiles() []string {
|
|
files, err := ioutil.ReadDir("example")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
names := []string{}
|
|
for _, file := range files {
|
|
names = append(names, path.Join("example", file.Name()))
|
|
}
|
|
return names
|
|
}
|
|
|
|
func referenceFile(infile string, suffix string) string {
|
|
return path.Join("test", path.Base(infile)+suffix)
|
|
}
|
|
|
|
func readFile(t *testing.T, file string) string {
|
|
bytes, err := ioutil.ReadFile(file)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
return string(bytes)
|
|
}
|
|
|
|
func TestLexer(t *testing.T) {
|
|
for _, file := range exampleFiles() {
|
|
t.Run(file, func(t *testing.T) {
|
|
input := readFile(t, file)
|
|
var buf bytes.Buffer
|
|
DebugLexer([]rune(input), &buf)
|
|
output := buf.String()
|
|
reference := readFile(t, referenceFile(file, ".lexer"))
|
|
if output != reference {
|
|
t.Fail()
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParser(t *testing.T) {
|
|
for _, file := range exampleFiles() {
|
|
t.Run(file, func(t *testing.T) {
|
|
input := readFile(t, file)
|
|
var buf bytes.Buffer
|
|
DebugParser([]rune(input), &buf)
|
|
output := buf.String()
|
|
reference := readFile(t, referenceFile(file, ".parser"))
|
|
if output != reference {
|
|
t.Fail()
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSearch(t *testing.T) {
|
|
available := map[float64]int{
|
|
20: 26,
|
|
10: 2,
|
|
5: 2,
|
|
2.5: 2,
|
|
1.25: 2,
|
|
0.5: 2,
|
|
}
|
|
tests := map[float64][]float64{
|
|
17: []float64{5, 2.5, 0.5},
|
|
42: []float64{20, 0.5},
|
|
80: []float64{20, 20},
|
|
}
|
|
for input, reference := range tests {
|
|
t.Run("search"+fmt.Sprint(input), func(t *testing.T) {
|
|
output := SearchPlates(available, input)
|
|
if len(reference) != len(output) {
|
|
t.Fail()
|
|
}
|
|
for i := 0; i < len(reference); i++ {
|
|
if reference[i] != output[i] {
|
|
t.Fail()
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|