merge simplified tree tests

2020-02-20 07:57:20 -03:00
parent 3deed0af7b
commit 1be97eea3e
8 changed files with 13 additions and 45 deletions
--- a/17-it-generator/tree/step2/test_tree.py
+++ b/17-it-generator/tree/step2/test_tree.py
@@ -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),
            ('Leaf1', 1),
            ('Leaf2', 1),
-            ('Leaf3', 1),
-            ('Leaf4', 1),
    ]
    result = list(tree(Branch))
    assert expected == result