def __exit__(self, type, value, traceback): self.exit_called = True self.exit_args = (type, value, traceback) return GeneratorContextManager.__exit__(self, type, value, traceback)
def __init__(self, gen): GeneratorContextManager.__init__(self, gen) self.enter_called = False self.exit_called = False self.exit_args = None
def __enter__(self): self.enter_called = True return GeneratorContextManager.__enter__(self)
def __init__(self, f, *a, **k): return GeneratorContextManager.__init__(self, f(*a, **k))