From bd53bbdf448ce30ec45a69fa54c5cae017abbba5 Mon Sep 17 00:00:00 2001 From: Xiaming Chen Date: Fri, 11 Nov 2016 10:49:24 +0800 Subject: [PATCH] Fix #64 update test cases --- tests/test_treelib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)