multicall.py

def wrapper(iterable):
    print("start")
    results = yield from iterable
    print("stop")
    results[:] = [1, 1, 1]
    return results


def multicall(*callables):
    all_results = []
    for c in callables:
        result = c()
        all_results.append(result)
        yield result

    return all_results


results = wrapper(multicall(lambda: 1, lambda: 2, lambda: 3))
try:
    print(next(results))
    print(next(results))
    print(next(results))
    print(next(results))
except StopIteration as e:
    assert e.value == [1, 1, 1]
    print(e.value)