def test_two_node_tree(self):
root = Node(None, None)
child = root.add_child("move")
self.assertTrue(root.is_root())
self.assertFalse(root.is_leaf())
self.assertFalse(child.is_root())
self.assertTrue(child.is_leaf())
self.assertIs(child.parent(), root)
self.assertEqual(child.action(), "move")
class TestMCTS(unittest.TestCase):
def setUp(self) -> None:
self.root = Node()
def test_is_leaf(self) -> None:
self.assertTrue(self.root.is_leaf())
def test_add_children(self) -> None:
children = [Estate(), Duchy(), Province()]
self.root.add_unique_children(children)
self.assertEquals(self.root.children[0].parent, self.root)
self.assertEquals(self.root.children[1].parent, self.root)
self.assertEquals(self.root.children[2].parent, self.root)
self.root.add_unique_children(children)
self.assertEquals(len(self.root.children), len(children))
def test_get_child(self) -> None:
children = [Estate(), Duchy(), Province()]
self.root.add_unique_children(children)
self.assertIsNotNone(self.root.get_child_node(Estate()))
self.assertIsNone(self.root.get_child_node(Colony()))
def test_single_node_tree(self):
n = Node(None, None)
self.assertTrue(n.is_root())
self.assertTrue(n.is_leaf())