test.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

class Context():
    def __enter__(self):
        print('__enter__')

    def __exit__(self, *args):
        print('__exit__')

def gen():
    with Context():
        yield 1


for i in gen():
    print(i / 0)