From 6b776d1b216e7754b6b6863b6d81d5abbae9f865 Mon Sep 17 00:00:00 2001
From: Broderick Westrope <broderickwestrope@gmail.com>
Date: Sun, 12 Nov 2023 18:13:07 +1100
Subject: [PATCH] test: add TestTranspose

---
 tetrimino_test.go | 92 ++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 76 insertions(+), 16 deletions(-)

diff --git a/tetrimino_test.go b/tetrimino_test.go
index d9dcdc0..5b7a5f9 100644
--- a/tetrimino_test.go
+++ b/tetrimino_test.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"reflect"
 	"testing"
 )
 
@@ -513,35 +514,60 @@ func TestTetrimino_MoveRight(t *testing.T) {
 	}
 }
 
-func TestIsOutOfBoundsHorizontally(t *testing.T) {
-	playfield := &Playfield{}
-
+func TestTranspose(t *testing.T) {
 	tt := []struct {
-		name    string
-		tetPosX int
-		cellCol int
-		expects bool
+		name          string
+		tet           *Tetrimino
+		expectedCells [][]bool
 	}{
 		{
-			"out left", 0, -1, true,
-		},
-		{
-			"in left", -1, 1, false,
+			"1x2",
+			&Tetrimino{
+				Cells: [][]bool{
+					{true, false},
+				},
+			},
+			[][]bool{
+				{true},
+				{false},
+			},
 		},
 		{
-			"in right", 10, -1, false,
+			"2x2",
+			&Tetrimino{
+				Cells: [][]bool{
+					{true, false},
+					{true, false},
+				},
+			},
+			[][]bool{
+				{true, true},
+				{false, false},
+			},
 		},
 		{
-			"out right", 10, 0, true,
+			"3x3",
+			&Tetrimino{
+				Cells: [][]bool{
+					{true, false, true},
+					{true, false, false},
+					{true, false, true},
+				},
+			},
+			[][]bool{
+				{true, true, true},
+				{false, false, false},
+				{true, false, true},
+			},
 		},
 	}
 
 	for _, tc := range tt {
 		t.Run(tc.name, func(t *testing.T) {
-			result := isOutOfBoundsHorizontally(tc.tetPosX, tc.cellCol, playfield)
+			tc.tet.transpose()
 
-			if result != tc.expects {
-				t.Errorf("got %v, want %v", result, tc.expects)
+			if !reflect.DeepEqual(tc.tet.Cells, tc.expectedCells) {
+				t.Errorf("got %v,\nwant %v", tc.tet.Cells, tc.expectedCells)
 			}
 		})
 	}
@@ -701,6 +727,40 @@ func TestCanRotate(t *testing.T) {
 	}
 }
 
+func TestIsOutOfBoundsHorizontally(t *testing.T) {
+	playfield := &Playfield{}
+
+	tt := []struct {
+		name    string
+		tetPosX int
+		cellCol int
+		expects bool
+	}{
+		{
+			"out left", 0, -1, true,
+		},
+		{
+			"in left", -1, 1, false,
+		},
+		{
+			"in right", 10, -1, false,
+		},
+		{
+			"out right", 10, 0, true,
+		},
+	}
+
+	for _, tc := range tt {
+		t.Run(tc.name, func(t *testing.T) {
+			result := isOutOfBoundsHorizontally(tc.tetPosX, tc.cellCol, playfield)
+
+			if result != tc.expects {
+				t.Errorf("got %v, want %v", result, tc.expects)
+			}
+		})
+	}
+}
+
 func TestIsOutOfBoundsVertically(t *testing.T) {
 	playfield := &Playfield{}