ó ßüÚ\c@s’ddlZddlmZddlmZddlmZddlmZddl m Z m Z ddl m Z m Z mZddlmZdd lmZe d eƒd „ƒZe d eƒd „ƒZe d eƒd „ƒZe d eƒd„ƒZd„Ze d eƒd„ƒZd„Ze d eƒd„ƒZd„Zd„Zd„Zd„Z d„Z!d„Z"e d eƒd„ƒZ#dS(iÿÿÿÿN(tdelayed(tClient(t futures_of(ttime(t gen_clustertinc(tclienttcluster_fixturetloop(t Serialized(tgenRc csit|j|jfdtƒV}t|j|jfdtƒV}|jtdƒƒV}|jd|ƒV}d|jdjksƒt ‚t |jdjddt ƒsªt ‚t j tƒ}|jd|ƒV}WdQXdt|jƒksðt ‚dt|jƒks t ‚|jjƒV}|dks-t ‚|jjƒV}|dksOt ‚|jƒV|jƒVdS(Nt asynchronousitdatatpublishtexists(R (R (RtiptporttTruetscattertrangetpublish_datasett extensionstdatasetstAssertionErrort isinstanceR tpytesttraisestKeyErrortstrtvaluet schedulert publish_listtclose( tstatbtctfR touttexc_infotresult((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_publish_simple s ' c cs t|j|jfdtƒV}t|j|jfdtƒV}z¯x¨d ddgD]—}|jtdƒƒV}|j|d|ƒV}||jdjksŸt ‚t |jdj|d t ƒsÆt ‚|j j ƒV}||ksQt ‚qQWWd|jƒ|jƒXdS( NR R"R#g"@iitnameR R (R"R#(RRRRRRRRRRRR RRR ( R!R"R#R$R%R*R R&R((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_publish_non_string_key's$ c cs_t|j|jfdtƒV}t|j|jfdtƒV}|jdddgƒV}|jd|ƒVd|j|djksˆt‚|j ddƒV}t |ƒt |ƒks¹t‚|j |ƒV}|dddgksät‚t j tƒ}|j ddƒV}WdQXd t|jƒks*t‚dt|jƒksEt‚|jƒV|jƒVdS( NR iiiR spublished-dataR*t nonexistents not found(RRRRRRt who_wantstkeyRt get_datasettlentgatherRRRRRR ( R!R"R#R$R%R R(R&R'((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_publish_roundtrip=s  c cs|jdddgƒV}|jd|ƒV|dj}~|jjddƒVd|jdjksjt‚tƒ}x:||j kr¯t j dƒVtƒ|dksvt‚qvWt j tƒ}|jddƒV}WdQXd t|jƒksöt‚dt|jƒkst‚dS( NiiiR R*R g{®Gáz„?is not found(RRR.Rtpublish_deleteRRRRR-R tsleepRRRR/RR( R$R!R"R#R R.tstartR'R(((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_unpublishVs  cCsœ|jdddgƒ}|jd|ƒ|jddƒtjtƒ}|jddƒ}WdQXdt|jƒks}t ‚dt|jƒks˜t ‚dS(NiiiR R*s not found( RRtunpublish_datasetRRRR/RRR(RR R'R(((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_unpublish_syncnsccsmttƒdƒ}ttƒdƒ}|jd|d|ƒV|jjƒV}t|ƒddhksit‚dS(Niitxty(RRRRRtsetR(R$R!R"R#R9R:R((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_publish_multiple_datasetszs cCsEttƒdƒ}ttƒdƒ}|jd|d|ƒ|jddƒtjtƒ}|jddƒ}WdQX|jƒ}t |ƒdhks›t ‚dt |j ƒks¶t ‚dt |j ƒksÑt ‚|jddƒtjtƒ}|jddƒ}WdQXdt |j ƒks&t ‚dt |j ƒksAt ‚dS(NiiR9R:R*s not found( RRRR7RRRR/t list_datasetsR;RRR(RR9R:R'R(R((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyt%test_unpublish_multiple_datasets_sync„s c cs¯tjdƒ}t|j|jfdtƒV}t|j|jfdtƒV}|jdddgƒ}|j|ƒ}tt |ƒƒdks’t ‚d„t |ƒDƒ}|t |j ƒksÃt ‚|j d|ƒVtt |ƒƒdksòt ‚|jdƒV} t | j jƒƒt |j jƒƒks2t ‚d „| j jƒDƒd „|j jƒDƒksjt ‚|j| ƒV} | dddgks•t ‚|jƒV|jƒVdS( Nsdask.bagR iiiicSsh|]}|j’qS((R.(t.0R%((s=lib/python2.7/site-packages/distributed/tests/test_publish.pys ¦s R cSsh|]}|j’qS((R.(R?R%((s=lib/python2.7/site-packages/distributed/tests/test_publish.pys °s (Rt importorskipRRRRt from_sequencetpersistR0RRR;tdaskRR/tkeystvaluestcomputeR ( R!R"R#tdbR$R%tbagtbagpRDR(R&((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_publish_bagœs"08 cCsLxEdddgD]4}d}||j|<|j|ƒ|kst‚qWdS(NR.iR(R.R.(RR/R(RR.R((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_datasets_setitem¸s cCsPxIdddgD]8}d}|j|d|ƒ|j||kst‚qWdS(NR.iRR*(R.R.(RRR(RR.R((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_datasets_getitem¿scCsYxRdddgD]A}d}|j|d|ƒ|j|=||jƒkst‚qWdS(NR.iRR*(R.R.(RRR=R(RR.R((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_datasets_delitemÆs  cCsg|jd„tdƒDƒt|jjƒƒ}|gtdƒD]}t|ƒ^qBksct‚dS(NcSsi|]}|t|ƒ“qS((R(R?tn((s=lib/python2.7/site-packages/distributed/tests/test_publish.pys Ïs i (RRtlistRRDRR(RRDRN((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_datasets_keysÎscCs5d\}}|jd|ƒ||jks1t‚dS(NR.R(R.R(RRR(RR.R((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_datasets_containsÔs cCsrgtdƒD] }|^q }|jd„|Dƒx5t|jƒD]$\}}|t|ƒksFt‚qFWdS(Ni cSsi|]}|t|ƒ“qS((R(R?R.((s=lib/python2.7/site-packages/distributed/tests/test_publish.pys Üs (RRt enumerateRRR(RRNRDR.((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_datasets_iterÚsccsÔt|jdtddgƒV}z¢|jddddgƒV|jdƒV}|d ks^t‚tjtƒ|jdd „ƒVWdQX|jd d „ƒVtjtƒ|jd ƒVWdQXWd|j ƒVXdS( NR t serializerstmsgpackR9iiiR:cSs|S(N((R9((s=lib/python2.7/site-packages/distributed/tests/test_publish.pytêttzcSs|S(N((R9((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyRVìRW(iii( RtaddressRRR/RRRt TypeErrorR (R$R!R"R#tc2R(((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyttest_pickle_safeás($RRCRt distributedRtdistributed.clientRtdistributed.metricsRtdistributed.utils_testRRRRRtdistributed.protocolR ttornadoR tFalseR)R+R2RR6R8R<R>RJRKRLRMRPRQRSR\(((s=lib/python2.7/site-packages/distributed/tests/test_publish.pyts.