def test_game_board_add_to_column_will_fill_up_column(self):
gb = GameBoard(4, 4, 2)
gb.add_to_column(1, self.p1)
self.assertEqual(gb.grid[1][0], self.p1)
gb.add_to_column(1, self.p1)
self.assertEqual(gb.grid[1][1], self.p1)
def test_a_horizonal_win_for_p2_and_test_game_state_changes_with_final_move(
self):
gb = GameBoard(4, 4, 4)
gb.grid = [[0], [0], [0], [0]]
game_state = gb.add_to_column(0, self.p2)
self.assertEqual(game_state, False)
game_state = gb.add_to_column(1, self.p2)
self.assertEqual(game_state, False)
game_state = gb.add_to_column(2, self.p2)
self.assertEqual(game_state, False)
game_state = gb.add_to_column(3, self.p2)
self.assertEqual(game_state, True)
def test_simple_vertical_win_for_p1(self):
gb = GameBoard(4, 4, 4)
gb.grid = [[1, 1, 1], [2, 2, 2], [], []]
game_state = gb.add_to_column(0, self.p1)
self.assertEqual(game_state, True)
def test_non_win(self):
gb = GameBoard(4, 4, 4)
gb.grid = [[1, 1, 1], [2, 2, 2], [], []]
game_state = gb.add_to_column(3, self.p1)
self.assertEqual(game_state, False)
def test_reverse_diagonal_win_for_p1(self):
gb = GameBoard(4, 4, 4)
gb.grid = [[0, 0, 0], [0, 0, 1, 0], [0, 1, 0, 0], [1, 0, 0, 0]]
game_state = gb.add_to_column(0, self.p1)
self.assertEqual(game_state, True)
def test_diagonal_non_win_for_p2(self):
gb = GameBoard(4, 4, 4)
gb.grid = [[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0], [0, 0, 0]]
game_state = gb.add_to_column(3, self.p2)
self.assertEqual(game_state, False)
def test_simple_horizonal_win_for_p1(self):
gb = GameBoard(4, 4, 4)
gb.grid = [[], [1], [1], [1]]
game_state = gb.add_to_column(0, self.p1)
self.assertEqual(game_state, True)