matrix_test.go

package matrix

import (
	"fmt"
	"sort"
	"testing"
)

var (
	m  = New[string](10, 10, false)
	m2 = New[string](3, 3, false)

	center = []Cell[string]{
		{X: 0, Y: 0},
		{X: 0, Y: 1},
		{X: 0, Y: 2},
		{X: 1, Y: 0},
		{X: 1, Y: 2},
		{X: 2, Y: 0},
		{X: 2, Y: 1},
		{X: 2, Y: 2},
	}

	left = []Cell[string]{
		{X: 0, Y: 0},
		{X: 1, Y: 0},
		{X: 1, Y: 1},
		{X: 0, Y: 2},
		{X: 1, Y: 2},
	}

	left_wrap = []Cell[string]{
		{X: 0, Y: 0},
		{X: 1, Y: 0},
		{X: 1, Y: 1},
		{X: 0, Y: 2},
		{X: 1, Y: 2},
		{X: 2, Y: 1},
		{X: 2, Y: 2},
		{X: 2, Y: 0},
	}

	right = []Cell[string]{
		{X: 1, Y: 2},
		{X: 2, Y: 2},
		{X: 1, Y: 1},
		{X: 1, Y: 0},
		{X: 2, Y: 0},
	}

	right_wrap = []Cell[string]{
		{X: 1, Y: 2},
		{X: 2, Y: 2},
		{X: 1, Y: 1},
		{X: 1, Y: 0},
		{X: 2, Y: 0},
		{X: 0, Y: 2},
		{X: 0, Y: 1},
		{X: 0, Y: 0},
	}

	bottomleft = []Cell[string]{
		{X: 0, Y: 1},
		{X: 1, Y: 1},
		{X: 1, Y: 0},
	}

	topright = []Cell[string]{
		{X: 1, Y: 2},
		{X: 1, Y: 1},
		{X: 2, Y: 1},
	}

	bottomleft_wrap = []Cell[string]{
		{X: 0, Y: 1},
		{X: 1, Y: 1},
		{X: 1, Y: 0},
		{X: 2, Y: 1},
		{X: 2, Y: 0},
		{X: 2, Y: 2},
		{X: 0, Y: 2},
		{X: 1, Y: 2},
	}

	topright_wrap = []Cell[string]{
		{X: 1, Y: 2},
		{X: 1, Y: 1},
		{X: 2, Y: 1},
		{X: 0, Y: 1},
		{X: 0, Y: 2},
		{X: 0, Y: 0},
		{X: 2, Y: 0},
		{X: 1, Y: 0},
	}

	m4 Matrix[int]
)

func init() {
	m.Set(0, 0, "1")
	m.Set(0, m.Height()-1, "2")
	m.Set(m.Width()-1, 0, "3")
	m.Set(m.Width()-1, m.Height()-1, "4")

	m2.Set(0, 0, "TL")
	m2.Set(0, 1, "ML")
	m2.Set(0, 2, "BL")
	m2.Set(1, 0, "TM")
	m2.Set(1, 1, "MM")
	m2.Set(1, 2, "BM")
	m2.Set(2, 0, "TR")
	m2.Set(2, 1, "MR")
	m2.Set(2, 2, "BR")

	m4, _ = FromSlice(4, false, []int{
		0, 1, 2, 3,
		4, 5, 6, 7,
		8, 9, 10, 11,
		12, 13, 14, 15,
	})
}

func TestNew(t *testing.T) {
	fmt.Printf("%#v\n", m)
}

func TestByRows(t *testing.T) {
	for i := 0; i < m.Height(); i++ {
		fmt.Println(m.Row(i))
	}
}

func TestByColumns(t *testing.T) {
	for i := 0; i < m.Width(); i++ {
		fmt.Println(m.Column(i))
	}
}

func TestClone(t *testing.T) {
	m2 := m.Clone()
	fmt.Printf("%#v\n", m2)
}

func TestFix(t *testing.T) {
	fmt.Printf("fix 0: %v\n", m.Fix(0))
	fmt.Printf("unfix 9: %v\n", m.Fix(9))
}

func TestFill(t *testing.T) {
	m2 := m.Clone()
	fmt.Printf("%#v\n", m2)

	m2.Fill(".")
	fmt.Printf("%#v\n", m2)
}

func compare(l1, l2 []Cell[string]) bool {
	if len(l1) != len(l2) {
		fmt.Println("XXX compare length", l1, l2)
		return false
	}

	sort.Slice(l1, func(i, j int) bool {
		v1 := l1[i].X*3 + l1[i].Y
		v2 := l1[j].X*3 + l1[j].Y

		return v1 < v2
	})

	sort.Slice(l2, func(i, j int) bool {
		v1 := l2[i].X*3 + l2[i].Y
		v2 := l2[j].X*3 + l2[j].Y

		return v1 < v2
	})

	for i := range l1 {
		if l1[i].X != l2[i].X || l1[i].Y != l2[i].Y {
			fmt.Println("XXX compare", i, l1, l2)
			return false
		}
	}

	return true
}

