Ejemplo n.º 1
0
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)