ZdddZddlZGddejZGddejZdZdS)ztests for extending pyzmqNc2eZdZUeed<ddeffd ZxZS) CustomSocket custom_attrrcZt||||_dSN)super__init__r)selfcontext socket_typer __class__s 2lib/python3.11/site-packages/zmq/tests/test_ext.pyr zCustomSocket.__init__ s+ +...&)r)__name__ __module__ __qualname__int__annotations__r __classcell__r s@rrrsW''#''''''''''rrc6eZdZUeed<eZddeffd ZxZS) CustomContext extra_argxcVt||_dSr)rr r)r rr s rr zCustomContext.__init__s$ "r)r) rrrstrrr _socket_classr rrs@rrrsYNNN M#############rrctd}t|tsJ|jdksJ|tjd}t|t sJ|jdksJ|j|usJ|j tjksJ| | dS)Ns )r) r isinstancersocketzmqPUSHrrr typecloseterm)ctxrs rtest_custom_contextr)s   C c= ) )))) =C     38 ,,A a & &&&& =B     9     6SX    GGIIIHHJJJJJr)__doc__r#SocketrContextrr)rrr.s '''''3:'''#####CK###     r