def g1():
yield 1
yield 2
def g2():
for i in range(2):
yield from g1()
def g3():
data = yield 1
yield data
class A:
def __init__(self):
self.items = [1, 2, 3]
def __iter__(self):
for x in self.items:
yield x
assrt.deepEqual([x for x in g1()], [1, 2])
assrt.deepEqual([x for x in g2()], [1, 2, 1, 2])
assrt.deepEqual([x for x in A()], [1, 2, 3])
g = g3()
assrt.equal(g.next().value, 1)
assrt.equal(g.next('a').value, 'a')
a = (x for x in range(3))
assrt.deepEqual(list(a), [0, 1, 2])
a = ([x, x**2] for x in range(3))
assrt.deepEqual(list(a), [[0, 0], [1, 1], [2, 4]])
assrt.deepEqual(list(x for x in range(3)), [0, 1, 2])
def t(a, b):
assrt.deepEqual(list(a), list(b))
t((x for x in range(1)), (y for y in range(1)))
