ó ßüÚ\c@sddlZddlZejdƒZddlmZddlmZm Z ej ie dƒd6ddd gd 6ƒZ ej je d eƒZejje d eƒZejjd eegd ddgƒd„ƒZd„Zejjd eegd ddgƒd„ƒZdS(iÿÿÿÿNtpyarrow(t gen_cluster(t deserializet serializetabctAiiitBtpreserve_indextobjtidst RecordBatchtTablecCs:t|ƒ\}}t||ƒ}|j|ƒs6t‚dS(N(RRtequalstAssertionError(Rtheadertframestnew_obj((sDlib/python2.7/site-packages/distributed/protocol/tests/test_arrow.pyttest_roundtripscCs|S(N((targ((sDlib/python2.7/site-packages/distributed/protocol/tests/test_arrow.pytechoscs)tdtƒ‡fd†ƒ}|ƒdS(Ntclientc3sB|jˆƒV}|jt|ƒ}|V}ˆj|ƒs>t‚dS(N(tscattertsubmitRR R (Rt schedulertworker1tworker2tobj_futtfuttresult(R(sDlib/python2.7/site-packages/distributed/protocol/tests/test_arrow.pytrun_tests(RtTrue(RR((RsDlib/python2.7/site-packages/distributed/protocol/tests/test_arrow.pyt test_scatters(tpandastpdtpytestt importorskiptpatdistributed.utils_testRtdistributed.protocolRRt DataFrametlisttdfR t from_pandastFalsettblR tbatchtmarkt parametrizeRRR(((sDlib/python2.7/site-packages/distributed/protocol/tests/test_arrow.pyts  ,0