def test_get_expected_success_rate(self):
game = TestGame()
node = Node(game)
node.cumulative_score = 7
node.visits_count = 2
self.assertEqual(node.get_expected_success_rate(), 2)
def test_update_uct_score(self):
game = TestGame()
parent = Node(game)
parent.visits_count = 1
node = Node(game, parent=parent)
node.cumulative_score = 1
node.visits_count = 1
node.update_uct_score()
self.assertEqual(node.uct_score, 1)
def test_select_child(self):
game = TestGame()
node = Node(game)
node.visits_count = 1
child1 = Node(game, parent=node)
child1.visits_count = 1
child2 = Node(game, parent=node)
child2.visits_count = 1
child2.cumulative_score = 2
child3 = Node(game, parent=node)
child3.visits_count = 1
node.children = [child1, child2, child3]
self.assertEqual(node.select_child(), child2)
