76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package writer
|
|
|
|
import (
|
|
"image/color"
|
|
"maze-solver/maze"
|
|
"testing"
|
|
|
|
"github.com/mazznoer/colorgrad"
|
|
)
|
|
|
|
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)
|
|
}
|
|
}
|
|
}
|