35 lines
662 B
Python
35 lines
662 B
Python
from tree import tree
|
|
|
|
|
|
def test_1_level():
|
|
class One: pass
|
|
expected = [('One', 0)]
|
|
result = list(tree(One))
|
|
assert expected == result
|
|
|
|
|
|
def test_2_levels_2_leaves():
|
|
class Branch: pass
|
|
class Leaf1(Branch): pass
|
|
class Leaf2(Branch): pass
|
|
expected = [
|
|
('Branch', 0),
|
|
('Leaf1', 1),
|
|
('Leaf2', 1),
|
|
]
|
|
result = list(tree(Branch))
|
|
assert expected == result
|
|
|
|
|
|
def test_3_levels_1_leaf():
|
|
class X: pass
|
|
class Y(X): pass
|
|
class Z(Y): pass
|
|
expected = [
|
|
('X', 0),
|
|
('Y', 1),
|
|
('Z', 2),
|
|
]
|
|
result = list(tree(X))
|
|
assert expected == result
|