ó ßüÚ\c @`sêddlmZmZmZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZmZddl m!Z!m"Z"d e#fd „ƒYZ$d „Z%d „Z&ee$e%e&ƒd „Z'd„Z(d„Z)d„Z*d„Z+ddlm,Z,ddl-m.Z.e,de/ƒd„ƒZ0e,de/ƒd„ƒZ1e,de/ƒd„ƒZ2de#fd„ƒYZ3d„Z4d„Z5ee3e4e5ƒd„Z6d„Z7d„Z8d„Z9d „Z:d!„Z;d"„Z<eƒd#„ƒZ=d$e#fd%„ƒYZ>dd&„Z@d'„ZAe,de/d(id)d*gd+6d,id)d*gd+6ƒd-„ƒZBe,de/ƒd.„ƒZCd/„ZDdS(0i(tprint_functiontdivisiontabsolute_importN(tidentity(twait( tregister_serializationt serializet deserializetnested_deserializet Serializet Serializedt to_serializetserialize_bytestdeserialize_bytestserialize_bytelisttregister_serialization_familytdask_serialize(tnbytes(tinctgen_test(t to_framest from_framestMyObjcB`seZd„Zd„ZRS(cC`s ||_dS(N(tdata(tselfR((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt__init__ scC`stdƒ‚dS(Ns Not picklable(t Exception(R((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt __getstate__#s(t__name__t __module__RR(((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRs cC`sitj|jƒgfS(N(tpickletdumpsR(tx((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pytserialize_myobj'scC`sttj|dƒƒS(Ni(RRtloads(theadertframes((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pytdeserialize_myobj+scC`sòx~ddddggD]g}t|ƒ\}}|ddksDt‚t|ƒdks\t‚t||ƒ}||kst‚qWtdƒ}t|ƒ\}}|ds¯t‚t|ƒdksÇt‚t||ƒ}|j|jksît‚dS(Ni{iiit serializerRttype(RtAssertionErrortlenRRR(R R#R$tresult((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_dumps_serialize2s cC`sjxcdtdƒfD]O}t|ƒ\}}|d|ksAt‚t||ƒ}||kst‚qWdS(Nt123t4567i(t bytearrayRR(R(tbR#R$tbb((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_serialize_bytestringsDs cC`sítdƒ}dt|ƒks$t‚|jdks9t‚tdƒ}t|ƒsWt‚tdƒ}||ksut‚||k sˆt‚||ksšt‚||k s­t‚t|ƒt|ƒksËt‚t|ƒt|ƒksét‚dS(Ni{R,ii(ii(R tstrR(Rthash(tstttu((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_SerializeLs   cC`s„ttdƒŒ}ttdƒŒ}ttdƒŒ}||ksHt‚||k s[t‚||ksmt‚||k s€t‚dS(Ni{ii(ii(R RR((R4R5R6((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_Serialized]scC`s¿idd6tdƒtdƒdgd6idttdƒŒgd 6d d 6d 6}tj|ƒ}t|ƒidd6dddgd6iddgd 6d d 6d 6ks©t‚||ks»t‚dS( Ntupdatetopi{iÈiR tabctdeftatghiR/ty(R R RtcopytdeepcopyRR((R tx_orig((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_nested_deserializegs* '(t gen_cluster(tdelayedtclientc c`sntdƒ}t|ƒ}ttƒ|ƒ}|j|ƒ}|V}t|tƒsUt‚|jdksjt‚dS(Ni{(RRERtcomputet isinstanceR(R( tcR4R=R/totvtv2tfutureR*((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_object_in_graph{s  cc`sntdƒ}|j|gƒV\}|j|ƒV|j|ƒV}t|tƒsUt‚|jdksjt‚dS(Ni{(Rt_scattert _replicatet_gatherRHR(R(RIR4R=R/RJRMto2((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt test_scatterˆs  cc`s„tdƒ}|j|gd|jƒV\}|jt|d|jƒ}|j|ƒV}t|tƒskt‚|jdks€t‚dS(Ni{tworkers( RROtaddresstsubmitRRQRHR(R(RIR4R=R/RJRMtfuture2RR((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_inter_worker_comms’s  tEmptycB`seZd„ZRS(cC`stdƒ‚dS(Ns Not picklable(R(R((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRs(RRR(((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRYœscC`s igfS(N((R ((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pytserialize_empty¡scC`stƒS(N(RY(R#R$((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pytdeserialize_empty¥scC`s4tƒ}tt|ƒŒ}t|tƒs0t‚dS(N(RYRRRHR((tete2((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt test_empty¬s cC`sWddlm}m}tƒ}||t|ƒgƒƒ}t|dtƒsSt‚dS(Ni(R"R(tdistributed.protocolR"RRYR RHR((R"RR\R]((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_empty_loads²s cC`seddlm}m}tƒ}||t|ƒgggƒƒ}t|dddtƒsat‚dS(Ni(R"R(R_R"RRYR RHR((R"RR\R]((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_empty_loads_deepºs !cC`sxxddtjdƒdtdƒgD]Q}t|ƒ}t|tƒsMt‚t|ƒ}t|ƒt|ƒks&t‚q&WdS(NiR;itabgЃA( tnptarangetintR RHtbytesR(R R2(R R/R?((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_serialize_bytesÂs ,  cC`s‡tjdƒtjdƒ}t|ƒ}ttt|ƒƒ|jdksPt‚dj |ƒ}t |ƒ}||kj ƒsƒt‚dS(Ntlz4i@Bit( tpytestt importorskipRctonesRtsumtmapRR(tjoinR tall(R tLR/R?((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_serialize_list_compressÊs  ( c`s(dtfd„ƒY‰dtf‡fd†ƒY}|ƒ}t|dgƒ\}}tjtƒ}t||ƒWdQXdt|jƒks“t‚dt|jƒks®t‚t|ddgƒ\}}tjtƒ}t||ƒWdQXdt|jƒks t‚dt|jƒks$t‚dS(Nt BadExceptioncB`seZd„ZRS(cS`s tdƒS(NsSneaky deserialization code(R(R((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt __setstate__×s(RRRt(((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRsÖstMyClassc`seZ‡fd†ZRS(c`s ˆƒ‚dS(N((R(Rs(sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRÛs(RRR((Rs(sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRuÚst serializerstSneakyR( RtobjectRRjtraisesRR2tvalueR((RutobjR#R$tinfo((RssHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_malicious_exceptionÕs c C`s²iittƒd6d6}t|dddgƒ\}}|ddksNt‚t|ddgƒ\}}|ddkst‚tjtƒt|ddgdd ƒWdQXdS( NtfooRRvtmsgpackRR&terrorton_errortraise(R RRR(RjRyt TypeError(tmsgR#R$((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt test_errorsïscc`s…titdƒd6ddgƒV}t|dddgƒV}|idd6ksUt‚tjtƒ}t|ddgƒVWdQXdS(NiÒR RvRt deserializersR~R(RR RR(RjRyRƒ(R$R*R|((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_err_on_bad_deserializerüs #tMyObjectcB`seZd„ZRS(cK`s|jj|ƒdS(N(t__dict__R9(Rtkwargs((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRs(RRR(((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRˆscC`sft|ƒjdkrYidd6}tj|jdtƒtj|dtƒg}||fStƒ‚dS(NRˆsmy-serR&t use_bin_type(R'RRRR‰tTruetNotImplementedError(R{tcontextR#R$((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pytmy_dumps s   cC`s?ttj|ddtƒ}tj|ddtƒ|_|S(Nitrawi(RˆRR"tFalseRŽ(R#R$R{((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pytmy_loadsst client_kwargssmy-serRRvt worker_kwargsc #`s5tdttƒz|jtddddd|jƒ}|jd„|d|jƒ}t|ƒV|j‰‡fd†}|j|d|jgƒV}i|jd 6|jd 6}||jd |jksÎt ‚|V} | j dkró| j dksùt ‚| j d |jkst ‚Wdd d l m} | d=XdS( Nsmy-serR iR?iRTcS`s|S(N((R ((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt,Ric`s|jˆ}|jS(N(RRŽ(t dask_workertmy_obj(tkey(sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pytcheck2s tsendert recipienti(tfamilies(RRR’RVRˆRURR˜trunR(R R?RŽtdistributed.protocol.serializeRœ( RIR4R=R/R R?R™R*texpectedtzRœ((R˜sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt#test_context_specific_serialization!s$   $ c #`sttttƒ|jtddddd|jƒ}|jd„|d|jƒ}t|ƒV|j‰‡fd†}|j|d|jgƒV}i|jd6|jd 6}||jd|jksËt ‚|V} | j dkrð| j dksöt ‚| j d|jkst ‚dS( NR iR?iRTcS`s|S(N((R ((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyR•KRic`s|jˆ}|jS(N(RRŽ(R–R—(R˜(sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyR™Qs RšR›( RRˆRR’RVRURR˜RR(R R?RŽ( RIR4R=R/R R?R™R*RŸR ((R˜sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyt)test_context_specific_serialization_classEs$   $cC`sxdtfd„ƒY}tj|ƒd„ƒ}tjtƒ}tt|ƒƒŒWdQXdt|j ƒkstt ‚dS(NtFoocB`seZRS((RR(((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyR£ascS`stdƒ‚dS(Ns Hello-123(R(tf((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyRdss Hello-123( RxRtregisterRjRyRRRR2RzR((R£RR|((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyttest_serialize_raises`s (Et __future__RRRR@RRtnumpyRcRjttoolzRt distributedRR_RRRRR R R R R RRRtdistributed.utilsRtdistributed.utils_testRRtdistributed.comm.utilsRRRxRR!R%R+R1R7R8RCRDtdaskRERŒRNRSRXRYRZR[R^R`RaRgRrR}R…R‡RˆtNoneRR’R¡R¢R¦(((sHlib/python2.7/site-packages/distributed/protocol/tests/test_serialize.pyts\     R                   !