array2d_test.go

package array2d

import (
	"testing"
)

func TestArray2D_stringEmpty(t *testing.T) {
	arr := New[int](3, 3)
	got := arr.String()
	want := "[[0 0 0] [0 0 0] [0 0 0]]"
	if got != want {
		t.Errorf("want %q, got %q", want, got)
	}
}

func TestArray2D_stringValues(t *testing.T) {
	arr := New[int](3, 3)
	arr.Set(0, 0, 1)
	arr.Set(1, 0, 2)
	arr.Set(2, 0, 3)
	arr.Set(0, 1, 4)
	arr.Set(1, 1, 5)
	arr.Set(2, 1, 6)
	arr.Set(0, 2, 7)
	arr.Set(1, 2, 8)
	arr.Set(2, 2, 9)
	got := arr.String()
	want := "[[1 2 3] [4 5 6] [7 8 9]]"
	if got != want {
		t.Errorf("want %q, got %q", want, got)
	}
}

func TestArray2D_fill(t *testing.T) {
	arr := New[int](64, 64)
	val := 42
	arr.Fill(20, 25, 40, 38, val)
	for x := 0; x < arr.Width(); x++ {
		for y := 0; y < arr.Height(); y++ {
			want := 0
			if x >= 20 && x <= 40 && y >= 25 && y <= 38 {
				want = val
			}
			got := arr.Get(x, y)
			if got != want {
				t.Errorf("x=%d, y=%d: want %d, got %d", x, y, want, got)
			}
		}
	}
}

func TestArray2D_rowSpan(t *testing.T) {
	arr := New[int](5, 5)
	span := arr.RowSpan(1, 3, 2)
	assertLen(t, 3, span)
	copy(span, []int{1, 2, 3})
	for x := 0; x < arr.Width(); x++ {
		for y := 0; y < arr.Height(); y++ {
			want := 0
			if x >= 1 && x <= 3 && y == 2 {
				want = x
			}
			got := arr.Get(x, y)
			if got != want {
				t.Errorf("x=%d, y=%d: want %d, got %d", x, y, want, got)
			}
		}
	}
}

func TestArray2D_row(t *testing.T) {
	arr := New[int](5, 5)
	span := arr.Row(2)
	assertLen(t, 5, span)
	copy(span, []int{1, 2, 3, 4, 5})
	for x := 0; x < arr.Width(); x++ {
		for y := 0; y < arr.Height(); y++ {
			want := 0
			if y == 2 {
				want = x + 1
			}
			got := arr.Get(x, y)
			if got != want {
				t.Errorf("x=%d, y=%d: want %d, got %d", x, y, want, got)
			}
		}
	}
}

func assertLen[E any](t *testing.T, want int, slice []E) {
	t.Helper()
	if len(slice) != want {
		t.Errorf("want len %d, got len %d", want, len(slice))
	}
}