diff --git a/src/tests/util.py b/src/tests/util.py --- a/src/tests/util.py +++ b/src/tests/util.py @@ -11,10 +11,14 @@ def simpleLoadSgf(filename): return [g(m) for m in re.finditer(r"\b[BW]\[([a-z]{2})\]",contents)] -def listStates(moves): +def listBoards(moves): g=Go() - res=[BoardState(g.board)] + res=[tuple(list(x for x in row) for row in g.board)] for m in moves: g.doMove(g.toMove,*m) - res.append(BoardState(g.board)) + res.append(tuple(list(x for x in row) for row in g.board)) return res + + +def listStates(moves): + return [BoardState(b) for b in listBoards(moves)]