x\c@soddlmZddlmZddlZddlZddlZddlm Z m Z m Z ddl Z ddl mZmZddlmZddljjZddlmZmZmZddlZddlmZmZmZmZdd l m!Z!dd l"m#Z#dd l$m%Z%m&Z&ddl'j(j)j*Z*dd l'm+Z+m,Z,dd l-m.Z.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7jj8Z9ddl7m:Z:m;Z;m<Z<m=Z=m>Z?m@Z@e4ZAddfZBe jCdde&DdZDe jCde*jEdZFe jCdZGe jCdZHdZIdZJdZKdZLdZMdZNeOe1jPe5ZQejRdeSfdYZTejRd eTfd!YZUd"eUfd#YZVd$eVfd%YZWd&ZXd'eSfd(YZYd)ZZd*Z[d+eSfd,YZ\ejRd-eSfd.YZ]ejRd/e]fd0YZ^d1e^fd2YZ_d3e]fd4YZ`ejRd5eSfd6YZad7eafd8YZbd9eafd:YZcd;eafd<YZde d=Zed>eSfd?YZfejRd@ZgejRdAZhie,dB6e!jidC6e+dD6ZEe jjjkdEe&e jjjkdFeEdGZldHZmdIZndJZodKZpdLZqdMZrdNZse jjjkdOdPdQfe jjjkdReteufe jjjkdSeteufdTZvdUZwdVZxdWeSfdXYZydS(Yi(t LooseVersion(tproductN(trandtrandinttrandn(tPY3treduce(tPerformanceWarning(tis_boolt is_list_liket is_scalar(t DataFrametPaneltSeriest date_range(tpytables(t_NUMEXPR_VERSION(tNumExprClobberingErrort_engines(tPandasExprVisitortPythonExprVisitor(t_NUMEXPR_INSTALLEDt _USE_NUMEXPR(t_arith_ops_symst_binary_math_opst_binary_ops_dictt_bool_ops_symst_special_case_arith_ops_symst_unary_math_ops(tassert_frame_equaltassert_numpy_array_equaltassert_produces_warningtassert_series_equaltmakeCustomDataframetrandbooltinsnot intparamsc csU|]K}tj|dtjj|dko.t ddjdtdtVqdS(tmarkstnumexprtreasons2numexpr enabled->{enabled}, installed->{installed}tenabledt installedN(tpytesttparamtmarktskipifRtformatR(t.0tengine((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pys &scCs|jS(N(R+(trequest((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR0%s cCs|jS(N(R+(R1((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pytparser2scCs,tr(ttdkr(tjdndS(Ns2.6.9snumexpr is >= 2.6.9R&(RRRR*tskip(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt ne_lt_2_6_97scCsDtr3ttdkrtStdtDSn tjddS(Ns2.6.9css!|]}|dkr|VqdS(tfloortceilN(R5R6((R/tx((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pys Dssnumexpr is not present(RRRRttupleR*R3(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pytunary_fns_for_ne>s cCs t|jS(N(Rt has_neg_frac(R0((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pytengine_has_neg_fracJscCslt|}t|r_y|||SWq_tk r[}t|jdrUtjSq_Xn|||S(Ns6negative number cannot be raised to a fractional power(RR;t ValueErrortstrt startswithtnptnan(tlhstcmp1trhsR0tcte((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt_eval_single_binNs   cCs^t|tr0t|tjr0|jdkp]t|to]t|tjo]|jdkS(Ni(t isinstanceR R?tndarraytndim(RARC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt_series_and_2d_ndarray[s!cCs:t|trt|tp9t|to9t|tS(N(RGR R (RARC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt_series_and_framebscCs%t|to$t|tjjjS(N(RGtbooltpdtcoretgenerictNDFrame(RARC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt_bool_and_framegscCs+to*t|ttjfo*tj|S(N(RRGtcomplexR?tcomplexfloatingtisnan(tresulttexpected((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt_is_py3_complex_incompatkstTestEvalNumexprPandascBseZedZedZdZdZdZdZe j j dZ dZ e j j dZd Zd Zejd Ze j j d Ze j j d Ze j j dZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"ddZ#dZ$dZ%dZ&d Z'd!Z(d"Z)d#Z*d$Z+d%Z,d&Z-d'Z.d(Z/d)Z0RS(*cCs+ddl}||_d|_d|_dS(NiR&tpandas(R&tneR0R2(tclsRZ((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt setup_classvs   cCs(|`|`t|dr$|`ndS(NRZ(R0R2thasattrRZ(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pytteardown_class}s cCs.ttdd}tj||dks(RtRhtfilterRtcurrent_engines(Retmethod((ResAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt setup_methods  cCs.|`|`|`|`|`|`|`dS(N(RcRdRbRaR`R_Rw(ReRx((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pytteardown_methodscCsdd}d}xQt|j||j|j|D].\}}}}}|j|||||q.WdS( Ns!=s==s<=s>=t(s!=s==s<=s>=R{R|(R|R{(RRcRdRotcheck_complex_cmp_op(ReRmRnRARBRCtbinoptcmp2((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_complex_cmp_opss  cCsttddttdtf}ttddttdtf}x9t|||jD]"\}}}|j|||qpWdS(Ntsizei i(i i(i(i i(i(R R"R RRmtcheck_simple_cmp_op(Ret bool_lhsest bool_rhsesRARCtcmp_op((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_simple_cmp_opss %cCsFx?t|j|j|jD]"\}}}|j|||qWdS(N(RRcRrRdtcheck_binary_arith_op(ReRAtopRC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_binary_arith_opss+cCs=x6t|j|jD]\}}|j|d|qWdS(Nt%(RRcRdt check_modulus(ReRARC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt test_moduluss"cCs=x6t|j|jD]\}}|j|d|qWdS(Ns//(RRcRdtcheck_floor_division(ReRARC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_floor_divisions"cCs=x6t|j|jD]\}}|j|d|qWdS(Ns**(RRcRdt check_pow(ReRARC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_pows"cCsFx?t|j|j|jD]"\}}}|j|||qWdS(N(RRcRmRdtcheck_single_invert_op(ReRARRC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_single_invert_ops+cCsFx?t|j|j|jD]"\}}}|j|||qWdS(N(RRcRmRdtcheck_compound_invert_op(ReRARRC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_compound_invert_ops+cCsd|j}d}xNt|j||||jD].\}}}}}|j|||||q.WdS(NR{R|(R{R|(RcRRdtcheck_chained_cmp_op(RetmidsRmRARBtmidRRC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_chained_cmp_ops  %cCst|tr"tj||nYt|trDtj||n7t|tjritj||n||ks{t dS(N( RGR ttmRR R R?RHRtAssertionError(ReRURV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt check_equalsc Csyt}djd|d|d|}t|oE||kpE||k}|rtjt$tj|d|jd|j WdQXtjt8tj|d|jd|j di|d6|d 6WdQXnt ||||j} t ||||j} t | t r,t | t r,|tkr,nIt | || |j} tj|d|jd|j } |j| | dS( Ns)(lhs {cmp1} rhs) {binop} (lhs {cmp2} rhs)RBR~RR0R2t local_dictRARC(t _scalar_skipR.R R*traisest TypeErrorRMtevalR0R2RFRGR R t_series_frame_incompatibleR( ReRARBRCR~Rt skip_thesetextscalar_with_in_notintlhs_newtrhs_newRVRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR}s,  % !csfd}||||}||||}|dk r|dk rdj||} dj||} dj||} t|d|j} xK| | | fD]7} tj| djdj}tj|| qWndS(Ncst|||jS(N(RFR0(tlefttrightR(Re(sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pytcheck_operands sslhs {0} mid {1} rhsslhs {0} mid and mid {1} rhss(lhs {0} mid) & (mid {1} rhs)t&R0R2( tNoneR.RFR0RMRR2Rtassert_almost_equal(ReRARBRRRCRRRtex1tex2tex3RVRRU((ResAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR s c Csdj|}|d krgt| rgtjttj|d|jd|jdi|d6|d6nIt ||||j}tj|d|jd|j}|j ||dS( Ns lhs {0} rhsR#snot inR0R2RRARC(R#snot in( R.R R*RRRMRR0R2RFR(ReRARBRCRRVRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs!cCsdj|}tj|d|jd|j}t||||j}tj||dj|}tj|d|jd|j}t||||j}|j||||dS(Ns lhs {0} rhsR0R2slhs {0} rhs {0} rhs( R.RMRR0R2RFRRtcheck_alignment(ReRAtarith1RCRRURVtnlhs((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR)s!!  cCshy|j|\}}Wntttfk r5n/X|jjdj|}tj|j |dS(Ns nlhs {0} ghs( talignR<RtAttributeErrorRZtevaluateR.RRtvalues(ReRURtghsRRV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR5s cCsdj|}tj|d|jd|j}||}tj|||jjdj|}t |t t frtj|j |ntj||j dS(Ns lhs {0} rhsR0R2sexpected {0} rhs(R.RMRR0R2RRRZRRGR R Rtitem(ReRARRCRRURV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRDs! c Csdj|}|jdkr\tj|d|jd|j}||}|j||n<tjttj|di|d6|d6d|jd|jdS(Ns lhs {0} rhstpythonR0R2RRARC( R.R0RMRR2RR*RR(ReRARRCRtresRV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRPs! cCsyt|d||j}Wn\tk rz}t|jdrt|jdkrhtjt|qwtj}q{nX|S(Ns**s6negative number cannot be raised to a fractional powerR( RFR0R<R=R>R*R3R?R@(ReRARCRVRE((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pytget_expected_pow_result\s cCsdj|}|j||}tj|d|jd|j}t|rt|rt||rtj t t j ||nnt j ||dj|}tj|d|jd|j}|j|j|||}t j ||dS(Ns lhs {0} rhsR0R2s(lhs {0} rhs) {0} rhs(R.RRMRR0R2R RWR*RRRRR(ReRARRCRRVRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRjs! !c Csx||fD]}y|jt}Wn)tk rQtjt|g}nX|}tjdd|jd|j}t j ||x9|j D].}t j |tjdd|d|jqWq WdS(Ns~elbR0R2( tastypeRLRR?tarrayRMRR0R2RRRw( ReRARBRCteltelbRVRUR0((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR|s !c CsTd }dj|}t|rl||krltjttj|d|jd|jdi|d6|d6nt|rt|rt d ||f\}}nt ||||j}t|r| }n|}tj|d|jd|j}t j ||xB|j D]7}tj|d|jd|j} t j | |qWdS( NR#snot ins~(lhs {0} rhs)R0R2RRARCcSstj|gS(N(R?R(R7((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRus(R#snot in(R.R R*RRRMRR0R2tmapRFRRRw( ReRARBRCRRRVRUR0tev((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs"!  !!RAcCsdj||S(Ns{0}{1}(R.(ReRtvar_name((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRsc Cs=|jd}ttdd}|jdkrmtjt&tj|d|jd|j }WdQXn7tjt &tj|d|jd|j }WdQXtt ddd}|jdkrtjt&tj|d|jd|j }WdQXn5|}tj|d|jd|j }t ||tt dddk}|}tj|d|jd|j }t ||tid d d gd 6t d dkd6}|jdkrtjt&tj|d|jd|j }WdQXn7tjt &tj|d|jd|j }WdQXdS(NRjiiR&R0R2Rg?taig@tbiRD(ii(RR RR0R*RtNotImplementedErrorRMRR2RRRRR<(ReRkRARUtexpect((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_frame_inverts0*'*! ! /*c Cs|jd}ttd}|jdkrjtjt&tj|d|jd|j }WdQXn7tjt &tj|d|jd|j }WdQXtt ddd}|jdkrtjt&tj|d|jd|j }WdQXn5|}tj|d|jd|j }t ||tt ddk}|}tj|d|jd|j }t ||tdd d g}|jdkrtjt&tj|d|jd|j }WdQXn7tjt &tj|d|jd|j }WdQXdS( NRjiR&R0R2Rg?Rig@(RR RR0R*RRRMRR2RRR RR<(ReRkRARUR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_series_inverts0*'*! ! *c CsC|jd}ttdd}| }tj|d|jd|j}t||ttddd }| }tj|d|jd|j}t||tt dddk}|jdkr t j t &tj|d|jd|j}WdQXn5| }tj|d|jd|j}t||dS( NRiiiR0R2Rg?R&(ii( RR RRMRR0R2RRRR*RR(ReRkRARRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_frame_negates ! ! *!c Cs=|jd}ttd}| }tj|d|jd|j}t||ttddd}| }tj|d|jd|j}t||tt ddk}|jdkrt j t &tj|d|jd|j}WdQXn5| }tj|d|jd|j}t||dS(NRiiR0R2Rg?R&( RR RRMRR0R2R RRR*RR(ReRkRARRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_series_negates ! ! *!cCs|jd}ttdd}|}tj|d|jd|j}t||ttddd}|}tj|d|jd|j}t||tt dddk}|}tj|d|jd|j}t||dS( Nt+iiR0R2Rg?(ii( RR RRMRR0R2RRR(ReRkRARRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_frame_pos+s! ! !cCs|jd}ttd}|}tj|d|jd|j}t||ttddd}|}tj|d|jd|j}t||tt ddk}|}tj|d|jd|j}t||dS(NRiR0R2Rg?( RR RRMRR0R2R RR(ReRkRARRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_series_pos@s! ! !c Cs0tjt$tjdd|jd|jWdQXtjdd|jd|jdksbttjdd|jd|jd ksttjdd|jd|jdksttjd d|jd|jd ksttjd d|jd|jd ksttjd d|jd|jtksFttjdd|jd|jt kstttjdd|jd|jt ksttjdd|jd|jt ksttjdd|jd|jt ksttjdd|jd|jt ks,tdS(Ns~1.0R0R2s-1.0gs+1.0g?s~1is-1is+1s~Trues~Falses-Trues-Falses+Trues+Falsei( R*RRRMRR0R2RtTruetFalse(Re((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_scalar_unaryUs0%cCsWttjdtjt ttt t ttt dddd g dtjdS(NsK[-True, True, ~True, +True,-False, False, ~False, +False,-37, 37, ~37, +37]ii%tdtypei(RRMRR?RRRtobject_(Re((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_unary_in_arrayps  c Csd }|d7}|d7}|d7}|d7}tjjdd d ttdd f\}}}}xC|D];}tjt$tj|d |j d |j WdQXqnWdS(Ns1 or 2s1 and 2sa and bsa or bs1 or 2 and (3 + 2) > 3s2 * x > 2 or 1 and 2s2 * df > 3 and 1 or aiiiR0R2(s1 or 2s1 and 2(sa and bsa or b(s1 or 2 and (3 + 2) > 3(s2 * x > 2 or 1 and 2(s2 * df > 3 and 1 or a( R?trandomRR R*RRRMRR0R2(RetexprsR7RRtdfR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_disallow_scalar_bool_opszs    9 cCsd}tjdd|jd|j}|dks9tt|sKtd}tjdd|jd|j}|dkstt|stt}tjdd|jd|j}| stt|stt|sttj dg}tjdd|jd|j}t j |tj dg|j dksRttj dg}tjdd|jd|j}t j |tj dg|j dksttj tg}tjdd|jd|j}t j |tj tg|j dkstdS( NiR7R0R2g?(i(i(i( RMRR0R2RR RRR?RRRtshape(ReR7RU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_identicals2!!! !!!cCs=d}tj|d|jd|j}|dks9tdS(Ns1 + 2 * 5 - 1 + 2 R0R2i (RMRR0R2R(RetexpRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_line_continuations!cCs d}tj|d|jd|j}tj|}||ksHttjidddgd6}d}|jd |}|j std }|jd |}|j d d gddf}t j ||d}|jd|}|j d gddf}t j ||dS(Ns1000000000.006R0R2g~eAg $eAg'1eAtAgeAsA < %.4fg eAsA > %.4fiis A == %.4f( RMRR0R2R?tfloat64RR tquerytemptytlocRR(ReRRURVRtcutofftexact((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_float_truncations$! cCstjdddggddddg}d}tjtd||jdWdQXtj}d |j_tjtd||jd WdQXdS( Nitcolumnstfootbartclasss4Python keyword not valid identifier in numexpr querytmatchs class == 0tlambdas lambda == 0(RMR R*Rt SyntaxErrorRtindextname(ReRtmsg((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_disallow_python_keywordss*  (1t__name__t __module__t classmethodR\R^RhRtRyRzR*R,tslowRRRRRttdtskip_if_windowsRRRRRR}RRRRRRRRRRRRRRRRRRRRRRRR(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRXssR        &        ' +      !  tTestEvalNumexprPythoncBs)eZedZdZdZRS(cCs>tt|jddl}||_d|_d|_dS(NiR&R(tsuperRR\R&RZR0R2(R[RZ((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\s    cCsttdtj|_|jddd|_gtjD]}|dkrA|^qA|_t|_ t |_ d |_ dS( NcSs |dkS(NR#snot in(R#snot in((R7((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRusitandtorRRiRj(RR(RRiRj( tlistRvRkRlRmRnRRoRRpRqRrRs(Rets((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRts   c CsKdj||}tjt$tj|d|jd|jWdQXdS(Nslhs {0} mid {1} rhsR0R2(R.R*RRRMRR0R2(ReRARBRRRCR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs(RRRR\RtR(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs tTestEvalPythonPythoncBs)eZedZdZdZRS(cCs)tt|jd|_d|_dS(NR(RRR\R0R2(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\s cCsvdj|}tj|d|jd|j}||}tj||t||||j}tj||dS(Ns lhs {0} rhsR0R2(R.RMRR0R2RRRF(ReRARRCRRURV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs ! cCs_y|j|\}}Wntttfk r5n&Xtdj|}tj||dS(Ns nlhs {0} ghs(RR<RRRR.RR(ReRURRRRV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs (RRRR\RR(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs tTestEvalPythonPandascBs eZedZdZRS(cCs)tt|jd|_d|_dS(NRRY(RRR\R0R2(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\s cCs tj||||||dS(N(RXR(ReRARBRRRC((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs(RRRR\R(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR scOs tjjS(N(R?RR(targstkwargs((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRustTestTypeCastingcBsVeZejjddddddgejjdejejgdZRS(RRRit*s**t/tdtcCstdddtd|}dj|}tj|d|d|}|jj|ks]t|jj|ksutt|t|dj|}tj|d|d|}|jj|kst|jj|kstt|t|dS( Niit data_gen_fRsdf {} 3R0R2s3 {} df( tmkdftfR.RMRRRRR(ReR0R2RRRRR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_binop_typecasting!s( RRR*R,t parametrizeR?tfloat32RR(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR s!cCst|jjtjS(N(t issubclassRttypeR?t datetime64(R7((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt _is_datetime7scGsDtttjd| }ttjtt|}|oC|S(Nt is_monotonic(tanyRtoperatort attrgetterRtxorR (Rtnot_monot only_one_dt((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt should_warn;st TestAlignmentcBseZd ZedZdZdZdZejj dZ dZ d Z d Z ejj d Zd ZRS(tituRRcCsKd}tdddt}tj|d|d|}t||ddS( Nsdf * ~2iiRR0R2ii(RRRMRR(ReR0R2RRR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_align_nested_unary_opFsc Cst|j|j|j}tjdttjdtx|D]\}}}tdddt d|d|}tdddt d|d|}t |j |j rt j t tjdd |d |} WdQXntjdd |d |} t| ||qEWWdQXdS( Ntrecordtalwaysi Rt r_idx_typet c_idx_typeisdf + df2R0R2(Rtlhs_index_typest index_typestwarningstcatch_warningsRt simplefiltertRuntimeWarningRRRRRRRMRR( ReR0R2Rt lr_idx_typet rr_idx_typeRRtdf2R((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_basic_frame_alignmentLs   $c Cst|jdd}x|D]\}}tdddtd|d|}tjdd|d |}t||dktt|j d |j d |j }tjd d|d |}t|||kqWdS( Ntrepeatii RRRsdf < 2R0R2RRsdf < df3( RRRRRMRRR RRRR( ReR0R2RRRRRtdf3((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_frame_comparison^s  c CsMt|j|j|j|j}tjdttjdtx|D]\}}}}tdddt d|d|}tdddt d|d|} td ddt d|d|} t |j | j | j rt j t tjd d |d |} WdQXntjd d |d |} t| || | qKWWdQXdS( NRRiiRRRiisdf + df2 + df3R0R2(RRRRR RR!R"RRRRRRRMRR( ReR0R2Rtr1tc1tr2tc2RR%R(R((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt#test_medium_complex_frame_alignmentks!!! c sfd}t|j|jd}tjdt?tjdtx'|D]\}}}||||qTWWdQXdS(Nc stdddtd|d|}t||}ttjjd|d }t|j|jrt j t  t j ddd}WdQXnt j ddd}|d ks|d krd kr|j|n||}n ||}t||dS( Ni RRRisdf + sR0R2RR&(RRtgetattrR R?RRRRRRR"RMRtaddR(RRt index_nameRRRRRV(R0R2(sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttestits $( RRRR(RR(RRRRR RR!R"(ReR0R2R2RRRR1((R0R2sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt!test_basic_frame_series_alignments c sfd}tdddgdddgd }tjdt?tjdtx'|D]\}}}||||q`WWdQXtd gd gd }tjdt?tjdtx'|D]\}}}||||qWWdQXdS( Nc stdddtd|d|}t||}ttjjd|d }t|j|jrt j t  t j ddd }WdQXnt j ddd }|d ks|d krd kr|j|n||}n ||}t||dS( Ni iRRRiss + dfR0R2RR&(RRR/R R?RRRRRRR"RMRR0R(RRR1RRRRRV(R0R2(sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR2s $( RRRRRRtignoreR(RR(RR(RRR RR!R"(ReR0R2R2RRRR1((R0R2sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt!test_basic_series_frame_alignments$cCst|j|jdd}tjdttjdtxn|D]f\}}}}tdddt d |d |}t ||} t t j jd | d } d j|} d j|} t|j| jr>tjt tj| d|d|} WdQXtjt tj| d|d|}WdQXn6tj| d|d|} tj| d|d|}|dkrE|dkrE|dkrt| |qqEqEWWdQXdS(NRRRRRRi RRRiss {0} dfsdf {0} sR0R2RR&(RR(RR(RRRRR RR!R"RRR/R R?RRR.RRRRRMRR(ReR0R2RRRRR1RRRRARCRR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_series_frame_commutativitys*  !$ cCs\ddl}t|j|j|j|j}d}d}d|}tjdttjdtx|D]\}} } } |j ddg} |j d d g} t ||d t d |d | }t ||d t d | d | }t t j| | }ttjj||| }| dks=| dkrh|dkr[|j|}qr||}n ||}|dks| dkr|dkr|j|}q||}n ||}t|j|j|jrtjt tjdd|d|}WdQXntjdd|d|}|j|jksAtt||qmWWdQXdS(NiiiiRRRRRR%RRRRR&s df2 + s + dfR0R2(RRRRRR RR!R"tchoiceRRR/tlocalstgetR R?RR0RRRRRMRRRR(ReR0R2RRtntm1tm2R*R,R+R-R1tobj_nameRR%RRt expected2RVR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt#test_complex_series_frame_alignments@  !!      c Csttdd}ttd}|dkr<t}nt}t|tjdd|d|WdQXttd}tttjdd|d|WdQXttdd}ttd}tttjdd|d|WdQXttdd}ttd}|dk}|sAt}nt}t|}tjdd|d||st|d kst t |d j } d j d d t j|j|jd } | | kst nWdQXdS( Nii i'R&sdf + sR0R2RiisAlignment difference on axis {0} is larger than an order of magnitude on term {1!r}, by more than {2:.4g}; performance may sufferR(R RR RRRRMRtlenRR=tmessageR.R?tlog10RR( ReR0R2RRtseentis_python_enginetwrntwRRV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt+test_performance_warning_for_poor_alignments8       &(RRR(R(RRRRRR&R)R*R,RR.R3R5R6R?RG(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRAs      $ -tTestOperationsNumExprPandasc BseZedZedZdZdZdZdZe j j ddZ dZ d Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZe j jdddddgejgd&ge j j ddZ e j jdddd'gdZ!e j jdddddgejgd(idd6gd Z"d!Z#d"Z$d#Z%d$Z&d%Z'RS()cCs)d|_d|_tjtj|_dS(NR&RY(R0R2RkRRlRr(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\7s  cCs|`|`dS(N(R0R2(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR^=scOsD|j|d<|j|d<|jddd|d 2] > 2(R R?RRR*Rt NameErrorR(ReR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt&test_failing_subscript_with_name_errorscCs\ttjjdd}|jddi|d6}|d|dk}t||dS(Niis(df + 1)[df > 2]RRii(R R?RRRR(ReRRURV((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_lhs_expression_subscriptsc Csttjjdddtd}d}|j|jk}d}|j|j|j}d}|j|j|j|jdk}|||f}|||f} x@t|| D]/\} } t | |j | d i|d 6qWdS( NiiRtabcs df.a < df.bsdf.a + df.b + df.csdf.a + df.b + df.c[df.b < 0]iRR( R R?RRRRRRDtzipR R( ReRtexpr1texpec1texpr2texpec2texpr3texpec3RtexpecsRERL((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_attr_expressions'$cCsvttjjdddtd}ttjjdd}d}tjt|j|di|d6|d6dS( NiiRR\sdf = df2RRR%( R R?RRRR*RR<R(ReRR%R^((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_assignment_failss 'csttjjdddtdjtjtj dtjtj dtjtj dj}|d|d |d<jj d d t t |j}|d|d |d <jj d d t t |fd}|j}d|d |dtjt$tjdd|jd|jWdQXtjt$tjdd|jd|jWdQXtjt$tjdd|jd|jWdQXtjt$tjdd|jd|jWdQXtjt$tjdd|jd|jWdQXtjt$tjdd|jd|jWdQXdS(NRs 1 in [1, 2]R0R2s 2 in (1, 2)s 3 in (1, 2)s3 not in (1, 2)s[3] not in (1, 2)s[3] in ([3], 2)s[[3]] in [[[3]], 2]s(3,) in [(3,), 2]s(3,) not in [(3,), 2]s[(3,)] in [[(3,)], 2]s[(3,)] in (1, 2, [(3,)])s[3] not in (1, 2, [[3]])(R2RMRR0RR*RR(ReR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_simple_in_opss\                   %%%(ii(ii(ii((RRRR\R^RRNRORPR*R,tfilterwarningsRQRRRTRURXRZR[ReRfRnRoRqRsRtRvRxRzRR?RR~RRRRRRR(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRH4sD        2    >       '+   tTestOperationsNumExprPythoncBs_eZedZdZdZdZdZdZdZ dZ dZ RS( cCsTtt|jd|_d|_tjtj|_t d|j|_dS(NR&RcSs |dkS(NR#snot in(R#snot in((R7((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRus( RRR\R0R2RkRRlRrRv(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\s    cCsVd}djdd}d}tj|d|jd|j}||ksRtdS(Nis * Ri!R0R2(tjoinRMRR0R2R(ReRRkRVR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_check_many_exprss !c CsTttjjdd}tjttjddi|d6d|j d|j dS(Niisdf > 2 and df > 3RRR2R0( R R?RRR*RRRMRR2R0(ReR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_fails_and sc CsTttjjdd}tjttjddi|d6d|j d|j dS(Niisdf > 2 or df > 3RRR2R0( R R?RRR*RRRMRR2R0(ReR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyt test_fails_orsc CsTttjjdd}tjttjddi|d6d|j d|j dS(Niis not df > 2RRR2R0( R R?RRR*RRRMRR2R0(ReR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_fails_notsc CsZttjjdd}d}tjt$tj|d|j d|j WdQXdS(Niis(df + 2)[df > 1] > 0 & (df > 0)R2R0( R R?RRR*RRRMRR2R0(ReRR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_fails_ampersandsc CsZttjjdd}d}tjt$tj|d|j d|j WdQXdS(Niis(df + 2)[df > 1] > 0 | (df > 0)R2R0( R R?RRR*RRRMRR2R0(ReRR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_fails_pipe"scCsxttjddD]\}}}dj|||}|dkrltjt|j|WdQXq|j|}t|}||kstqWdS( NRRs {0} {1} {2}RR(RR(RR(RR( RRkRR.R*RRRR(ReRRARCRRR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRP(s  c CsxttjttfttfD]\}}}dj|}|dkrtjt$t j |d|j d|j WdQXq"t j |d|j d|j }t |}||ks"t q"WdS(Ns lhs {0} rhsRRR0R2(RR(RRkRRRR.R*RRRMRR0R2R(ReRRARCRRR((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRO4s (! ( RRRR\RRRRRRRPRO(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRs       tTestOperationsPythonPythoncBseZedZRS(cCsRtt|jd|_|_tjtj|_t d|j|_dS(NRcSs |dkS(NR#snot in(R#snot in((R7((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRuHs( RRR\R0R2RkRRlRrRv(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\Cs  (RRRR\(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRAstTestOperationsPythonPandascBseZedZRS(cCs<tt|jd|_d|_tjtj|_dS(NRRY( RRR\R0R2RkRRlRr(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\Ns  (RRRR\(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyRLstTestMathPythonPythoncBseZedZedZdZdZdZdZdZ dZ dZ d Z d Z d Zd ZRS( cCs(d|_d|_t|_t|_dS(NRRY(R0R2Rt unary_fnsRt binary_fns(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR\Ys   cCs|`|`dS(N(R0R2(R[((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR^`scOsD|j|d<|j|d<|jddd|d 0)R0R2Rs not (cat > 0)( RMR R|RRRR*RR(R0R2RRVRU((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_negate_lt_eq_lens- "t TestValidatecBseZdZRS(c CsYdddddgdg}x7|D]/}tjttjdd|WdQXq"WdS(NiRiig@s2+2Rh(R*RR<RMR(Retinvalid_valuestvalue((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyttest_validate_bool_argss (RRR(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyR}s(ztdistutils.versionRt itertoolsRRRtnumpyR?t numpy.randomRRRR*t pandas.compatRRt pandas.errorsRtpandas.util._test_decoratorstutilt_test_decoratorsRtpandas.core.dtypes.commonRR R RYRMR R R Rtpandas.core.computationRtpandas.core.computation.checkRtpandas.core.computation.enginesRRtpandas.core.computation.exprRNt computationRkRRt#pandas.core.computation.expressionsRRtpandas.core.computation.opsRRRRRRtpandas.util.testingttestingRRRRR R!RR"RRtfixtureR0RR2R4R9R;RFRJRKRQRWtsett differenceRqt skip_if_no_netobjectRXRRRRRR RRRHRRRRRRRRRRRt ExprVisitorR,RRRRRRRRRRRRRRR(((sAlib/python2.7/site-packages/pandas/tests/computation/test_eval.pyts     "..        a      I  t    !     *