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)