def test_diagonal_winner():
    text = """\
OX.
XOO
XXO
"""
    board = TicTacToeState(text)
    expected_winner = board.O_PLAYER
    winner = board.get_winner()

    assert winner == expected_winner
def test_horizontal_winner():
    text = """\
XXX
.OO
...
"""
    board = TicTacToeState(text)
    expected_winner = board.X_PLAYER
    winner = board.get_winner()

    assert winner == expected_winner
def test_vertical_winner():
    text = """\
XXO
.OO
XXO
"""
    board = TicTacToeState(text)
    expected_winner = board.O_PLAYER
    winner = board.get_winner()

    assert winner == expected_winner
def test_no_winner():
    text = """\
XX.
.OO
...
"""
    board = TicTacToeState(text)
    expected_winner = board.NO_PLAYER
    winner = board.get_winner()

    assert winner == expected_winner