def should_add_node(self): root = StringTree('root') node = root.add_node(['node 1', 'node 1.0', 'node 1.0.0']) assert len(root.childs) == 1 assert len(root.find('node 1').childs) == 1 assert len(root.find('node 1').find('node 1.0').childs) == 1 assert node.path() == ['root', 'node 1', 'node 1.0', 'node 1.0.0']
def should_find_childs_by_name(self): root = StringTree('root') name_11 = ['node 1', 'node 11'] name_21 = ['node 2', 'node 21'] node_11 = StringTree(name_11, root) node_21 = StringTree(name_21, root) assert node_11.parent == root.find('node 1') assert node_11.parent.find('node 11') == node_11 assert node_21.parent == root.find('node 2')