def rollout_node():
model = mock_model.MockModel()
# testing: model responds with all ones
when(model).inference(...).thenReturn((np.ones(MOVE_TENSOR_SHAPE), 1))
statemachine = state_machine.StateMachine()
node = azts_node.AztsNode(statemachine, model)
for i in range(10):
node.rollout()
return node
def _init_tree(self):
'''
keep this as separate function
because it needs to be called
after every move if tree is
not reused
'''
self.root = azts_node.AztsNode(\
statemachine=self.statemachine, \
model=self.model, \
color=self.color, \
exploration=self.exploration, \
payoffs=self.payoffs)
def lose_node():
model = mock_model.MockModel()
statemachine = state_machine.StateMachine()
statemachine.set_to_fen_state(WIN_STATE)
node = azts_node.AztsNode(statemachine, model, WHITE)
return node
def stale_node():
model = mock_model.MockModel()
statemachine = state_machine.StateMachine()
statemachine.set_to_fen_state(STALE_MATE)
node = azts_node.AztsNode(statemachine, model, BLACK)
return node
def initial_node():
model = mock_model.MockModel()
statemachine = state_machine.StateMachine()
node = azts_node.AztsNode(statemachine, model)
return node