merge simplified tree tests

17-it-generator/tree/step5/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