def rubyyielder(gen):
    def wrapped_gen(block):
        for elem in gen():
            block(elem)
    return wrapped_gen

@rubyyielder
def test():
    print("In test")
    yield 1
    print("back in test")
    yield 2

@test
def _(a):
    print("You are in block %s" % a)