diff --git a/tests/test_treelib.py b/tests/test_treelib.py index d007824..28352b4 100644 --- a/tests/test_treelib.py +++ b/tests/test_treelib.py @@ -200,7 +200,7 @@ def test_leaves(self): for nid in self.tree.expand_tree(): self.assertEqual((self.tree[nid].is_leaf()) == (self.tree[nid] \ in leaves), True) - leaves = self.tree.leaves(root='jane') + leaves = self.tree.leaves(nid='jane') for nid in self.tree.expand_tree(nid='jane'): self.assertEqual(self.tree[nid].is_leaf() == (self.tree[nid] in leaves), True)