132 lines
2.1 KiB
Go
132 lines
2.1 KiB
Go
|
|
package part1
|
|
|
|
import (
|
|
"testing"
|
|
"reflect"
|
|
"bufio"
|
|
"strings"
|
|
)
|
|
|
|
const basicInput =
|
|
".....\n" +
|
|
".S-7.\n" +
|
|
".|.|.\n" +
|
|
".L-J.\n" +
|
|
".....\n"
|
|
|
|
var basicMap = []string{
|
|
".....",
|
|
".S-7.",
|
|
".|.|.",
|
|
".L-J.",
|
|
".....",
|
|
}
|
|
|
|
const complexInput =
|
|
"7-F7-\n" +
|
|
".FJ|7\n" +
|
|
"SJLL7\n" +
|
|
"|F--J\n" +
|
|
"LJ.LJ\n"
|
|
|
|
var complexMap = []string{
|
|
"7-F7-",
|
|
".FJ|7",
|
|
"SJLL7",
|
|
"|F--J",
|
|
"LJ.LJ",
|
|
}
|
|
|
|
func TestReadMap(t *testing.T) {
|
|
reader := bufio.NewScanner(strings.NewReader(basicInput));
|
|
pipes := ReadMap(reader)
|
|
equal(t, pipes, basicMap)
|
|
}
|
|
|
|
func TestReadMapComplex(t *testing.T) {
|
|
reader := bufio.NewScanner(strings.NewReader(complexInput));
|
|
pipes := ReadMap(reader)
|
|
equal(t, pipes, complexMap)
|
|
}
|
|
|
|
func TestFindStart(t *testing.T) {
|
|
x,y := FindStart(basicMap)
|
|
equal(t, x, 1)
|
|
equal(t, y, 1)
|
|
}
|
|
|
|
func TestFindStartComplex(t *testing.T) {
|
|
x,y := FindStart(complexMap)
|
|
equal(t, x, 0)
|
|
equal(t, y, 2)
|
|
}
|
|
|
|
func TestFindFirstBasic(t *testing.T) {
|
|
x,y := 1,1
|
|
side,newX,newY := FindFirst(basicMap, x,y)
|
|
equal(t, side, 0)
|
|
equal(t, newX, 2)
|
|
equal(t, newY, 1)
|
|
}
|
|
|
|
func TestFindFirstComplex(t *testing.T) {
|
|
x,y := 0, 2
|
|
side,newX,newY := FindFirst(complexMap, x,y)
|
|
equal(t, side, 1)
|
|
equal(t, newX, 1)
|
|
equal(t, newY, 2)
|
|
}
|
|
|
|
func TestGetSide(t *testing.T) {
|
|
side,ok := GetSide(basicMap, 1,1, 2,1)
|
|
equal(t, side, 0)
|
|
equal(t, ok, true)
|
|
}
|
|
|
|
func TestFindNext(t *testing.T) {
|
|
cx,cy := 2,1
|
|
|
|
s,nx,ny := FindNext(basicMap, 0, cx,cy)
|
|
equal(t, nx, 3)
|
|
equal(t, ny, 1)
|
|
|
|
s,nx,ny = FindNext(basicMap, s, nx,ny)
|
|
equal(t, nx, 3)
|
|
equal(t, ny, 2)
|
|
|
|
s,nx,ny = FindNext(basicMap, s, nx,ny)
|
|
equal(t, nx, 3)
|
|
equal(t, ny, 3)
|
|
|
|
s,nx,ny = FindNext(basicMap, s, nx,ny)
|
|
equal(t, nx, 2)
|
|
equal(t, ny, 3)
|
|
|
|
s,nx,ny = FindNext(basicMap, s, nx,ny)
|
|
equal(t, nx, 1)
|
|
equal(t, ny, 3)
|
|
|
|
s,nx,ny = FindNext(basicMap, s, nx,ny)
|
|
equal(t, nx, 1)
|
|
equal(t, ny, 2)
|
|
|
|
s,nx,ny = FindNext(basicMap, s, nx,ny)
|
|
equal(t, nx, 1)
|
|
equal(t, ny, 1)
|
|
}
|
|
|
|
func TestLoopSize(t *testing.T) {
|
|
size := LoopSize(basicMap)
|
|
equal(t, size, 8)
|
|
size = LoopSize(complexMap)
|
|
equal(t, size, 16)
|
|
}
|
|
|
|
func equal(t *testing.T, actual any, target any) {
|
|
if !reflect.DeepEqual(actual, target) {
|
|
t.Errorf("expected %v, got %v", target, actual)
|
|
}
|
|
}
|
|
|