contextmanager のテスト

contextman.py

#-*- coding:utf-8 -*-
import contextlib

def gencon():
    u'''
    ふつうの contextmanager
    '''

    print 'enter'

    yield 10

    print 'exit'



def gencon():
    u'''
    __exit__ で 'generator が終わってねーぞあほか' って言われる
    '''

    print 'enter'

    yield 10

    print 'exit'

    yield 20


if __name__ == '__main__':
    
    with contextlib.GeneratorContextManager(gencon()) as x:
        print x

yield書かずに関数抜ければStopIteration例外がでるのでその場合の例。

@contextmanager
def foo():
    print 'enter'
    return 1
    print 'what?'
    yield 2
    print 'exit'

with foo() as x:
    print x

RuntimeError: generator didn't yield