test_pytest.py

import unittest

import ttt


class TestTTT(unittest.TestCase):

    def test_invalid_wrong_size(self):
        board = [
            'x', '.', 'o',
        ]
        self.assertEqual(ttt.GameStates.invalid, ttt.game_state(board))

    def test_invalid_illegal_char(self):
        board = [
            'x', '.', 'o',
            'x', '.', 'o',
            'x', '.', 'O',
        ]
        self.assertEqual(ttt.GameStates.invalid, ttt.game_state(board))

    def test_invalid_too_many_o_moves(self):
        board = [
            'x', '.', 'o',
            'x', '.', 'o',
            'x', 'o', 'o',
        ]
        self.assertEqual(ttt.GameStates.invalid, ttt.game_state(board))

    def test_x_win_0(self):
        board = [
            'x', 'x', 'x',
            'o', '.', 'o',
            '.', '.', '.',
        ]
        self.assertEqual(ttt.GameStates.x_wins, ttt.game_state(board))

    def test_x_win_1(self):
        board = [
            'o', '.', 'o',
            'x', 'x', 'x',
            '.', '.', '.',
        ]
        self.assertEqual(ttt.GameStates.x_wins, ttt.game_state(board))

    def test_x_win_2(self):
        board = [
            'o', '.', 'o',
            '.', '.', '.',
            'x', 'x', 'x',
        ]
        self.assertEqual(ttt.GameStates.x_wins, ttt.game_state(board))

    def test_x_win_vert(self):
        board = [
            'x', '.', 'o',
            'x', '.', '.',
            'x', '.', 'o',
        ]
        self.assertEqual(ttt.GameStates.x_wins, ttt.game_state(board))

    def test_o_win_vert(self):
        board = [
            'x', 'x', 'o',
            '.', '.', 'o',
            'x', '.', 'o',
        ]
        self.assertEqual(ttt.GameStates.o_wins, ttt.game_state(board))

    def test_x_win_vert_middle(self):
        board = [
            'o', 'x', '.',
            'o', 'x', '.',
            '.', 'x', '.',
        ]
        self.assertEqual(ttt.GameStates.x_wins, ttt.game_state(board))

    def test_x_win_diag_0(self):
        board = [
            'x', '.', 'o',
            '.', 'x', 'o',
            '.', '.', 'x',
        ]
        self.assertEqual(ttt.GameStates.x_wins, ttt.game_state(board))

    def test_x_win_diag_1(self):
        board = [
            'x', '.', 'x',
            'o', 'x', 'o',
            'x', 'o', '.',
        ]
        self.assertEqual(ttt.GameStates.x_wins, ttt.game_state(board))

    def test_invalid_two_winners(self):
        board = [
            'o', 'x', '.',
            'o', 'x', '.',
            'o', 'x', '.',
        ]
        self.assertEqual(ttt.GameStates.invalid, ttt.game_state(board))

    def test_incomplete_0(self):
        board = [
            'x', 'x', 'o',
            'o', 'x', 'o',
            'x', 'o', '.',
        ]
        self.assertEqual(ttt.GameStates.unfinished, ttt.game_state(board))

    def test_draw(self):
        board = [
            'x', 'x', 'o',
            'o', 'x', 'x',
            'x', 'o', 'o',
        ]
        self.assertEqual(ttt.GameStates.draw, ttt.game_state(board))