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")
def test_single_node_tree(self): n = Node(None, None) self.assertTrue(n.is_root()) self.assertTrue(n.is_leaf())