def test_GameStateIsLegal(self):
gameState = GameState()
stoneGroups = self.InitializeGameStateStoneGroups()
board = self.InitilizeGameStateBoard()
assert(gameState.IsLegal(stoneGroups, board, (0, 1)) == False)
assert(gameState.IsLegal(stoneGroups, board, (1, 3)) == True)
def test_GameStateCopyStoneGroups(self):
gameState = GameState()
stoneGroups = self.InitializeGameStateStoneGroups()
stoneGroupsCopy = gameState.CopyStoneGroups(stoneGroups)
for stone in stoneGroups:
assert(stoneGroups[stone] == stoneGroupsCopy[stone])
stoneGroupsCopy[(2, 0)] = {(2, 0)}
assert(((2, 0) not in stoneGroups) == True)
assert(((2, 0) in stoneGroupsCopy) == True)
def test_GameStateCopyBoard(self):
gameState = GameState()
board = self.InitilizeGameStateBoard()
boardCopy = gameState.CopyBoard(board)
for rowIndex, row in enumerate(board):
for colIndex, stone in enumerate(row):
assert(stone == boardCopy[rowIndex][colIndex])
boardCopy[2][0] = 'W'
assert(board[2][0] == None)
assert(boardCopy[2][0] is not None)
def test_MergeGroups(self):
gameState = GameState()
stoneGroups = self.InitializeGameStateStoneGroups()
gameState.MergeGroups(stoneGroups, (3, 0), (3, 4))
assert(((3, 1) in stoneGroups[(4, 2)]) == True)
for stone in stoneGroups[(3, 0)]:
assert((stone in stoneGroups[3, 4]) == True)
for stone in stoneGroups[(3, 4)]:
assert((stone in stoneGroups[3, 0]) == True)
def test_MergeSetsInDict(self):
gameState = GameState()
stoneGroups = self.InitializeGameStateStoneGroups()
testDict = {}
testDict[(0, 0)] = {(0, 0)}
testDict[(1, 1)] = {(1, 1)}
gameState.MergeSetsInDict(testDict, (0, 0), (1, 1))
assert(((0, 0) in testDict[(0, 0)]) == True)
assert(((1, 1) in testDict[(0, 0)]) == True)
assert(((0, 0) in testDict[(1, 1)]) == True)
assert(((1, 1) in testDict[(1, 1)]) == True)
def test_GetActions(self):
gameState = GameState()
gameState.board = self.InitilizeGameStateBoard()
gameState.stoneGroups = self.InitializeGameStateStoneGroups()
actions = gameState.GetActions()
expected = {(1, 3), (1, 4), (2, 0), (2, 1), (2, 2), (2, 3), (2, 4), "pass"}
assert(((0, 1) in actions) == False)
assert(((1, 3) in actions) == True)
for e in expected:
assert((e in actions) == True)
for k in actions:
assert ((k in expected) == True)
def test_MergeNeighboringStones(self):
gameState = GameState()
stoneGroups = self.InitializeGameStateStoneGroups()
board = self.InitilizeGameStateBoard()
board[2][0] = 'W'
stoneGroups[(2, 0)] = {(2, 0)}
gameState.MergeNeighboringStones((2, 0), stoneGroups, board)
gameState.MergeNeighboringStones((2, 0), stoneGroups, board)
assert(((2, 0) in stoneGroups[(3, 0)]) == True)
for stone in stoneGroups[(3, 0)]:
assert((stone in stoneGroups[2, 0]) == True)
for stone in stoneGroups[(2, 0)]:
assert((stone in stoneGroups[3, 0]) == True)
def test_SwapMany(self):
gameState = GameState()
gameState.board = self.InitilizeGameStateBoard()
gameState.stoneGroups = self.InitializeGameStateStoneGroups()
gameState.turn = 'W'
newGameState = gameState.Result((2, 4))
expectedBoardAfterMove = [
['B', None, 'W', 'B', 'W'],
['W', 'W', 'B', None, None],
[None, None, None, None, 'W'],
['W', 'W', 'W', 'W', 'W'],
['B', 'W', 'W', 'W', 'W']
]
expectedGroupAfterMove = {(2, 4), (3, 0), (3, 1), (3, 2), (3, 3), (3, 4), (4, 1), (4, 2), (4, 3), (4, 4)}
for rowIndex, row in enumerate(newGameState.board):
for colIndex, stone in enumerate(row):
assert(stone == expectedBoardAfterMove[rowIndex][colIndex])
for stone in newGameState.stoneGroups[(4, 1)]:
assert((stone in expectedGroupAfterMove) == True)
for stone in expectedGroupAfterMove:
assert((stone in newGameState.stoneGroups[(4, 1)]) == True)
for stone in newGameState.stoneGroups[(2, 4)]:
assert((stone in expectedGroupAfterMove) == True)
for stone in expectedGroupAfterMove:
assert((stone in newGameState.stoneGroups[(2, 4)]) == True)
def test_UpdateStoneCount(self):
gameState = GameState()
gameState.board = self.InitilizeGameStateBoard()
gameState.stoneGroups = self.InitializeGameStateStoneGroups()
gameState.UpdateGameStateVariables()
assert(gameState.Utility() == (8-9))
assert(gameState.turn == 'W')