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 package writer
import ( import (
"bytes"
"image/color"
"io"
"maze-solver/maze"
"os"
"testing" "testing"
"github.com/mazznoer/colorgrad"
)
const (
OUT_DIR = "./out"
EXPECTED_DIR = "../../assets/solved"
) )
func TestImageWriter(t *testing.T) { func TestImageWriter(t *testing.T) {
// pathGradient, err := colorgrad.NewGradient().Colors(color.White).Build() if _, err := os.Stat(OUT_DIR); os.IsNotExist(err) {
// if err != nil { os.Mkdir(OUT_DIR, 0700)
// panic(err) }
// }
// tests := []struct {
// tests := []struct { name string
// name string filename string
// filename string m *maze.SolvedMaze
// m *maze.SolvedMaze CellWidth, cellHeight int
// CellWidth, cellHeight int pathColor, wallColor color.Color
// pathColor, wallColor color.Color gradient colorgrad.Gradient
// gradient colorgrad.Gradient }{
// }{ {
// { "Trivial",
// "Trivial", "trivial.png",
// "../../out/trivial_sol.png", trivial(),
// trivial(), 20, 20,
// 40, 40, color.White, color.Black,
// color.White, color.Black, colorgrad.Warm(),
// colorgrad.Warm(), },
// }, {
// { "Trivial Bigger",
// "Bigger", "trivial-bigger.png",
// "../../out/bigger_sol.png", bigger(),
// bigger(), 20, 20,
// 40, 40, color.White, color.Black,
// color.White, color.Black, colorgrad.Warm(),
// colorgrad.Warm(), },
// }, {
// { "Trivial Bigger Staggered",
// "Bigger Staggered", "trivial-bigger-staggered.png",
// "../../out/bigger_staggered_sol.png", bigger_staggered(),
// bigger_staggered(), 20, 20,
// 40, 40, color.White, color.Black,
// color.White, color.Black, colorgrad.Warm(),
// pathGradient, },
// }, {
// { "Normal",
// "Normal", "normal.png",
// "../../out/normal_sol.png", normal(),
// normal(), 20, 20,
// 40, 40, color.White, color.Black,
// color.White, color.Black, colorgrad.Warm(),
// colorgrad.Warm(), },
// }, }
// }
// for _, test := range tests {
// for _, test := range tests { writer := ImageWriter{
// writer := ImageWriter{ Filename: OUT_DIR + "/" + test.filename,
// Filename: test.filename, Maze: test.m,
// Maze: test.m, CellWidth: test.CellWidth,
// CellWidth: test.CellWidth, CellHeight: test.cellHeight,
// CellHeight: test.cellHeight, WallColor: test.wallColor,
// WallColor: test.wallColor, PathColor: test.pathColor,
// PathColor: test.pathColor, SolutionGradient: test.gradient,
// SolutionGradient: test.gradient, }
// }
// err := writer.Write()
// err := writer.Write() if err != nil {
// if err != nil { t.Fatalf("%s: couldn't write solution, got following error\n%v", test.name, err)
// 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)
}
}
} }