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)
