def test_filter_tree_complex_case(): child = SpedNode('child', []) parent = SpedNode('', [child]) uncle = SpedNode('', []) root = SpedNode('', [parent, uncle]) root.filter(lambda n: n.values[0] != 'child') assert list(root) == [root, parent, uncle]
def test_filter_tree_simple_case(): node = SpedNode('', [ SpedNode('foo', []), SpedNode('bar', []) ]) node.filter(lambda n: n.values[0] != 'foo') expected = SpedNode('', [SpedNode('bar', [])]) assert node == expected