class TestColumns(unittest.TestCase):
def setUp(self):
self.game = Board()
def test_column1_player1(self):
self.game.board = [['X', '_', '_'], ['X', '_', '_'], ['X', '_', '_']]
self.assertEqual(self.game.check_columns(), 1)
def test_column1_player2(self):
self.game.board = [['O', '_', '_'], ['O', '_', '_'], ['O', '_', '_']]
self.assertEqual(self.game.check_columns(), 2)
def test_column2_player1(self):
self.game.board = [['_', 'X', '_'], ['_', 'X', '_'], ['_', 'X', '_']]
self.assertEqual(self.game.check_columns(), 1)
def test_column2_player2(self):
self.game.board = [['_', 'O', '_'], ['_', 'O', '_'], ['_', 'O', '_']]
self.assertEqual(self.game.check_columns(), 2)
def test_column3_player1(self):
self.game.board = [['_', '_', 'X'], ['_', '_', 'X'], ['_', '_', 'X']]
self.assertEqual(self.game.check_columns(), 1)
def test_column3_player2(self):
self.game.board = [['_', '_', 'O'], ['_', '_', 'O'], ['_', '_', 'O']]
self.assertEqual(self.game.check_columns(), 2)
def test_empty(self):
self.game.board = [['_', '_', '_'], ['_', '_', '_'], ['_', '_', '_']]
self.assertEqual(self.game.check_columns(), 0)
def test_no_cols(self):
self.game.board = [['X', 'X', 'X'], ['_', 'O', 'O'], ['_', 'X', 'O']]
self.assertEqual(self.game.check_columns(), 0)