daviddoji/example-code-2e
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

merge simplified tree tests

Browse Source
This commit is contained in:
Luciano Ramalho
2020-02-20 07:57:20 -03:00
parent 3deed0af7b
commit 1be97eea3e
8 changed files with 13 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
17-it-generator/tree/extra/test_pretty_tree.py
View File

@@ -48,19 +48,15 @@ def test_4_levels_1_leaf():
assert expected == result
def test_2_levels_4_leaves():
def test_2_levels_2_leaves():
class Branch: pass
class Leaf1(Branch): pass
class Leaf2(Branch): pass
class Leaf3(Branch): pass
class Leaf4(Branch): pass
result = list(render_lines(tree(Branch)))
expected = [
'Branch',
'├── Leaf1',
'├── Leaf2',
'├── Leaf3',
'└── Leaf4',
'└── Leaf2',
]
assert expected == result

8
17-it-generator/tree/extra/test_tree.py
View File

@@ -8,18 +8,14 @@ def test_1_level():
assert expected == result
def test_2_levels_4_leaves():
def test_2_levels_2_leaves():
class Branch: pass
class Leaf1(Branch): pass
class Leaf2(Branch): pass
class Leaf3(Branch): pass
class Leaf4(Branch): pass
expected = [
('Branch', 0, True),
('Leaf1', 1, False),
('Leaf2', 1, False),
('Leaf3', 1, False),
('Leaf4', 1, True),
('Leaf2', 1, True),
]
result = list(tree(Branch))
assert expected == result