class TestDiagonal(unittest.TestCase):
def setUp(self):
self.game = Board()
def test_diagonal1_player1(self):
self.game.board = [['X', '_', '_'], ['_', 'X', '_'], ['_', '_', 'X']]
self.assertEqual(self.game.check_diagonals(), 1)
def test_diagonal1_player2(self):
self.game.board = [['O', '_', '_'], ['_', 'O', '_'], ['_', '_', 'O']]
self.assertEqual(self.game.check_diagonals(), 2)
def test_diagonal2_player1(self):
self.game.board = [['_', '_', 'X'], ['_', 'X', '_'], ['X', '_', '_']]
self.assertEqual(self.game.check_diagonals(), 1)
def test_empty(self):
self.game.board = [['_', '_', '_'], ['_', '_', '_'], ['_', '_', '_']]
self.assertEqual(self.game.check_diagonals(), 0)
def test_no_cols(self):
self.game.board = [['X', 'X', 'X'], ['_', 'O', 'O'], ['_', 'X', 'O']]
self.assertEqual(self.game.check_diagonals(), 0)
