Exemplos de SpedNode.filter em Python

Exemplo n.º 1

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]

Exemplo n.º 2

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