1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
|
# -*- coding: utf-8 -*-
from anytree import Node
from anytree.util import commonancestors, leftsibling, rightsibling
from .helper import eq_
def test_commonancestors():
"""commonancestors."""
udo = Node("Udo")
marc = Node("Marc", parent=udo)
lian = Node("Lian", parent=marc)
dan = Node("Dan", parent=udo)
jet = Node("Jet", parent=dan)
joe = Node("Joe", parent=dan)
eq_(commonancestors(jet, joe), (udo, dan))
eq_(commonancestors(jet, marc), (udo,))
eq_(commonancestors(jet), (udo, dan))
eq_(commonancestors(), ())
eq_(commonancestors(jet, lian), (udo,))
def test_leftsibling():
"""leftsibling."""
dan = Node("Dan")
jet = Node("Jet", parent=dan)
jan = Node("Jan", parent=dan)
joe = Node("Joe", parent=dan)
eq_(leftsibling(dan), None)
eq_(leftsibling(jet), None)
eq_(leftsibling(jan), jet)
eq_(leftsibling(joe), jan)
def test_rightsibling():
"""rightsibling."""
dan = Node("Dan")
jet = Node("Jet", parent=dan)
jan = Node("Jan", parent=dan)
joe = Node("Joe", parent=dan)
eq_(rightsibling(dan), None)
eq_(rightsibling(jet), jan)
eq_(rightsibling(jan), joe)
eq_(rightsibling(joe), None)
|