U lu fã@s<dZddlZGdd„dejƒZGdd„dejƒZdd„ZdS) ztests for extending pyzmqéNcs.eZdZUeed<dedœ‡fdd„ Z‡ZS)Ú CustomSocketÚ custom_attrr©rcstƒ ||¡||_dS©N)ÚsuperÚ__init__r)ÚselfÚcontextÚ socket_typer©Ú __class__©ú6/tmp/pip-unpacked-wheel-h6ekxre8/zmq/tests/test_ext.pyr szCustomSocket.__init__)r)Ú__name__Ú __module__Ú __qualname__ÚintÚ__annotations__rÚ __classcell__r r r rrs rcs2eZdZUeed<eZdedœ‡fdd„ Z‡ZS)Ú CustomContextÚ extra_argÚx)rcstƒ ¡||_dSr)rrr)rrr r rrs zCustomContext.__init__)r) rrrÚstrrrZ _socket_classrrr r r rrs rcCs‚tdƒ}t|tƒst‚|jdks$t‚|jtjdd}t|tƒsBt‚|jdksPt‚|j |ks^t‚|j tjksnt‚|  ¡|  ¡dS)NÚsé r) rÚ isinstanceÚAssertionErrorrÚsocketÚzmqZPUSHrrr ÚtypeÚcloseZterm)Úctxrr r rÚtest_custom_contextsr")Ú__doc__rZSocketrÚContextrr"r r r rÚs