ch11-24: clean up by @eumiro & sync with Atlas

17-it-generator/tree/step2/test_tree.py

@@ -0,0 +1,21 @@
+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