karma/maze-solver-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Finally written some good tests for the image writer

Browse Source
This commit is contained in:
Karma Riuk 2023-08-14 15:08:27 +02:00
parent 94af003adc
commit 346cfe9705
1 changed files with 114 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

177
io/writer/image_test.go
View File

@ -1,71 +1,122 @@
package writer
import (
"bytes"
"image/color"
"io"
"maze-solver/maze"
"os"
"testing"
"github.com/mazznoer/colorgrad"
)
const (
OUT_DIR = "./out"
EXPECTED_DIR = "../../assets/solved"
)
func TestImageWriter(t *testing.T) {
// pathGradient, err := colorgrad.NewGradient().Colors(color.White).Build()
// if err != nil {
// panic(err)
// }
//
// tests := []struct {
// name string
// filename string
// m *maze.SolvedMaze
// CellWidth, cellHeight int
// pathColor, wallColor color.Color
// gradient colorgrad.Gradient
// }{
// {
// "Trivial",
// "../../out/trivial_sol.png",
// trivial(),
// 40, 40,
// color.White, color.Black,
// colorgrad.Warm(),
// },
// {
// "Bigger",
// "../../out/bigger_sol.png",
// bigger(),
// 40, 40,
// color.White, color.Black,
// colorgrad.Warm(),
// },
// {
// "Bigger Staggered",
// "../../out/bigger_staggered_sol.png",
// bigger_staggered(),
// 40, 40,
// color.White, color.Black,
// pathGradient,
// },
// {
// "Normal",
// "../../out/normal_sol.png",
// normal(),
// 40, 40,
// color.White, color.Black,
// colorgrad.Warm(),
// },
// }
//
// for _, test := range tests {
// writer := ImageWriter{
// Filename: test.filename,
// Maze: test.m,
// CellWidth: test.CellWidth,
// CellHeight: test.cellHeight,
// WallColor: test.wallColor,
// PathColor: test.pathColor,
// SolutionGradient: test.gradient,
// }
//
// err := writer.Write()
// if err != nil {
// t.Fatalf("%s: couldn't write solution, got following error\n%v", test.name, err)
// }
// }
if _, err := os.Stat(OUT_DIR); os.IsNotExist(err) {
os.Mkdir(OUT_DIR, 0700)
}
tests := []struct {
name string
filename string
m *maze.SolvedMaze
CellWidth, cellHeight int
pathColor, wallColor color.Color
gradient colorgrad.Gradient
}{
{
"Trivial",
"trivial.png",
trivial(),
20, 20,
color.White, color.Black,
colorgrad.Warm(),
},
{
"Trivial Bigger",
"trivial-bigger.png",
bigger(),
20, 20,
color.White, color.Black,
colorgrad.Warm(),
},
{
"Trivial Bigger Staggered",
"trivial-bigger-staggered.png",
bigger_staggered(),
20, 20,
color.White, color.Black,
colorgrad.Warm(),
},
{
"Normal",
"normal.png",
normal(),
20, 20,
color.White, color.Black,
colorgrad.Warm(),
},
}
for _, test := range tests {
writer := ImageWriter{
Filename: OUT_DIR + "/" + test.filename,
Maze: test.m,
CellWidth: test.CellWidth,
CellHeight: test.cellHeight,
WallColor: test.wallColor,
PathColor: test.pathColor,
SolutionGradient: test.gradient,
}
err := writer.Write()
if err != nil {
t.Fatalf("%s: couldn't write solution, got following error\n%v", test.name, err)
}
assertEqualFile(t, EXPECTED_DIR+"/"+test.filename, OUT_DIR+"/"+test.filename, test.name)
os.Remove(writer.Filename)
}
os.Remove(OUT_DIR)
}
const chunkSize = 64000
func assertEqualFile(t *testing.T, file1, file2, name string) {
f1, err := os.Open(file1)
if err != nil {
t.Fatal(err)
}
defer f1.Close()
f2, err := os.Open(file2)
if err != nil {
t.Fatal(err)
}
defer f2.Close()
for {
b1 := make([]byte, chunkSize)
_, err1 := f1.Read(b1)
b2 := make([]byte, chunkSize)
_, err2 := f2.Read(b2)
if err1 != nil || err2 != nil {
if err1 == io.EOF && err2 == io.EOF {
return
} else if err1 == io.EOF || err2 == io.EOF {
t.Fatalf("%s: files are not equal. Got %q, wanted %q", name, file1, file2)
}
}
if !bytes.Equal(b1, b2) {
t.Fatalf("%s: files are not equal. Got %q, wanted %q", name, file1, file2)
}
}
}