about summary refs log tree commit diff
path: root/scratch/facebook/moderate/tic-tac-toe-checker.py
blob: 342c29be6bdbdbf3969b8f8c3b3b9a6508c3c037 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import random

def print_board(board):
    result = []
    for row in range(len(board)):
        r = []
        for col in range(len(board[row])):
            cell = board[row][col]
            if not cell:
                r.append("-")
            else:
                r.append(cell)
        result.append(" | ".join(r))
    print("\n---------\n".join(result))

def init_board():
    result = []
    for row in range(3):
        r = []
        for col in range(3):
            r.append(None)
        result.append(r)
    return result

def check(board, player):
    print_board(board)
    print()
    if player not in "XO":
        raise Exception("Only checking the board for Xs or Os. You supplied {}".format(player))
    dn, ax, ddg, udg = "DOWN", "ACROSS", "DOWN_DIAGONAL", "UP_DIAGONAL"
    ways = [
        [[dn, ax, ddg], [dn], [dn, udg]],
        [[ax], [], []],
        [[ax], [], []],
    ]
    for row in range(len(board)):
        for col in range(len(board[row])):
            if board[row][col] == player:
                xs = ways[row][col]
                for x in xs:
                    if x == dn:
                        if {player} == {board[row+1][col], board[row+2][col]}:
                            return True
                    if x == ax:
                        if {player} == {board[row][col+1], board[row][col+2]}:
                            return True
                    if x == ddg:
                        if {player} == {board[row+1][col+1], board[row+2][col+2]}:
                            return True
                    if x == udg:
                        if {player} == {board[row+1][col-1], board[row+2][col-2]}:
                            return True
    return False

def op(player):
    return "X" if player == "O" else "O"

dn_win = lambda p: [
    [op(p), p, None],
    [op(p), p, None],
    [None,  p, None],
]

ax_win = lambda p: [
    [p, p, p],
    [op(p), op(p), None],
    [None, None, None],
]

ddg_win = lambda p: [
    [p, None, None],
    [op(p), p, None],
    [op(p), None, p],
]

udg_win = lambda p: [
    [op(p), None, p],
    [op(p), p, None],
    [p, None, None],
]

# Down
p = random.choice(["X", "O"])
assert check(dn_win(p), p) == True
assert check(dn_win(p), op(p)) == False
# Across
p = random.choice(["X", "O"])
assert check(ax_win(p), p) == True
assert check(ax_win(p), op(p)) == False
# Down Diagonally
p = random.choice(["X", "O"])
assert check(ddg_win(p), p) == True
assert check(ddg_win(p), op(p)) == False
# Down Diagonally
p = random.choice(["X", "O"])
assert check(udg_win(p), p) == True
assert check(udg_win(p), op(p)) == False
# Success
print("Tests pass!")