from .utils import listify
def test_listify():
@listify
def gen():
yield 1
yield 2
assert [1, 2] == gen()