*VfjdZddlZGddejZGddejZdZy)ztests for extending pyzmqNc2eZdZUeed<ddeffd ZxZS) CustomSocket custom_attrc4t|||||_yN)super__init__r)selfcontext socket_typer __class__s [/var/www/html/software/conda/envs/catlas/lib/python3.12/site-packages/zmq/tests/test_ext.pyr zCustomSocket.__init__ s +.&)r)__name__ __module__ __qualname__int__annotations__r __classcell__r s@rrrs'#''rrc6eZdZUeed<eZddeffd ZxZS) CustomContext extra_argc0t|||_yr)rr r)r rr s rr zCustomContext.__init__s "r)x) rrrstrrr _socket_classr rrs@rrrsN M####rrctd}t|tsJ|jdk(sJ|jtj d}t|t sJ|jdk(sJ|j|usJ|jtj k(sJ|j|jy)Ns )r) r isinstancersocketzmqPUSHrrr typecloseterm)ctxrs rtest_custom_contextr)s  C c= )) ) ==C   388 ,A a && & ==B   99   66SXX  GGIHHJr)__doc__r#SocketrContextrr)rrr.s0 '3::'#CKK# r