def test_extract_data_normal_end(self): player = MCTSPlayer(DummyNet()) player.initialize_game() player.tree_search() player.play_move(None) player.tree_search() player.play_move(None) self.assertTrue(player.root.is_done()) player.set_result(player.root.position.result(), was_resign=False) data = list(player.extract_data()) self.assertEqual(2, len(data)) position, _, result = data[0] # White wins by komi self.assertEqual(go.WHITE, result) self.assertEqual("W+{}".format(player.root.position.komi), player.result_string)
def test_extract_data_resign_end(self): player = MCTSPlayer(DummyNet()) player.initialize_game() player.tree_search() player.play_move((0, 0)) player.tree_search() player.play_move(None) player.tree_search() # Black is winning on the board self.assertEqual(go.BLACK, player.root.position.result()) # But if Black resigns player.set_result(go.WHITE, was_resign=True) data = list(player.extract_data()) position, _, result = data[0] # Result should say White is the winner self.assertEqual(go.WHITE, result) self.assertEqual("W+R", player.result_string)