a < b @sUddlmZddlmZddlmZddlZddlZddlZ ddl Z ddl m Z ddl mmZddlmZmZmZmZddlZddlmZmZmZddlmZddlmZdd l m!Z!m"Z"ddl#m$m%m&Z&dd l#m'Z'm(Z(m)Z)dd l*m+Z+m,Z,dd l-m.Z.m/Z/m0Z0m1Z1m2Z2e j3d de!DdddZ4e j3e&j5dddZ6ddZ7e j3e7dddZ8ddZ9ddZ:ddZ;ddZe?e@e.Ae/ZBe j3eCeDd$gd%d&d'd(ZEeEZFeEZGGd)d*d*ZHd+d,ZIGd-d.d.ZJd/d0ZKd1d2ZLGd3d4d4ZMGd5d6d6ZNGd7d8d8ZOe jPQd9ZRGd:d;d;ZSejTdd?d@dAZXejTdBdCZYejTdDdEZZe)ej[e(dFZ\dGe]dH<e jVWde!e jVWde\dIdJZ^dKdLZ_dMdNZ`e jVWdOgdPdQdRZadSdTZbdUdVZcdWdXZddYdZZee jVWd[d\e jVWd(efegfe jVWd]efegfd^d_Zhe jVWd`dadbgdcddZidedfZjdgdhZke jVWdidjdkgdldmZldndoZmGdpdqdqZndS)r) annotations)reduce)productN)PerformanceWarning)is_boolis_float is_list_like is_scalar) DataFrameSeries date_range)pytables)ENGINESNumExprClobberingError)BaseExprVisitorPandasExprVisitorPythonExprVisitor)NUMEXPR_INSTALLED USE_NUMEXPR)ARITH_OPS_SYMSSPECIAL_CASE_ARITH_OPS_SYMS_binary_math_ops_binary_ops_dict_unary_math_opsc csD|]<}tj|tjj|dkot dtdtdtjgdVqdS)numexprznumexpr enabled->z , installed->reason)ZmarksN)pytestparammarkZskipifrrtd skip_if_no_ne.0enginer%r/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/computation/test_eval.py 5s  r')paramscCs|jSNrrequestr%r%r&r$4sr$cCs|jSr)r*r+r%r%r&parserHsr-cCstr ttSgSr))rlistrr%r%r%r&_get_unary_fns_for_neMsr/cCs|jSr)r*r+r%r%r&unary_fns_for_neQsr0cCs t|jSr))rZ has_neg_fracr$r%r%r&engine_has_neg_fracVsr2c Csnt|}t|rdz |||WStyb}z.t|drLtjWYd}~SWYd}~n d}~00|||S)Nz6negative number cannot be raised to a fractional power)rr2 ValueErrorstr startswithnpnan)lhscmp1rhsr$cer%r%r&_eval_single_binZs r=cCs@t|tr t|tjr |jdkp>t|to>t|tjo>|jdkS)N) isinstancer r6ndarrayndimr8r:r%r%r&_series_and_2d_ndarrayhsrCcCs(t|trt|tp&t|to&t|tSr))r?r r rBr%r%r&_series_and_framensrDcCst|tot|tjjjSr))r?boolpdcoreZgenericZNDFramerBr%r%r&_bool_and_frametsrHcCst|ttjfot|Sr))r?complexr6Zcomplexfloatingisnan)resultexpectedr%r%r&_is_py3_complex_incompatxsrM)r r Z SeriesNaNZ DataFrameNaNfloat)r(idscCsjttjdd}tj||dk<ttjddttjdtddtjtjdg|tjf}||jS)N rN?r>)r r6randomrandr7randnr r)r,Znan_df1optsr%r%r&r8sr8c @sNeZdZejddddZeddZeddZ ej j d gd gd d ej j d ddgddgd ej de j ddZej de jddZej de jddZej d ddgej d ddgddZddZdd Zd!d"Zej d#ed$d%Zd&d'Zd(d)Zd*d+Zejd,d-Zd.d/Zd0d1Zd2d3Z d4d5Z!d6d7Z"d8d9Z#ej d:e$e%j&'d;d<e$e%j&j(d;d=d>e$e%j&)d;d<d?kgd@dAZ*ej d:e+e%j&'d;e+e%j&j(d;d;d>e+e%j&)d;d?kgdBdCZ,dDdEZ-dFdGZ.ej dHe%j/e%j0gdIdJZ1dKdLZ2dMdNZ3dOdPZ4dQdRZ5dSdTZ6dUS)VTestEvalTZautousecCs||_||_dSr)r$r-selfr$r-r%r%r&set_engine_parser_attrssz TestEval.set_engine_parser_attrscCsddl}||_dS)Nr)rne)clsr^r%r%r& setup_classszTestEval.setup_classcsfddtDS)Nc3s|]}|jkr|VqdSr)r1r"r\r%r&r'z+TestEval.current_engines..)rrar%rar&current_enginesszTestEval.current_enginesr9)z!=z==z<=z>=<>)r^eqlegeltgt)rPcmp2rerdrjribinopc Cs|jdkrt|dvrtd}tjt|d<d|d|d|d}tj||j|jd }Wdn1sf0YdSt||||j} t||||j} t| || |j} d|d|d|d}tj||j|jd }||| dS) Npythonandor"'BoolOp' nodes are not implementedmatch(lhs z rhs) z (lhs  rhs)rZ) r-rraisesNotImplementedErrorrFevalr$r= check_equal) r\r9rkrlr8r:msgexrKlhs_newrhs_newrLr%r%r&test_complex_cmp_opss2zTestEval.test_complex_cmp_opscmp_opcCsx|dk}|dk}|jdkrf|dvrfd}tjt|d||||Wdn1sX0YdS||||dS)Nrrminnot in&'(In|NotIn)' nodes are not implementedrr)r-rrvrwcheck_simple_cmp_op)r\rr8r:rzr%r%r&test_simple_cmp_opss,zTestEval.test_simple_cmp_opsopcCs|jdkrV|dvrVd}tjt|d||||Wdn1sH0YdSt|rt|s|dvr|jdkr|jdkrtjjdd}|j |||||dS)Nrmrrrrpandasz]Looks like expected is negative, unclear whether expected is incorrect or result is incorrectr) r-rrvrwcheck_compound_invert_oprr$rxfailnodeZ add_marker)r\rr8r:r,rzrr%r%r&test_compound_invert_ops(, z TestEval.test_compound_invert_opcCs||||||dSr))check_chained_cmp_op)r\r9rkr8midhsr:r%r%r&test_chained_cmp_opszTestEval.test_chained_cmp_opcCsZt|trt||n>t|tr0t||n&t|tjrJt||n ||ksVJdSr)) r?r tmassert_frame_equalr assert_series_equalr6r@assert_numpy_array_equalr\rKrLr%r%r&rys   zTestEval.check_equalcCs|jdkrfd|d|d}d}tjt|d$tj||j|jdWdn1sX0YdSt||||j}t||||j} |dur| durd|d|d}d|d|d} d |d |d } t|d | |j} || | fD]$} tj| |j|jd}t || qdS) Nrmlhs  mid  rhsrqrrrZ mid and mid rt mid) & (mid ru&) r-rrvrwrFrxr$r=rassert_almost_equal)r\r8r9midrkr:ex1rzr|r}ex2ex3rLr{rKr%r%r&rs  2zTestEval.check_chained_cmp_opcCsd|d}dddg}|dvrvt|svtjt|d,tj||j|j||dd Wdq1sj0Yn0t ||||j}tj||j|jd }| ||dS) Nrr|zxonly list-like( or dict-like)? objects are allowed to be passed to (DataFrame\.)?isin\(\), you passed a (\[|')bool(\]|')z'argument of type 'bool' is not iterablerrrrBr$r- local_dictrZ) joinrrrv TypeErrorrFrxr$r-r=ry)r\r8r9r:r{rzrLrKr%r%r&r s" &zTestEval.check_simple_cmp_oparith1cCsd|d}tj||j|jd}t||||j}t||d|d|d}tj||j|jd}t||||j}|||||dS)NrrrZz rhs )rFrxr$r-r=rrcheck_alignment)r\rr8r:r{rKrLnlhsr%r%r&test_binary_arith_ops"s  zTestEval.test_binary_arith_opsc Cszz||\}}Wntttfy*YnL0|jdkrZ|jd|d}t|j |nt d|d}t||dS)Nrznlhs z ghs) Zalignr3rAttributeErrorr$r^evaluaterrvaluesrx)r\rKrZghsrrLr%r%r&r.s zTestEval.check_alignmentcCsd}tj||j|jd}||}t|||jdkrr|jd}t|t t fr`t|j |qt|| nt |d||j}t||dS)Nz lhs % rhsrZrzexpected % rhs%)rFrxr$r-rrr^rr?r r ritemr=)r\r8r:r{rKrLr%r%r& test_modulusDs   zTestEval.test_moduluscCsd}|jdkr8tj||j|jd}||}|||nNd}tjt|d,tj|||d|j|jdWdn1s|0YdS)Nz lhs // rhsrmrZzGunsupported operand type\(s\) for //: 'VariableNode' and 'VariableNode'rrrBrr$r-)r$rFrxr-ryrrvr)r\r8r:r{resrLrzr%r%r&test_floor_divisionTs zTestEval.test_floor_divisioncCsd}t|d||j}tj||j|jd}t|rt|rt||rd}tjt |dt ||Wdq1sv0YnPt ||d}tj||j|jd}t|d||j}t|d||j}t ||dS)Nz lhs ** rhs**rZz-(DataFrame.columns|numpy array) are differentrrz(lhs ** rhs) ** rhs) r=r$rFrxr-r rMrrvAssertionErrorrrr)r\r8r:r{rLrKrzZmiddler%r%r&test_powhs$, zTestEval.test_powc Csz|t}Wn"ty0tt|g}Yn0|}tjd|j|jd}t |||j D]}t |tjd||jdq^dS)Nz~elbrZ) astyperErr6arrayrFrxr$r-rrrc)r\r8ZelbrLrKr$r%r%r&check_single_invert_ops   zTestEval.check_single_invert_opc Csddg}d|d}dddg}t|r~||vr~tjt|d,tj||j|j||d d Wdn1sr0Ynt|rt|rt d d ||f\}}t ||||j}t|r| }n|}tj||j|jd }t |||j D]$} tj||j|jd } t | |qdS)Nrrz~(lhs rurzyonly list-like( or dict-like)? objects are allowed to be passed to (DataFrame\.)?isin\(\), you passed a (\[|')float(\]|')z(argument of type 'float' is not iterablerrrBrcSs t|gSr))r6rxr%r%r&rbz3TestEval.check_compound_invert_op..rZ)rr rrvrrFrxr$r-mapr=rrrc) r\r8r9r:Z skip_theser{rzrLrKr$Zevr%r%r&rs4 &  z!TestEval.check_compound_invert_opcCs6d}ttjdd}|jdkrhd}tjt|d$tj ||j|j d}Wdq1s\0YnFd}tjt |d$tj ||j|j d}Wdn1s0Yttjj dd d }|jdkrd }tjt|d$tj ||j|j d}Wdn1s 0Yn&|}tj ||j|j d}t ||ttjddd k}|}tj ||j|j d}t ||tgd tjdd kd}|jdkrtjtdd$tj ||j|j d}Wdn1s0YnHd}tjt |d$tj ||j|j d}Wdn1s(0YdS)N~lhsrNrSr-couldn't find matching opcode for 'invert_dd'rrrZ0ufunc 'invert' not supported for the input typesrNrSsize)couldn't find matching opcode for 'invertrRar>g@)br;unknown type object#bad operand type for unary ~: 'str')r r6rTrVr$rrvrwrFrxr-rrandintrrrUr3r\exprr8rzrKexpectr%r%r&test_frame_inverts8 42 6   6zTestEval.test_frame_invertcCs d}ttjd}|jdkrfd}tjt|d$tj ||j|j d}Wdq1sZ0YnFd}tjt |d$tj ||j|j d}Wdn1s0Yttjj ddd}|jdkrd }tjt|d$tj ||j|j d}Wdn1s 0Yn&|}tj ||j|j d}t ||ttjdd k}|}tj ||j|j d}t ||tgd }|jdkrtjtd d$tj ||j|j d}Wdn1s0YnHd }tjt |d$tj ||j|j d}Wdn1s0YdS)NrrNrrrrrZrrrrRrrr)r r6rTrVr$rrvrwrFrxr-rrrrrUr3rr%r%r&test_series_inverts8 42 6    6zTestEval.test_series_invertcCsd}ttjdd}| }tj||j|jd}t ||ttjj ddd}| }tj||j|jd}t ||ttj dddk}|jdkrd }t j t|d $tj||j|jd}Wdn1s0Yn&| }tj||j|jd}t ||dS) N-lhsrNrSrZrrrRr*couldn't find matching opcode for 'neg_bb'rr)r r6rTrVrFrxr$r-rrrrUrrvrwr\rr8rrKrzr%r%r&test_frame_negates"   4zTestEval.test_frame_negatecCsd}ttjd}| }tj||j|jd}t ||ttjj ddd}| }tj||j|jd}t ||ttj ddk}|jdkrd}t j t|d$tj||j|jd}Wdn1s0Yn&| }tj||j|jd}t ||dS) NrrNrZrrRrrrr)r r6rTrVrFrxr$r-rrrrUrrvrwrr%r%r&test_series_negate)s"   4zTestEval.test_series_negater8rNrSrrrRcCs,d}|}tj||j|jd}t||dSNz+lhsrZ)rFrxr$r-rrr\r8rrrKr%r%r&test_frame_posCs zTestEval.test_frame_poscCs,d}|}tj||j|jd}t||dSr)rFrxr$r-rrrr%r%r&test_series_posUs zTestEval.test_series_poscCsd}tjt|d$tjd|j|jdWdn1s<0Ytjd|j|jddksbJtjd|j|jdd ks~Jtjd |j|jdd ksJtjd |j|jdd ksJtjd|j|jddksJtjd|j|jdd ksJtjd|j|jdd ks Jtjd|j|jdd ks*Jtjd|j|jddksHJtjd|j|jddksfJtjd|j|jddksJdS)Nz%bad operand type for unary ~: 'float'rrz~1.0rZz-1.0r-r$gz+1.0?z~1z-1z+1r>z~Truez~Falsez-Truez-Falserz+Truez+False)rrvrrFrxr$r-)r\rzr%r%r&test_scalar_unarygs2zTestEval.test_scalar_unarycCs:tjtdtjd}tjgdtjd}t||dS)NzK[-True, True, ~True, +True,-False, False, ~False, +False,-37, 37, ~37, +37]dtype) rTrr>rFrri%ir)r6rrFrxZobject_rrrr%r%r&test_unary_in_arrayxszTestEval.test_unary_in_arrayrcCs\tdtjdg|di}|d}|jtdgks8J|d}|jtdgksXJdS)Nrrrzx < -0.1Fz-5 > x)r r6rrxr)r\rdfrr%r%r&test_float_comparison_bin_ops   z%TestEval.test_float_comparison_bin_opc Csd}|d7}|d7}|d7}|d7}tjddd}}}ttjdd}|D]J}d }tjt|d $tj||j |j d WdqR1s0YqRdS) N)z1 or 2z1 and 2)za and bza or b)z1 or 2 and (3 + 2) > 3)z2 * x > 2 or 1 and 2)z2 * df > 3 and 1 or arr>rS;cannot evaluate scalar only bool ops|'BoolOp' nodes are notrrrZ) r6rTrVr rrvrwrFrxr$r-)r\exprsrrrrr{rzr%r%r&test_disallow_scalar_bool_opssz&TestEval.test_disallow_scalar_bool_opscCsfd}tjd|j|jd}|dks$Jt|s0Jd}tjd|j|jd}|dksTJt|s`Jd}tjd|j|jd}|rJt|sJt|sJtdg}tjd|j|jd}t |tdg|j dksJtdg}tjd|j|jd}t |tdg|j dksJtdg}tjd|j|jd}t |tdg|j dksbJdS)Nr>rrZg?F)r>) rFrxr$r-r rr6rrrshape)r\rrKr%r%r&test_identicals2         zTestEval.test_identicalcCs(d}tj||j|jd}|dks$JdS)Nz1 + 2 * 5 - 1 + 2 rZ )rFrxr$r-)r\exprKr%r%r&test_line_continuationszTestEval.test_line_continuationcCsd}tj||j|jd}t|}||ks.Jtdgdi}d}|d|d}|js^Jd}|d |d}|j d d gddf}t ||d }|d |d}|j d gddf}t ||dS)Nz1000000000.006rZA)g~eA $eAg'1eAgeAzA < z.4fg eAzA > r>rSrA == ) rFrxr$r-r6float64r queryemptylocrr)r\rrKrLrcutoffexactr%r%r&test_float_truncations     zTestEval.test_float_truncationcCstgdggdd}d}tjt|d|dWdn1sH0Yt}d|j_tjt|d|dWdn1s0YdS) N)rrr)Zfoobarclasscolumnsz4Python keyword not valid identifier in numexpr queryrrz class == 0lambdaz lambda == 0)r rrv SyntaxErrorrindexnamer\rrzr%r%r&test_disallow_python_keywordss(z&TestEval.test_disallow_python_keywordsN)7__name__ __module__ __qualname__rfixturer] classmethodr`propertyrcr parametrizer BOOL_OPS_SYMSr~ CMP_OPS_SYMSrrrryrr_good_arith_opsrrrrr skip_if_windowsrrrrrrrr r6rTrVrrUrr rrrfloat32rrrrrrrr%r%r%r&rXs|           %+/   !rXcOs tjSr))r6rTrV)argskwargsr%r%r&rrbrc@s<eZdZejdgdejdejejgddZ dS)TestTypeCastingr)+-*r/dtcCstjddt|d}d|d}tj|||d}|jj|ks>J|jj|ksNJt|t|d|d}tj|||d}|jj|ksJ|jj|ksJt|t|dS) NrNr) data_gen_frdf z 3rZz3  df)rmakeCustomDataframefrFrxrrr)r\r$r-rrrsrr%r%r&test_binop_typecastings  z&TestTypeCasting.test_binop_typecastingN) rrrrrrr6r rrr%r%r%r&r sr cCst|jjtjSr)) issubclassrtyper6Z datetime64rr%r%r& _is_datetimesrcGs0tttd| }ttjtt|}|o.|S)NZ is_monotonic)anyroperator attrgetterrxorr)r Znot_monoZ only_one_dtr%r%r& should_warnsr c @seZdZgdZedgZddZejdeejdeejdedd Z ejd eejded d Z ejd eejdeejdeejdeddZ ejdddgejdeejd eddZ ejdddgejde egdgddgejdddZejdeejd eejdddgejddd gd!d"Zejd eejdeejdeejded#d$Zd%d&Zd'S)( TestAlignment)iurrcCs8d}tjddtd}tj|||d}t||ddS)Nzdf * ~2rNr)rrZ)rrrrFrxr)r\r$r-rrrr%r%r&test_align_nested_unary_op)sz(TestAlignment.test_align_nested_unary_op lr_idx_type rr_idx_type c_idx_typec Cstjddtdttjddt||d}tjddt||d}t|j|jrt t t j d||d}Wdq1s0Ynt j d||d}t |||Wdn1s0YdS) NTrecordalwaysrQr r_idx_typer(zdf + df2rZ warningscatch_warnings simplefilterRuntimeWarningrrrr rassert_produces_warningrFrxr) r\r$r-r&r'r(rdf2rr%r%r&test_basic_frame_alignment/s    0z(TestAlignment.test_basic_frame_alignmentr-cCsttjddt||d}tjd||d}t||dkttjj |j |j |j d}tjd||d}t|||kdS)NrQr,df < 2rZrS)rrzdf < df3) rrrrFrxrr r6rTrVrrr)r\r$r-r-r(rrdf3r%r%r&test_frame_comparisonFs z#TestAlignment.test_frame_comparisonr1c1r2c2c Cstjddtdttjddt||d}tjddt||d}tjddt||d} t|j|j| jrt t t j d ||d } Wdq1s0Ynt j d ||d } t | ||| Wdn1s0YdS) NTr)r+rrSr,rNzdf + df2 + df3rZr/) r\r$r-r:r;r<r=rr5r8rr%r%r&#test_medium_complex_frame_alignmentSs      0z1TestAlignment.test_medium_complex_frame_alignment index_namerrc Cstjddtdttjddt||d}t||}tt j d|dd}t |j |j rtt tjd||d} Wdq1s0Yntjd||d} |d ks|d kr|d kr||n||} n||} t| | Wdn1s0YdS) NTr)r+rQr,rNdf + srZrr)r0r1r2r3rrrgetattrr r6rTrVr rr4rFrxaddr r\r$r-r@r-r(rrrrrLr%r%r&!test_basic_frame_series_alignmentls    0z/TestAlignment.test_basic_frame_series_alignmentzr_idx_type, c_idx_type)r"r#r)rrzignore::RuntimeWarningc Cstjddt||d}t||}ttjd|dd}t|j |j rt t  t j d||d} Wdq1sv0Ynt j d||d} |dks|dkr|dkr||n||} n||} t| | dS) NrQr,rNzs + dfrZrr)rrrrBr r6rTrVr rr4r3rFrxrCrrDr%r%r&!test_basic_series_frame_alignments    0z/TestAlignment.test_basic_series_frame_alignmentrr rc Cs^tjdd:tdttjddt||d}t||}tt j d|dd} d|d} d |d } t |j | j rtt tj| ||d } Wdn1s0Ytt tj| ||d } Wdn1s0Yn tj| ||d } tj| ||d } |d kr:|d kr:|d kr:t| | Wdn1sP0YdS)NTr)r+rQr,rNzs rrz srZrr)r0r1r2r3rrrrBr r6rTrVr rr4rFrxr)r\r$r-r@rr-r(rrrr8r:rrr%r%r&test_series_frame_commutativitys&      . 0 z-TestAlignment.test_series_frame_commutativityc Csddl}d}d} d| } tjdd|tdt|dd g} |d d g} tj| |t||d } tj| |t||d }t t | | }t t j||d|}|d ks|d kr|dkr||}q||}n||}|d ks|d kr |dkr|| }n|| }n|| }t|j|j| jrftt tjd||d}Wdn1sZ0Yntjd||d}|j|jksJt||Wdn1s0YdS)NrrrNrSTr)r+rrrr5r,rrzdf2 + ser + dfrZ)rTr0r1r2r3choicerrrrBlocalsgetr r6rVrCr rr4rFrxrr)r\r$r-r:r;r<r=rTnm1m2r@Zobj_namerr5rZserZ expected2rLrr%r%r&#test_complex_series_frame_alignments@         2z1TestAlignment.test_complex_series_frame_alignmentc Csttjdd}ttjd}|dkr0t}nd}t| tj d||dWdn1sd0Yttjd}td tj d||dWdn1s0Yttjdd}ttjd}td tj d||dWdn1s 0Yttjdd}ttjd}|dk}|sLt}nd}t|x}tj d||d|st |d ksJt |d j } t |j|jd } d | d d } | | ksJWdn1s0YdS)NirQi'rFrArZrmr>rz_Alignment difference on axis 1 is larger than an order of magnitude on term 'df', by more than z.4gz; performance may suffer.)r r6rTrVr rrr4rFrxlenr4messagelog10rr) r\r$r-rrseenZis_python_engineZwrnwrzZloggedrLr%r%r&+test_performance_warning_for_poor_alignments> . . 0 z9TestAlignment.test_performance_warning_for_poor_alignmentN)rrrZ index_typesZlhs_index_typesr%rrrr6r9r?rEr.rfilterwarningsrGrHrOrUr%r%r%r&r!$sJ                   .r!c @s*eZdZejddddZddZddZej d dd gej d dd gej d e j d dZ ej d dd gej d dd gej d e j ddZ ddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zejjd%d&d'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Z ej d7d8d9d8d:ge!"gd;gej#d<d=d>Z$ej d7gd?d@dAZ%ej dBd8d9d8d:ge!"gd;d8d:igdCdDZ&dEdFZ'dGdHZ(dIdJZ)dKdLZ*dMdNZ+dOdPZ,ej dQgdRdSdTZ-ej dUdVdWgdXdYZ.dZS)[TestOperationsTrYcCs||_||_dSr)rZr[r%r%r&r]$sz&TestOperations.set_engine_parser_attrscOs8|j|d<|j|d<|ddd|d<tj|i|SNr$r-levelrr>r$r-poprFrxr\r r r%r%r&rx*s  zTestOperations.evalc sdg|jdkrddgfddtjtjD}dd|D}|D]}d|d }d |d }d|d }|d vrd }tjt|d$tj||j |jdWdn1s0YqBt d|d|j }|j||j |jd} | |ksJt | |d|j }|j|d| i|j |jd} | |ks$Jt d|| d|j }|j|d| i|j |jd} | |ksBJqBdS)Nrmrrcsg|]}|vr|qSr%r%r#rZ exclude_arithr%r& 5sz8TestOperations.test_simple_arith_ops..css|]}|dkr|VqdS)z//Nr%r]r%r%r&r';rbz7TestOperations.test_simple_arith_ops..z1 z 1zx z (x + 1)rz&argument of type 'int' is not iterablerrrZr>rr) r-rrrrrvrrFrxr$r=) r\Z arith_opsopsrr{rrrzexpecryr%r^r&test_simple_arith_ops0s8       4 z$TestOperations.test_simple_arith_opsr:Fr8rcCs|d|d|}|jdkrf|dvrfd}tjt|d||Wdn1sX0YdS||}t|}||ksJdSN rmrnrqrrr-rrvrwrxr\r:r8rr{rzrrr%r%r&test_simple_bool_opsWs( z#TestOperations.test_simple_bool_opscCs|d|d|}|jdkrf|dvrfd}tjt|d||Wdn1sX0YdS||}t|}||ksJdSrdrfrgr%r%r&test_bool_ops_with_constantsgs( z+TestOperations.test_bool_ops_with_constantscCsltjdddd}ttjd}d}tjt|d$|jd||d d Wdn1s^0YdS) Nrr>rNrQz?N-dimensional objects, where N > 2, are not supported with evalrrx + yrrbr)r6rTrVr rrvrwrx)r\rrbrzr%r%r&test_4d_ndarray_failsws z$TestOperations.test_4d_ndarray_failscCs|d}|dksJdS)N1r>)rx)r\rr%r%r& test_constant~s zTestOperations.test_constantcCs4ttjdd}|jdd|id}t||dS)NrQrSrrmr r6rTrVrxrr)r\rr5r%r%r&test_single_variablesz#TestOperations.test_single_variablecCstdg}d}tt|j|dd}Wdn1s>0Yt|tdgtt|j|dd}Wdn1s0Yt|tdgtt|jddd}Wdn1s0Yd}||ksJtt|jddd}Wdn1s 0Yd}||kszs / 1F)truedivrTz1 / 2rRzs / 2)r6rrr4 FutureWarningrxr)r\rr{rrar%r%r& test_truedivs0  , , ,  . . .zTestOperations.test_truedivcCsNttjdd}tjtdd|dWdn1s@0YdS)NrNrzname 'x' is not definedrrz df[x > 2] > 2)r r6rTrVrrv NameErrorrx)r\rr%r%r&&test_failing_subscript_with_name_errorsz5TestOperations.test_failing_subscript_with_name_errorcCsDttjdd}|jdd|id}|d|dk}t||dS)NrNrz(df + 1)[df > 2]rrmr>rSrqr\rrKrLr%r%r&test_lhs_expression_subscriptsz,TestOperations.test_lhs_expression_subscriptc Csttjddtdd}d}|j|jk}d}|j|j|j}d}|j|j|j|jdk}|||f}|||f} t|| D]"\} } t | |j | d |id q~dS) NrNrabcrz df.a < df.bzdf.a + df.b + df.czdf.a + df.b + df.c[df.b < 0]rrrm) r r6rTrVr.rrr;ziprrrx) r\rexpr1Zexpec1Zexpr2Zexpec2Zexpr3Zexpec3rZexpecsr<rar%r%r&test_attr_expressions   z#TestOperations.test_attr_expressioncCszttjddtdd}ttjdd}d}d}tjt|d$|j|||dd Wdn1sl0YdS) NrNrrzrzdf = df2%cannot assign without a target objectrr)rr5rm) r r6rTrVr.rrvr3rx)r\rr5r|rzr%r%r&test_assignment_failss z$TestOperations.test_assignment_failscs<ttjddtddtjtdd dWdn1sP0Yd}tjt|d d Wdn1s0Yd }tjt|d d Wdn1s0Y}|d |d |d <j dddt |}|d |d |d<j dddt |fdd}|}d|d |d <t |fdd}|j dddd}tjt|d dWdn1s0Y|j ddidd}|d |d |d<t |dS)NrNrSabrinvalid syntaxrrz d c = a + bz5left hand side of an assignment must be a single namez d,c = a + bzcannot assign to function callzTimestamp("20131001") = a + brr a = a + bTinplacer; c = a + bcs}d}|jddd|S)Nr>z a = 1 + bTr)copyrx)rr)orig_dfr%r&rsz0TestOperations.test_assignment_column..fr>csJd}j}jddd|j}tj|jdd|jdusFJdS)Nr>rTrFZ check_names)rrrxrrrr)rZold_arK)rr%r&rs   z#can only assign a single expressionz c = a = bzc = df.a + df.br)rtargetr) r r6rTrVr.rrrvrrxrr)r\rzrLrr%)rrr&test_assignment_columnsL(((     *z%TestOperations.test_assignment_columncCs:tdgdgd}|d}tdg}tj||dddS)N irrza in [11, -32]TFr)r rxr rrrxr%r%r&test_column_ins  zTestOperations.test_column_inz%Unknown: Omitted test_ in name prior.rcCs`ttjddtdd}|jddd}|dus4J|}|d|d |d <t||dS) NrNrSrrrFrrrr;) r r6rTrVr.rxrrr)r\ractualrLr%r%r&test_assignment_not_inplaces  z*TestOperations.test_assignment_not_inplacecCstgdgdd}|}|d|d|d<|d|d|d<|jdd d }t|||duslJ|dd |d<|dd |d <|jdd d }t|||dusJd}tjt|d|jddd Wdn1s0YdS)Nr>rSrr>rNrjrrrr;d$ c = a + b d = c + bTrr>rSr<$ a = a - 1 e = a + 2z@Multi-line expressions are only valid if all expressions containrrz( a = b + 2 b - 2F)r rrxrrrrvr3)r\rrLanswerrzr%r%r&test_multi_line_expression#s0    z)TestOperations.test_multi_line_expressioncCstgdgdd}|}|d|d|d<|d|d|d<|jdd d }t|||dd |d<|dd |d <|jdd d }t||dS)Nrrrrrr;rrFrr>rSr<rr rrxrrr\rrLr%r%r&&test_multi_line_expression_not_inplaceHs  z5TestOperations.test_multi_line_expression_not_inplacecCsltgdgdd}|}d}|d||d<|d||d<|jdd d }t|||dushJdS) NrrrrFrr;rz? c = a * @local_var d = c + @local_var Trr)r\rrLZ local_varrr%r%r&)test_multi_line_expression_local_variableas z8TestOperations.test_multi_line_expression_local_variablecCs|tgdgdd}dd}|}|d|dd|d <|d |dd|d <|jd d d }t|||dusxJdS)NrrrcSs|Sr)r%rr%r%r& local_funcwszUTestOperations.test_multi_line_expression_callable_local_variable..local_funcrr>rFr;rzM c = a * @local_func(1, 7) d = c + @local_func(1, 7) Trrr\rrrLrr%r%r&2test_multi_line_expression_callable_local_variabless zATestOperations.test_multi_line_expression_callable_local_variablecCstgdgdd}dd}|}|d|ddd |d <|d |ddd |d <|jd d d}t|||dus|JdS)NrrrcSs|Sr)r%rr%r%r&rszaTestOperations.test_multi_line_expression_callable_local_variable_with_kwargs..local_funcrrFr>)rrr;rzU c = a * @local_func(b=7, a=1) d = c + @local_func(b=7, a=1) Trrrr%r%r&>test_multi_line_expression_callable_local_variable_with_kwargss zMTestOperations.test_multi_line_expression_callable_local_variable_with_kwargscCsjtgdgdd}|}d}tjt|d|dWdn1sP0Yt||dS)Nrrrr~rrza = 1)r rrrvr3rrr)r\rZdf_origrzr%r%r&test_assignment_in_querys (z'TestOperations.test_assignment_in_querycCsttgdgdd}|}||ddk}|jdddt||i}dd i}|jd |dd t||dS) NrrrrrSza == 2Trr a = 1 + 2rr)r rrrrrxZassert_dict_equalrr%r%r&test_query_inplaces z!TestOperations.test_query_inplaceinvalid_targetr>catrSr>rzignore::FutureWarningcCsd}d}tjt|d |j||ddWdn1s<0Yt|drtjt|d |j||ddWdn1s0YdS)Nz)Cannot assign expression output to targetrrrTrrF)rrvr3rxhasattrr\rrz expressionr%r%r&test_cannot_item_assigns. z&TestOperations.test_cannot_item_assign)r>rrcCsJd}d}tjt|d |j||ddWdn1s<0YdS)Nz"Cannot return a copy of the targetrrrFr)rrvr3rxrr%r%r&test_cannot_copy_itemsz$TestOperations.test_cannot_copy_itemrcCsbd}|j||dddksJd}tjt|d |j||ddWdn1sT0YdS)N1 + 2Frrz0Cannot operate inplace if there is no assignmentrrT)rxrrvr3)r\rrrzr%r%r&test_inplace_no_assignments z)TestOperations.test_inplace_no_assignmentcCsRtjddtddd}|dk}|jdd|id}|dk}t||t||dS)NrSpr"rr(r-r7rrmrrrrxr)r\rr<rrr%r%r&*test_basic_period_index_boolean_expressions  z9TestOperations.test_basic_period_index_boolean_expressioncCsBtjddtddd}|jdd|id}||dk}t||dS) NrSrr"rzdf[df < 2 + 3]rrmrNrr\rrr<r%r%r&,test_basic_period_index_subscript_expressions z;TestOperations.test_basic_period_index_subscript_expressioncCsRtjddtddd}|jdd|id}|||dkdk|d}t||dS)NrSrr"rzdf[df[df < 2] < 2] + df * 2rrmrrr%r%r&-test_nested_period_index_subscript_expressionszz * Z!aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarZ)rrFrxr$r-)r\rrrLrr%r%r&test_check_many_exprs*s  z$TestOperations.test_check_many_exprsr)zdf > 2 and df > 3zdf > 2 or df > 3z not df > 2cCsttjdd}|jdkrvd}d|vr,d}tjt|d*tj |d|i|j|j d Wdq1sj0Yntj |d|i|j|j d dS) NrNrrmrqnot'Not' nodes are not implementedrrr)rr-r$ r r6rTrVr-rrvrwrFrxr$)r\rrrzr%r%r&test_fails_and_or_not1s$  &z$TestOperations.test_fails_and_or_notcharrrcCsttjdd}d|d}|jdkrpd}tjt|d$tj ||j|j dWdq1sd0Yntj ||j|j ddS) NrNrz(df + 2)[df > 1] > 0 z (df > 0)rmz$cannot evaluate scalar only bool opsrrrr)r\rrr{rzr%r%r&test_fails_ampersand_pipePs  4z(TestOperations.test_fails_ampersand_pipeN)/rrrrrr]rxrcrrrrrhrirnrprrrurwryr}rrrrrrrrrrrrr6rrVrrrrrrrrrrrr%r%r%r&rW#sf  '  " E  %     &   7 rWc@seZdZejddddZddZddZej d e d d Z d d Z ddZ ddZddZejddZddZddZdS)TestMathTrYcCs||_||_dSr)rZr[r%r%r&r]^sz TestMath.set_engine_parser_attrscOs8|j|d<|j|d<|ddd|d<tj|i|SrXrZr\r%r%r&rxds  z TestMath.evalcCstdtjdi}|j}|}|d}||}tjddtt||}Wdn1sb0Ytj ||dddS)NrrQz(a)ignoreallFr) r r6rTrVrrxerrstaterBrr)r\r0rrfnrgotrr%r%r&test_unary_functionsjs  ,zTestMath.test_unary_functionsrcCsttjdtjdd}|j}|j}|d}||}tjdd tt|||}Wdn1sp0Yt j ||dddS)NrQrz(a, b)rrFr) r r6rTrVrrrxrrBrr)r\rrrrrrrr%r%r&test_binary_functionsvs  .zTestMath.test_binary_functionscCsdttjdtjdd}|jd|j|jdd|j}tt |j |j }t j ||dddS)NrQrze = arctan2(sin(a), b)Tr$r-rFr)r r6rTrVrxr$r-r<Zarctan2sinrrrrr\rrrr%r%r&test_df_use_caseszTestMath.test_df_use_casecCs`ttjdtjdd}|jd|j|jdd|j}t|j |j }t j ||dddS)NrQrze = sin(a + b)TrFr) r r6rTrVrxr$r-r<rrrrrrr%r%r& test_df_arithmetic_subexpressions z)TestMath.test_df_arithmetic_subexpressioncCstdtjd|i}|jj|ks*J|jd|j|j dd|j }t |j}|j|jksbJ||jkspJt j ||dddS)NrrQz b = sin(a)TrFr)r r6rTrVrrrrxr$r-rrrr)r\rZ expect_dtyperrrr%r%r&check_result_types zTestMath.check_result_typecCsD|tjtj|tjtj|tjtj|tjtjdSr))rr6Zint32rZint64r rar%r%r&test_result_typesszTestMath.test_result_typescCs|tjtjdSr))rr6Z complex128rar%r%r&test_result_complex128szTestMath.test_result_complex128cCs^tdtjdi}d}tjt|d$|jd|j|j dWdn1sP0YdS)NrrQz#"mysin" is not a supported functionrrzmysin(a)rZ) r r6rTrVrrvr3rxr$r-rr%r%r&test_undefined_funcszTestMath.test_undefined_funccCs^tdtjdi}d}tjt|d$|jd|j|j dWdn1sP0YdS)NrrQz1Function "sin" does not support keyword argumentsrrzsin(x=a)rZ) r r6rTrVrrvrrxr$r-rr%r%r&test_keyword_argszTestMath.test_keyword_argN)rrrrrr]rxrrrrrrrrrr rrrrr%r%r%r&r]s       rrQc@s$eZdZddZddZddZdS) TestScopecCs$d}ttdtj|||ddS)Nz _var_s * 2rSrZ)rr_var_srFrx)r\r$r-r<r%r%r&test_global_scopeszTestScope.test_global_scopecCsDd}t}tjd|||dt}|d||ks@JdS)Nr>x + 1rlcls)rJrrFrxr[)r\r$r-rrZlcls2r%r%r&test_no_new_localss    zTestScope.test_no_new_localscCs8d}t}tjd||dt}||ks4JdS)Nr>rrZ)globalsrrFrx)r\r$r-rZgblsZgbls2r%r%r&test_no_new_globalss   zTestScope.test_no_new_globalsN)rrrrrrr%r%r%r&rsrcCsLd}tjt|d&tjddddddWdn1s>0YdS) NzInvalid engine 'asdf' passedrrrkr>rSrlasdf)rr$rrvKeyErrorrFrxrzr%r%r&test_invalid_enginesr) use_numexprrL))Tr)FrmcCsPddlm}td|$|d}||ks.JWdn1sB0YdS)Nr) _check_enginecompute.use_numexpr)Zpandas.core.computation.evalrrFoption_context)rrLrrKr%r%r&test_numexpr_option_respecteds rcCsztddZtgdgdd}|d}tddgddgdd dgd }t||Wdn1sl0YdS) NrF)TFTFNN)r>rSrr>rNrj)rBz A.isnull()rNrjr>)r)rFrr rrr)rrKrLr%r%r&#test_numexpr_option_incompatible_ops rcCsLd}tjt|d&tjddddddWdn1s>0YdS) NzInvalid parser 'asdf' passedrrrkr>rSrlr)rr-rrr%r%r&test_invalid_parsersr)rmr rz dict[str, type[BaseExprVisitor]]_parsersc Csbt|}|d||}|jD]B}d}tjt|dt||Wdq1sR0YqdS)Nrznodes are not implementedrr)rZunsupported_nodesrrvrwrB)r$r-Z VisitorClassinstr`rzr%r%r&test_disallowed_nodes s   rcCsFd}tjtdd tj|||dWdn1s80YdS)Nzs +rrrrZrrvrrFrx)r$r-r<r%r%r&test_syntax_error_exprssrcCsJd}d}tjt|d tj|||dWdn1s<0YdS)Nzs + tzname 's' is not definedrrrZ)rrvrvrFrx)r$r-r<rzr%r%r&test_name_error_exprssrexpress)za + @bz@a + bz@a + @bcCsd\}}|dkrPtjtdd tj|||dWdq1sD0Yn>tjtdd tj|||dWdn1s0YdS)Nr>rSrzThe '@' prefix is onlyrrrZzThe '@' prefix is notr)r$r-rrrr%r%r&%test_invalid_local_variable_reference#s 0rcCsxd\}}|dkrTd}tjt|d tjd||dWdqt1sH0Yn tjd||d}|||kstJdS)NrrzVariables in expression .+rrzsin + dotted_linerZ)rrvrrFrx)r$r-rZ dotted_linerzrr%r%r&test_numexpr_builtin_raises/s0rcCsHd}tjtdd"tjd|||dWdn1s:0YdS)N)*g@zResolver of type .+rrr) resolversr$r-)rrvrrFrx)r$r-Zcannot_resolver%r%r&test_bad_resolver_raises:srcCsBtjtdd tjd||dWdn1s40YdS)Nzexpr cannot be an empty stringrrrZrrvr3rFrxrZr%r%r&test_empty_string_raises@srcCsBtjtdd tjd||dWdn1s40YdS)Nz#only a single expression is allowedrrz 1 + 1; 2 + 2rZrrZr%r%r&$test_more_than_one_expression_raisesFsrcmprnr:c Cstddttjji}||}||}||}d|d|d}d|d|d}d|d|d } ||| fD]F} d } tjt| d  tj | ||d Wdqr1s0YqrdS) NcSs tjdS)NrQ)r6rTrr%r%r%r&rOrbz0test_bool_ops_fails_on_scalars..rrrrrtrrurrrrZ) intrOr6rTrVrrvrwrFrx) r8rr:r$r-genrrrrr{rzr%r%r&test_bool_ops_fails_on_scalarsKs   rotherz'x'z...cCsJtdgdi}|d|}tgddd}tr:d|_t||dS)Nr)rrr;r)FFF)r)r rxr rrrr)rrrKrLr%r%r&test_equals_various^s rcCs*d}tj}tj|||d}||ks&JdS)Nzinf + 1rZ)r6infrFrx)r$r-rrLrKr%r%r&test_infpsrcCsd}tt"}tjd||ddWdn1s60Yt|dksPJ|t|djvsfJtt"}tjd||ddWdn1s0Yt|dksJ|t|djvsJdS)Nz0The `truediv` parameter in pd.eval is deprecatedz1+1T)r$r-rsr>rF)rr4rtrFrxrPr4rQ)r$r-rsmr%r%r&test_truediv_deprecatedws 0 0rcolumnu Temp(°C)uCapacitance(μF)cCsTttjdd|dgd}|||dk}d|d}|j||d}t||dS)NrNrSrr`z` > 5r1)r r6rTrVrrr)r$rrrLZ query_stringrKr%r%r&test_query_tokens  rcCstddgddggddgd}||jdk}|jd||d }t|||d krd }tjt|d  |jd ||d Wdq1s0Yn|jd ||d }t||dS)NrrQr>r.rcountrz ~(cat > 0)rZrmrrrz not (cat > 0))r rrrrrrvrw)r$r-rrLrKrzr%r%r&test_negate_lt_eq_les 0rc@s.eZdZejdddgddgddZdS) TestValidatevaluer>Truerg@cCsDd}tjt|dtjd|dWdn1s60YdS)Nz8For argument "inplace" expected type bool, received typerrz2+2rr)r\rrzr%r%r&test_validate_bool_argssz$TestValidate.test_validate_bool_argsN)rrrrrrr r%r%r%r&rsr)o __future__r functoolsr itertoolsrrr0Znumpyr6rZ pandas.errorsrZpandas.util._test_decoratorsutilZ_test_decoratorsr Zpandas.core.dtypes.commonrrrr rrFr r r Zpandas._testingZ_testingrZpandas.core.computationr Zpandas.core.computation.enginesrrZpandas.core.computation.exprrGZ computationrrrrZ#pandas.core.computation.expressionsrrZpandas.core.computation.opsrrrrrrr$ZPARSERSr-r/r0r2r=rCrDrHrMsortedset differencerr.ranger8r:rrXrr rr r!rWrrTrVrrr!rrrrrrZPyTablesExprVisitorr__annotations__rrrrrrrrrrOrrrrrrrr%r%r%r&s              h>`