def test_get_valid_moves(): text = """\ X.. .O. ... """ expected_moves = np.array([0, 1, 1, 1, 0, 1, 1, 1, 1]) board = TicTacToeState(text) moves = board.get_valid_moves() assert np.array_equal(moves, expected_moves)
def test_get_valid_moves_after_win(): """ Win gets checked in search, so don't also check in valid moves. """ text = """\ XXX OO. ... """ expected_moves = np.array([0, 0, 0, 0, 0, 1, 1, 1, 1]) board = TicTacToeState(text) moves = board.get_valid_moves() assert np.array_equal(expected_moves, moves)