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)