def test_nested(self): def myg(items): for x in items: yield x flat = flat_generator(myg([1, myg([2, myg([3, myg([4, myg([5])])])])])) assert [1, 2, 3, 4, 5] == [f[0] for f in flat]