def test_diagonal_left_to_right():
sm = StateManager()
class MockState:
def __init__(self):
self.board = {
"status": [[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0], [0, 0, 0, 1]],
"size": 4
}
assert sm.is_game_complete(MockState()) == (True, False)
def test_win_middle_column():
sm = StateManager()
class MockState:
def __init__(self):
self.board = {
"status": [[0, 1, 0], [0, 1, 0], [0, 1, 0]],
"size": 3
}
assert sm.is_game_complete(MockState()) == (True, False)
def test_incomplete_left_column_one_move():
sm = StateManager()
class MockState:
def __init__(self):
self.board = {
"status": [[1, 0, 0], [0, 0, 0], [0, 0, 0]],
"size": 3
}
assert sm.is_game_complete(MockState()) == (False, False)
def test_cats_game():
sm = StateManager()
class MockState:
def __init__(self):
self.board = {
"status": [[1, 2, 1, 2], [2, 1, 2, 1], [2, 1, 2, 1], [1, 2, 1, 2]],
"size": 4
}
assert sm.is_game_complete(MockState()) == (True, True)
