blueloveTH 2 лет назад
Родитель
Сommit
399170727c
1 измененных файлов с 10 добавлено и 1 удалено
  1. 10 1
      tests/45_yield.py

+ 10 - 1
tests/45_yield.py

@@ -36,4 +36,13 @@ def f():
     yield from g()
 
 a = [i for i in f()]
-assert a == [1, 2, 3]
+assert a == [1, 2, 3]
+
+def f(n):
+    for i in range(n):
+        yield i
+        for j in range(i):
+            yield j
+
+t = f(3)
+assert list(t) == [0, 1, 0, 2, 0, 1]