func TestAdjacent(t *testing.T) {
	cells := m2.Adjacent(1, 1, false)
	if !compare(cells, center) {
		t.Logf("(1,1) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(0, 1, false)
	if !compare(cells, left) {
		t.Logf("(0,1) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(2, 1, false)
	if !compare(cells, right) {
		t.Logf("(2,1) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(0, 0, false)
	if !compare(cells, bottomleft) {
		t.Logf("(0,0) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(2, 2, false)
	if !compare(cells, topright) {
		t.Logf("(2,2) %v %#v\n", len(cells), cells)
		t.Fail()
	}
}

func TestAdjacentRoll(t *testing.T) {
	cells := m2.Adjacent(1, 1, true)
	if !compare(cells, center) {
		t.Logf("(1,1) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(0, 1, true)
	if !compare(cells, left_wrap) {
		t.Logf("(0,1) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(2, 1, true)
	if !compare(cells, right_wrap) {
		t.Logf("(2,1) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(0, 0, true)
	if !compare(cells, bottomleft_wrap) {
		t.Logf("(0,0) %v %#v\n", len(cells), cells)
		t.Fail()
	}

	cells = m2.Adjacent(2, 2, true)
	if !compare(cells, topright_wrap) {
		t.Logf("(2,2) %v %#v\n", len(cells), cells)
		t.Fail()
	}
}

func TestNewLike(t *testing.T) {
	n := NewLike(m)
	if n.Width() != m.Width() || n.Height() != m.Height() {
		t.Logf("expected %vx%v, got %vx%v", m.Width(), m.Height(), n.Width(), n.Height())
		t.Fail()
	}
}

func TestSubmatrix(t *testing.T) {
	cart := m4.Cartesian()

	s := m4.Submatrix(1, 1, 2, 2)
	r, _ := FromSlice(2, cart, []int{5, 6, 9, 10})

	if !s.Equals(r) {
		t.Logf("expected %#v got %#v", r, s)
		t.Fail()
	}

	s = m4.Submatrix(0, 0, 2, 2)
	r, _ = FromSlice(2, cart, []int{0, 1, 4, 5})

	if !s.Equals(r) {
		t.Logf("expected %#v got %#v", r, s)
		t.Fail()
	}

	s = m4.Submatrix(2, 2, 3, 3)
	r, _ = FromSlice(2, cart, []int{10, 11, 14, 15})

	if !s.Equals(r) {
		t.Logf("expected %#v got %#v", r, s)
		t.Fail()
	}
}

func TestCopy(t *testing.T) {
	cart := m4.Cartesian()
	src := m4.Submatrix(0, 0, 3, 3)
	dst := New[int](5, 5, cart)

	dst.Copy(0, 0, src)

	r, _ := FromSlice(5, cart, []int{
		0, 1, 2, 0, 0,
		4, 5, 6, 0, 0,
		8, 9, 10, 0, 0,
		0, 0, 0, 0, 0,
		0, 0, 0, 0, 0,
	})

	if !dst.Equals(r) {
		t.Logf("expected %#v got %#v", r, dst)
		t.Fail()
	}

	dst.Fill(0)
	dst.Copy(1, 1, src)

	r, _ = FromSlice(5, cart, []int{
		0, 0, 0, 0, 0,
		0, 0, 1, 2, 0,
		0, 4, 5, 6, 0,
		0, 8, 9, 10, 0,
		0, 0, 0, 0, 0,
	})

	if !dst.Equals(r) {
		t.Logf("expected %#v got %#v", r, dst)
		t.Fail()
	}

	dst.Fill(0)
	dst.Copy(2, 2, src)

	r, _ = FromSlice(5, cart, []int{
		0, 0, 0, 0, 0,
		0, 0, 0, 0, 0,
		0, 0, 0, 1, 2,
		0, 0, 4, 5, 6,
		0, 0, 8, 9, 10,
	})

	if !dst.Equals(r) {
		t.Logf("expected %#v got %#v", r, dst)
		t.Fail()
	}

	dst.Fill(0)
	dst.Copy(3, 3, src)

	r, _ = FromSlice(5, cart, []int{
		0, 0, 0, 0, 0,
		0, 0, 0, 0, 0,
		0, 0, 0, 0, 0,
		0, 0, 0, 0, 1,
		0, 0, 0, 4, 5,
	})

	if !dst.Equals(r) {
		t.Logf("expected %#v got %#v", r, dst)
		t.Fail()
	}
}

func TestVisited(t *testing.T) {
	m, _ := FromSlice(5, false, []int{
		5, 0, 0, 0, 0,
		5, 0, 0, 0, 0,
		5, 5, 5, 0, 0,
		0, 0, 5, 0, 0,
		0, 0, 5, 0, 0,
	})

	l := m.Connected(0, 0, -1)
	if len(l) != 7 {
		t.Logf("expected connected=7, got %v", len(l))
		t.Fail()
	}

	if l[0].Value != 5 {
		t.Logf("expected value=5, got %v", l[0].Value)
		t.Fail()
	}

	m, _ = FromSlice(5, true, []int{
		5, 0, 0, 0, 0,
		5, 0, 0, 0, 0,
		5, 5, 5, 0, 0,
		0, 0, 5, 0, 0,
		0, 0, 5, 0, 0,
	})

	l = m.Connected(0, 0, -1)
	if len(l) != 4 {
		t.Logf("expected connected=7, got %v", len(l))
		t.Fail()
	}

	if l[0].Value != 0 {
		t.Logf("expected value=5, got %v", l[0].Value)
		t.Fail()
	}
}