Tic-tac-toe simple solver

test_cases.py

from tictactoe import check


TESTCASES = {
    '#########': '#',
    'XXX######': 'X',
    '###XXX###': 'X',
    '######XXX': 'X',
    'X##X##X##': 'X',
    '#X##X##X#': 'X',
    '##X##X##X': 'X',
    'X###X###X': 'X',
    '##X#X#X##': 'X',
    'OOO######': 'O',
    '###OOO###': 'O',
    '######OOO': 'O',
    'O##O##O##': 'O',
    '#O##O##O#': 'O',
    '##O##O##O': 'O',
    'O###O###O': 'O',
    '##O#O#O##': 'O'
}


def test_check():
    for data, expected in TESTCASES.items():
        assert check(data) == expected

tictactoe.py

import sys


# Layout, in terms of array indexes:
# 012
# 345
# 678

VALID_WINS = [
    '012',
    '345',
    '678',
    '036',
    '147',
    '258',
    '048',
    '246'
]


def check(data):
    for condition in VALID_WINS:
        indexes = [int(e) for e in condition]
        if data[indexes[0]] == data[indexes[1]] == data[indexes[2]] and data[indexes[0]] != '#':
            return data[indexes[0]]
    return '#'


if __name__ == '__main__':
    test_data = sys.stdin.readline().strip()
    print(check(test_data))

Requires pytest to run the tests, or a bit of manual work.