ó šxŠ\c@s²ddlmZddlZddlZddlmZmZmZddl m Z m Z m Z m Z ddlmZddljjZddlmZdefd„ƒYZdS( iÿÿÿÿ(tprint_functionN(tPY36tlrangetrange(t DataFrametIndext MultiIndextSeries(tTestData(tassert_frame_equaltTestDataFrameMutateColumnscBs·eZd„Zd„Zd„Zd„Zejje ddƒd„ƒZ ejje ddƒd„ƒZ d „Z d „Z d „Zd „Zd „Zd„Zd„Zd„ZRS(cCs£tidddgd6dddgd6ƒ}|jƒ}|jd |j|jƒ}|jƒ}dd dg|d staxistcolumnsii cSs|j|jS(N(R R (R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyR4s(RtcopytassignR R R tdroptrename(tselftdftoriginaltresulttexpected((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyt test_assigns*,       (   cCs¿tddgddgddggddd gƒ}|jd d d d gd|jdd„ƒ}tddd ddgddd ddgddd ddggdtdƒƒ}t||ƒdS(NiiiiiiRR R R iii tDtEcSs|jS(N(R (R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyR9stABCDE(RRR tlistR (RRRR((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyttest_assign_multiple7s 3-''cCsKtddgddggdddgƒ}|jd|j|jd |j|jƒ}tr•tdddd gddd d ggdtd ƒƒ}n6tddd dgddd d ggdtd ƒƒ}t||ƒ|jd |j|jd|j|jƒ}tddd dgddd d ggdtd ƒƒ}t||ƒdS(NiiiiRR R RR iÿÿÿÿitABDCtABCD(RRR R RRR (RRRR((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyttest_assign_order>s*,'' ,'c Cstidddgd6dddgd6ƒ}tjtƒ|jd „ƒWdQXtjtƒ(|jd |jd |j|jƒWdQXdS( NiiiR iiiR cSs|jS(N(R (R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyRVsR R(Rtpytesttraisest TypeErrorRtAttributeErrorR R (RR((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyttest_assign_badQs ,treasonsDIssue #14207: valid for python 3.6 and abovec Cs”tidddgd6dddgd6ƒ}tjtƒ!|jd d „d d „ƒWdQXtjtƒ!|jd |jd d „ƒWdQXdS(NiiiR iiiR R cSs|jS(N(R (R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyRasRcSs|d|dS(NR R ((R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyRbscSs|d|dS(NR R ((R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyRds(RR$R%tKeyErrorRR (RR((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyt test_assign_dependent_old_pythonZs ,sHIssue #14207: not valid for python 3.5 and belowcCsìtiddgd6ddgd6ƒ}|jd|jdd „ƒ}tddddgddddggd td ƒƒ}t||ƒ|jdd „dd „ƒ}tddddgddddggd td ƒƒ}t||ƒdS(NiiR iiR R RcSs|d|dS(NR R ((R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyRksRR"cSs|jS(N(R (R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyRpscSs|d|dS(NR R ((R((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyRqs(RRR RR (RRRR((sElib/python2.7/site-packages/pandas/tests/frame/test_mutate_columns.pyttest_assign_dependentfs&'  'cCs!tidddgd6dddgd6d d d gd 6ƒjdƒ}tiddddgd6d dddgd6ƒjdƒ}d}tjtd|ƒ||ds  "