U
|
ý°dõ ã @ sv U d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZd dl Z
|
d dl Z d dl m Z mZmZ d dlm mZ d dlmZmZmZmZ d dlZd dlmZm Z m Z d dl m Z d dl!m"Z"m#Z# d d l$m%Z% d d
|
l&m'Z'm(Z(m)Z) d d l*m+Z+m,Z, d d l-m.Z.m/Z/m0Z0m1Z1m2Z2 d d l3m4Z4 e j5dd e%D ddd Z6e j5e"j7ddd Z8dd Z9e j5e:e;dddddd gd d d Z<e<Z=e<Z>G d d! d!Z?d"d# Z@G d$d% d%ZAd&d' ZBG d(d) d)ZCG d*d+ d+ZDG d,d- d-ZEe
|
j Fd.¡ZGG d/d0 d0ZHejId1d2 ZJejIe jK Ld3d4¡d5d6 ZMejId7d8 ZNejId9d: ZOe)e#jPe(d;ZQd<eRd=< e jK Lde%¡e jK LdeQ¡d>d? ZSd@dA ZTdBdC ZUe jK LdDdEdFdGg¡dHdI ZVdJdK ZWdLdM ZXdNdO ZYdPdQ ZZe jK LdRdS¡e jK Ld e[e\f¡e jK LdTe[e\f¡dUdV Z]e jK LdWdXdYg¡dZd[ Z^d\d] Z_e jK Ld^d_d`g¡dadb Z`dcdd Zae jK Ld^e4 b¡ ¡dedf Zcdgdh ZdG didj djZedS )ké )Ú annotations)Úreduce)ÚproductN)ÚNumExprClobberingErrorÚPerformanceWarningÚUndefinedVariableError)Úis_boolÚis_floatÚ is_list_likeÚ is_scalar)Ú DataFrameÚSeriesÚ
|
date_range)ÚexprÚpytables)ÚENGINES)ÚBaseExprVisitorÚPandasExprVisitorÚPythonExprVisitor)ÚNUMEXPR_INSTALLEDÚ USE_NUMEXPR)ÚARITH_OPS_SYMSÚSPECIAL_CASE_ARITH_OPS_SYMSÚ_binary_math_opsÚ_binary_ops_dictÚ_unary_math_ops)ÚDEFAULT_GLOBALSc c sD | ]<}t j|t jj|d ko t dt dt dtjgdV qdS )Únumexprznumexpr enabled->z , installed->©Úreason©ZmarksN)ÚpytestÚparamÚmarkÚskipifr r ÚtdÚ skip_if_no_ne)Ú.0Úengine© r) úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/computation/test_eval.pyÚ <genexpr>: s õ þúþr+ )Úparamsc C s | j S ©N©r" ©Úrequestr) r) r* r( 9 s r( c C s | j S r- r. r/ r) r) r* ÚparserM s r1 c
|
C sj t | }t| jr`z || |W S tk
|
r^ } z"t| d¡rLtj W Y ¢S W 5 d }~X Y nX || |S )Nz6negative number cannot be raised to a fractional power)r r Z has_neg_fracÚ
|
ValueErrorÚstrÚ
|
startswithÚnpÚnan)ÚlhsÚcmp1Úrhsr( ÚcÚer) r) r* Ú_eval_single_binR s
|
ÿr< é r r Z SeriesNaNZ DataFrameNaNÚfloat)r, Úidsc C sj t tj dd¡}tj||dk< t tj dd¡ttj d¡tddtjtjdg|tj ¡ f}|| j S )Né
|
r= ç à?é é )r r5 ÚrandomÚrandr6 Úrandnr r" )r0 Znan_df1Úoptsr) r) r* r7 a s ûr7 c
|
@ sP e Zd Zejjdddddddgdd d
|
d d d gdejjdddgd d gdej dej¡dd Zej dej ¡dd Z
|
ej dej ¡dd Z ej dddg¡ej dddg¡dd Z ej de ee e¡¡d d Zd d Zd d! Zejd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Z ej d.e e j d/d0¡e e j j!d/d1d2e e j "d/d0¡d3kg¡d4d5 Z#ej d.e$e j d/¡e$e j j!d/d/d2e$e j "d/¡d3kg¡d6d7 Z%d8d9 Z&d:d; Z'ej d<e j(e j)g¡ej d=d>d?g¡d@dA Z*dBdC Z+ej dDdE¡dFdG Z,dHdI Z-dJdK Z.dLdM Z/dNdO Z0dPdQ Z1dRdS Z2dTS )UÚTestEvalr8 z!=z==z<=z>=ú<ú>ÚneÚeqÚleÚgeÚltÚgt)r? Úcmp2Úbinopc C s¼ |dkrZ|dkrZd}t jt|d. d| d| d| d} tj| ||d W 5 Q R X d S t||||}
|
t||||} t|
|
|| |} d| d| d| d} tj| ||d } t | | ¡ d S )
|
NÚpython©ÚandÚorú"'BoolOp' nodes are not implemented©Úmatchú(lhs z rhs) z (lhs ú rhs)©r( r1 )r! ÚraisesÚNotImplementedErrorÚpdÚevalr< ÚtmÚ assert_equal)Úselfr8 rQ rR r7 r9 r( r1 ÚmsgÚexÚlhs_newÚrhs_newÚexpectedÚresultr) r) r* Útest_complex_cmp_opsx s z TestEval.test_complex_cmp_opsÚcmp_opc
|
C sê |dk }|dk }|dkr^|dkr^d}t jt|d" d| d}tj|||d W 5 Q R X d S d| d}d d
|
d g¡}|dkr¼t|s¼t jt|d tj|||||d d W 5 Q R X n*t||||}tj|||d} t
|
| |¡ d S )Nr rS ©Úinúnot inú&'(In|NotIn)' nodes are not implementedrX úlhs ú rhsr\ ú|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 iterable©r7 r9 ©r( r1 Ú
|
local_dict) r! r] r^ r_ r` Újoinr
|
Ú TypeErrorr< ra rb )
|
rc rk r7 r9 r( r1 rd re rh ri r) r) r* Útest_simple_cmp_ops s2 üÿüz TestEval.test_simple_cmp_opsÚopc C sf |dkrN|dkrNd}t jt|d" d| d}tj|||d W 5 Q R X d S t|rt|s|dkr|dkr|dkrt jjd d
|
} |j | ¡ d d g}
|
d| d}d
|
ddg¡}t |rö||
|
kröt jt |d tj|||||dd W 5 Q R X nlt |r t |r t dd ||f\}}t||||} t | r@| } n| } tj|||d} t | | ¡ d S )NrS rl ro rX z~(lhs r[ r\ Úpandasz]Looks like expected is negative, unclear whether expected is incorrect or result is incorrectr rm rn rr zyonly 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 iterablers rt c S s t | g¡S r- )r5 Úarray©Úxr) r) r* Ú<lambda>Ý ó z2TestEval.test_compound_invert_op.<locals>.<lambda>)r! r] r^ r_ r` r r# ÚxfailÚnodeÚ
|
add_markerrv r rw Úmapr< ra Úassert_almost_equal) rc ry r7 r9 r0 r( r1 rd re r# Z
|
skip_theserh ri r) r) r* Útest_compound_invert_op² sV ÿþýüûÿ üÿü
|
z TestEval.test_compound_invert_opc C sð |}|dkrPd| d| d} d}
|
t jt|
|
d tj| ||d W 5 Q R X d S t||||} t||||} | d k rì| d k rìd| d| d} d| d| d} d | d
|
| d }t| d | |}| | |fD ] }tj|||d}t ||¡ qÊd S ) NrS rp ú mid rq rW rX r\ ú mid and mid rZ ú mid) & (mid r[ ú&)r! r] r^ r_ r` r< ra r )rc r8 rQ r7 Úmidhsr9 r( r1 ÚmidÚex1rd rf rg Úex2Úex3rh re ri r) r) r* Útest_chained_cmp_opæ s" z TestEval.test_chained_cmp_opÚarith1c
|
C sè d| d}t j|||d}t||||}t ||¡ d| d| d}t j|||d}t||||} z| |¡\} }
|
W n tttfk
|
r Y d S X |dkrÈdd l } |
|
d| d¡}t |j |¡ n td| d}t ||¡ d S ) Nrp rq r\ z rhs r r znlhs z ghs) r_ r` r< ra r Zalignr2 rw ÚAttributeErrorr ÚevaluateÚvalues) rc r r7 r9 r( r1 re ri rh ZnlhsZghsrK r) r) r* Útest_binary_arith_opsÿ s" z TestEval.test_binary_arith_opsc C s d}t j|||d}|| }t ||¡ |dkrrdd l}| d¡}t|ttfr`t |j |¡ qt ||
|
¡ ¡ nt |d||}t ||¡ d S )Nz lhs % rhsr\ r r zexpected % rhsú%) r_ r` ra r r r Ú
|
isinstancer r r Úitemr< ) rc r7 r9 r( r1 re ri rh rK r) r) r* Ú test_modulus" s
|
zTestEval.test_modulusc C sl d}|dkr2t j|||d}|| }t ||¡ n6d}tjt|d t j|||d||d W 5 Q R X d S )Nz
|
lhs // rhsrS r\ zGunsupported operand type\(s\) for //: 'VariableNode' and 'VariableNode'rX rs ©ru r( r1 )r_ r` ra rb r! r] rw ) rc r7 r9 r( r1 re Úresrh rd r) r) r* Útest_floor_division4 s ÿüz TestEval.test_floor_divisionc
|
C sÄ d}t |d||}tj|||d}t|rxt|rxt|ttjfrxt |¡rxd}t j
|
t |d t ||¡ W 5 Q R X nHt ||¡ d}tj|||d}t |d||} t | d||}t ||¡ d S )Nz
|
lhs ** rhsú**r\ z-(DataFrame.columns|numpy array) are differentrX z(lhs ** rhs) ** rhs)r< r_ r` r r Úcomplexr5 ZcomplexfloatingÚisnanr! r] ÚAssertionErrorra Úassert_numpy_array_equalr )
|
rc r7 r9 r( r1 re rh ri rd Zmiddler) r) r* Útest_powH s( ÿþýü zTestEval.test_powc C sZ z| t¡}W n$ tk
|
r2 t t|g¡}Y nX | }tjd||d}t ||¡ d S )Nz~elbr\ ) ÚastypeÚboolr r5 r{ r_ r` ra r )rc r7 r( r1 Zelbrh ri r) r) r* Útest_check_single_invert_opb s z$TestEval.test_check_single_invert_opc C sª d}t tj dd¡}|dkrNd}tjt|d tj|||d W 5 Q R X n.d}tjt |d tj|||d W 5 Q R X t tjj
|
dd d
|
}|dkrÈd }tjt|d tj|||d W 5 Q R X n"| }tj|||d}t ||¡ t tj dd¡d k}| }tj|||d}t ||¡ t d ddgtj d¡d kd}|dkrxtjtdd tj|||d W 5 Q R X n.d}tjt |d tj|||d W 5 Q R X d S )Nú~lhsr= rC r ú-couldn't find matching opcode for 'invert_dd'rX r\ ú0ufunc 'invert' not supported for the input types©r= rC ©Úsizeú)couldn't find matching opcode for 'invertrA ÚarB ç @é )Úbr: úunknown type objectú#bad operand type for unary ~: 'str')r r5 rD rF r! r] r^ r_ r` rw Úrandintra Úassert_frame_equalrE r2 )rc r( r1 r r7 rd Úexpectri r) r) r* Útest_frame_invertl s8
|
zTestEval.test_frame_invertc C s d}t tj d¡}|dkrLd}tjt|d tj|||d}W 5 Q R X n.d}tjt |d tj|||d W 5 Q R X t tjj
|
ddd}|dkrÆd }tjt|d tj|||d W 5 Q R X n"| }tj|||d}t ||¡ t tj d¡d
|
k}| }tj|||d}t ||¡ t d d d g}|dkrbtjtdd tj|||d W 5 Q R X n.d}tjt |d tj|||d W 5 Q R X d S )Nr¥ r= r r¦ rX r\ r§ r© r« rA r¬ rB r r° r± )r r5 rD rF r! r] r^ r_ r` rw r² ra Úassert_series_equalrE r2 )rc r( r1 r r7 rd ri r´ r) r) r* Útest_series_invert s8
|
zTestEval.test_series_invertc C sâ d}t tj dd¡}| }tj|||d}t ||¡ t tjjddd}| }tj|||d}t ||¡ t tj dd¡dk}|dkr¼d }t
|
j t |d
|
tj|||d W 5 Q R X n"| }tj|||d}t ||¡ d S ) Nú-lhsr= rC r\ r¨ r© rA r ú*couldn't find matching opcode for 'neg_bb'rX ) r r5 rD rF r_ r` ra r³ r² rE r! r] r^ ©rc r( r1 r r7 r´ ri rd r) r) r* Útest_frame_negateÆ s" zTestEval.test_frame_negatec C sÞ d}t tj d¡}| }tj|||d}t ||¡ t tjjddd}| }tj|||d}t ||¡ t tj d¡dk}|dkr¸d}t
|
j t |d tj|||d W 5 Q R X n"| }tj|||d}t ||¡ d S ) Nr¸ r= r\ r© rA r r¹ rX ) r r5 rD rF r_ r` ra r¶ r² rE r! r] r^ rº r) r) r* Útest_series_negateà s" zTestEval.test_series_negater7 r= rC r¨ r© rA c C s( d}|}t j|||d}t ||¡ d S ©Nz+lhsr\ )r_ r` ra r³ ©rc r7 r( r1 r r´ ri r) r) r* Útest_frame_posú s zTestEval.test_frame_posc C s( d}|}t j|||d}t ||¡ d S r½ )r_ r` ra r¶ r¾ r) r) r* Útest_series_pos s zTestEval.test_series_posc C s@ d}t jt|d tjd||d W 5 Q R X tjd||ddksFttjd||dd ks^ttjd
|
||dd ksvttjd ||dd ksttjd||ddks¦ttjd||dd ks¾ttjd||dd ksÖttjd||dd ksîttjd||ddksttjd||ddks"ttjd||ddks<td S )Nz%bad operand type for unary ~: 'float'rX z~1.0r\ z-1.0©r1 r( g ð¿z+1.0g ð?z~1éþÿÿÿz-1éÿÿÿÿz+1rB z~Truez~Falsez-Truez-Falser z+Truez+False)r! r] rw r_ r` r )rc r( r1 rd r) r) r* Útest_scalar_unary s zTestEval.test_scalar_unaryc C sN t jt d¡t jd}t jddddddddd d
|
d d
|
g t jd}t ||¡ d S ) NzK[-True, True, ~True, +True,-False, False, ~False, +False,-37, 37, ~37, +37]©Údtyperà Tr rB r FiÛÿÿÿé% iÚÿÿÿ)r5 r{ r_ r` Zobject_ra r )rc ri rh r) r) r* Útest_unary_in_array/ s. ÿúôñz TestEval.test_unary_in_arrayrÆ r zx < -0.1z-5 > xc C s< t dtjdg|di}| |¡}|jt dg¡ks8td S )Nr} r rÅ F)r r5 r{ r` r r )rc rÆ r Údfr r) r) r* Ú test_float_comparison_bin_opN s
|
z%TestEval.test_float_comparison_bin_opc C sl t dddtjgi}| d¡}|j d¡}tj||t d | d¡}|jj ddd}tj||t d d S ) Nr} r rB z x.fillna(-1)rà ©Z check_nameszx.shift(1, fill_value=-1))Z
|
fill_value)
|
r r5 r6 r` r} Zfillnara r¶ r Úshift©rc rÉ ri rh r) r) r* Útest_unary_in_functionV s
|
|
z TestEval.test_unary_in_functionre )z1 or 2z1 and 2za and bza or bz1 or 2 and (3 + 2) > 3z2 * x > 2 or 1 and 2z2 * df > 3 and 1 or ac C s\ t j d¡dd }}}tt j dd¡}d}tjt|d tj|||d W 5 Q R X d S )Nr® rB rC ú;cannot evaluate scalar only bool ops|'BoolOp' nodes are notrX r\ ) r5 rD rF r r! r] r^ r_ r` ) rc re r( r1 r} r¬ r¯ rÉ rd r) r) r* Ú test_disallow_scalar_bool_opsd s
|
z&TestEval.test_disallow_scalar_bool_opsc C sN d}t jd||d}|dks tt|s,td}t jd||d}|dksLtt|sXtd}t jd||d}|rttt|stt|stt dg¡}t jd||d}t |t dg¡¡ |j dksÊtt dg¡}t jd||d}t |t dg¡¡ |j dks
|
tt dg¡}t jd||d}t |t dg¡¡ |j dksJtd S )NrB r} r\ g ø?F)rB )
|
r_ r` r r r r5 r{ ra r Úshape)rc r( r1 r} ri r) r) r* Útest_identicalx s2 zTestEval.test_identicalc C s$ d}t j|||d}|dks td S )Nz1 + 2 * 5 - 1 + 2 r\ é ©r_ r` r )rc r( r1 Úexpri r) r) r* Útest_line_continuation s z TestEval.test_line_continuationc C sÎ d}t j|||d}t |¡}||ks*ttddddgi}d}| d|d ¡}|js\td
|
}| d |d ¡}|jd d gd d
f }t
|
||¡ d}| d|d ¡}|jd gd d
f }t
|
||¡ d S )Nz1000000000.006r\ ÚAg~ eÍÍAg $ eÍÍAg'1 eÍÍAg© eÍÍAzA < z.4fgÅ eÍÍAzA > rB rC úA == ) r_ r` r5 Úfloat64r r ÚqueryÚemptyÚlocra r³ ) rc r( r1 rÕ ri rh rÉ ÚcutoffÚexactr) r) r* Útest_float_truncation s
|
|
z TestEval.test_float_truncationc C sx t dddggdddgd}d}tjt|d | d¡ W 5 Q R X t }d |j_tjt|d | d
|
¡ W 5 Q R X d S ) Nr ZfooÚbarÚclass©Úcolumnsz4Python keyword not valid identifier in numexpr queryrX z
|
class == 0Úlambdaz lambda == 0)r r! r] Ú SyntaxErrorrÚ ÚindexÚname©rc rÉ rd r) r) r* Ú test_disallow_python_keywords¶ s z&TestEval.test_disallow_python_keywordsc C s: t d¡dkstt d¡dks$tt d¡dks6td S )Nznot Truer z not Falserà zTrue and not Truer rÔ )rc r) r) r* Útest_true_false_logic s z TestEval.test_true_false_logicc C sR t ddi}t |j d¡j ¡s&tt |j d¡joD|j d¡j ¡sNtd S )Nr¬ Zhello)r r_ r` r3 rY r¬ r )rc Úeventr) r) r* Útest_and_logic_string_matchÈ s z$TestEval.test_and_logic_string_matchN)3Ú__name__Ú
|
__module__Ú __qualname__r! r# Ú parametrizer Ú BOOL_OPS_SYMSrj Ú CMP_OPS_SYMSrx r
r ÚsortedÚsetr Ú
|
differencer r r r r% Úskip_if_windowsr¡ r¤ rµ r· r» r¼ r r5 rD rF r² rE r¿ r rÀ rÄ rÈ Úfloat32rÙ rÊ rÎ rÐ rÒ rÖ rß ré rê rì r) r) r) r* rH w s| ý
|
"
|
3 ÿ
|
|
|
+/úþ
|
úþ
|
þ
|
! rH c O s
|
t j ¡ S r- )r5 rD rF )ÚargsÚkwargsr) r) r* r~ Ï r r~ c @ sT e Zd Zej ddddddg¡ej dejejg¡ej dd d
|
g¡d d Z d S )ÚTestTypeCastingry ú+ú-Ú*r ú/ÚdtÚ
|
left_right)rÉ Ú3)r rÉ c C sr t jddt|d}|\}}| d| d| } tj| ||d}
|
|jj|ksNt|
|
jj|ks^tt |
|
t| ¡ d S )Nr= r® )Ú
|
data_gen_frÆ ú r\ ) ra ÚmakeCustomDataframeÚfr_ r` r rÆ r r³ ) rc r( r1 ry rÿ r rÉ ÚleftÚrightÚsr r) r) r* Útest_binop_typecasting× s z&TestTypeCasting.test_binop_typecastingN)
|
rí rî rï r! r# rð r5 r÷ rÙ r r) r) r) r* rú Ö s rú c G s4 t tt d¡| }ttjtdd | }|o2|S )NZis_monotonic_increasingc S s t | jjtjS r- )Ú
|
issubclassrÆ Útyper5 Z
|
datetime64r| r) r) r* r~ í r z should_warn.<locals>.<lambda>)Úanyr ÚoperatorÚ
|
attrgetterr Úxor)rø Znot_monoZ only_one_dtr) r) r* Ú should_warnê s ÿr c @ sÐ e Zd ZdddgZedg Zdd Zej de¡ej de¡ej de¡d d
|
Z ej d e¡ej de¡d d Z
|
ej de¡ej de¡ej de¡ej de¡dd Z ej dddg¡ej de¡ej d e¡dd Z ej dddg¡ej de eddgddgdg ¡ej d¡d d Zej de¡ej d e¡ej dddg¡ej d d d g¡d!d" Zej de¡ej de¡ej de¡ej de¡d#d$ Zd%d& Zd'S )(Ú TestAlignmentÚir rÿ c C s8 d}t jddtd}tj|||d}t ||d ¡ d S )Nzdf * ~2r= r® )r r\ éýÿÿÿ)ra r r r_ r` r³ )rc r( r1 r rÉ r r) r) r* Útest_align_nested_unary_opö s z(TestAlignment.test_align_nested_unary_opÚ lr_idx_typeÚ rr_idx_typeÚ
|
c_idx_typec
|
C s¦ t jdd t dt¡ tjddt||d}tjddt||d}t|j|jrxt t¡ t
|
j d||d}W 5 Q R X nt
|
j d||d}t ||| ¡ W 5 Q R X d S ) NT©ÚrecordÚalwaysr@ ©r Ú
|
r_idx_typer é zdf + df2r\ © ÚwarningsÚcatch_warningsÚ simplefilterÚRuntimeWarningra r r r ræ Úassert_produces_warningr_ r` r³ ) rc r( r1 r r r rÉ Údf2r r) r) r* Útest_basic_frame_alignmentü s* ÿ ÿ z(TestAlignment.test_basic_frame_alignmentr c C st t jddt||d}tjd||d}t ||dk ¡ ttjj |j
|
|j |j d}tjd||d}t |||k ¡ d S )Nr@ r údf < 2r\ rC )ræ rã zdf < df3) ra r r r_ r` r³ r r5 rD rF rÑ ræ rã )rc r( r1 r r rÉ r Údf3r) r) r* Útest_frame_comparison s ÿ z#TestAlignment.test_frame_comparisonÚr1Úc1Úr2Úc2c
|
C s t jdd¬ t dt¡ tjddt||d}tjddt||d}tjddt||d} t|j|j| jrt t¡ t
|
j d ||d
|
}
|
W 5 Q R X nt
|
j d ||d
|
}
|
t |
|
|| | ¡ W 5 Q R X d S ) NTr r r® rC r é r= zdf + df2 + df3r\ r ) rc r( r1 r) r* r+ r, rÉ r$ r' r r) r) r* Ú#test_medium_complex_frame_alignment s8 ÿ ÿ ÿ z1TestAlignment.test_medium_complex_frame_alignmentÚ
|
index_nameræ rã c
|
C sæ t jddÐ t dt¡ tjddt||d}t||}tt j
|
d¡|d d
}t |j |j rt t¡ tjd||d} W 5 Q R X ntjd||d} |d ks¨|d krÄ|d
|
krº| |¡n|| }
|
n|| }
|
t | |
|
¡ W 5 Q R X d S ) NTr r r@ r r= údf + sr\ rÿ r )r r r! r" ra r r Úgetattrr r5 rD rF r ræ r# r_ r` Úaddr³ ) rc r( r1 r/ r r rÉ ræ r r rh r) r) r* Ú!test_basic_frame_series_alignment8 s& ÿ
|
z/TestAlignment.test_basic_frame_series_alignmentzr_idx_type, c_idx_type)rÿ rÿ zignore::RuntimeWarningc
|
C s( |dkrb|dkrb|dkrb|dkrb|dkrbd| d| d| d | d
|
|
|
}|j tjj|d d ¡ tjd dt||d}t||} t t
|
j d¡| d d
}
|
t |
|
j|jrÐt t¡ tjd||d} W 5 Q R X ntjd||d} |dksò|dkr|dkr| |
|
¡n|
|
| } n|
|
| } t | | ¡ d S )Nr )rz rS ræ r r z"Flaky column ordering when engine=z , parser=z , index_name=z , r_idx_type=z , c_idx_type=F)r Ústrictr@ é r r= zs + dfr\ rÿ )r r r! r# r ra r r r1 r r5 rD rF r ræ r# r" r_ r` r2 r³ ) rc r0 r( r1 r/ r r r rÉ ræ r r rh r) r) r* Ú!test_basic_series_frame_alignmentR s<
|
ÿþýüû ÿ ÿ
|
z/TestAlignment.test_basic_series_frame_alignmentry rû rý c
|
C s t jdd t dt¡ tjddt||d}t||}tt j
|
d¡|d d
} d| d}
|
d | d
|
} t |j | j rÈt t¡ tj|
|
||d } W 5 Q R X t t¡ tj| ||d } W 5 Q R X n tj|
|
||d } tj| ||d } |d kr|d kr|d krt | | ¡ W 5 Q R X d S )NTr r r@ r r= zs z dfzdf z sr\ rÿ r )r r r! r" ra r r r1 r r5 rD rF r ræ r# r_ r` r³ )rc r( r1 r/ ry r r rÉ ræ r r7 r9 r¬ r¯ r) r) r* Ú test_series_frame_commutativityy s. ÿ
|
|
z-TestAlignment.test_series_frame_commutativityc
|
C s d}d}d| } t jddZ t dt¡ t ddg¡}
|
t d d
|
g¡} tj||t||d } tj| |t||d } t t
|
| ¡|
|
}t t j |¡|d |
}|d ks¬|d krÊ|d krÀ| |¡}qÒ| | }n| | }|d ksä|d kr|d krø| | ¡}n|| }n|| }t| j|j| jrFt t¡ tjd||d}W 5 Q R X ntjd||d}|j|jkshtt ||¡ W 5 Q R X d S )Nr® r= rC Tr r ræ rã rÉ r$ r rÿ r zdf2 + ser + dfr\ )r r r! r" rD Úchoicera r r r1 ÚlocalsÚgetr r5 rF r2 r ræ r# r_ r` rÑ r r³ )rc r( r1 r) r* r+ r, ÚnÚm1Úm2r/ Zobj_namerÉ r$ ræ ÚserZ expected2rh r r) r) r* Ú#test_complex_series_frame_alignment sN ÿ ÿ
|
|
z1TestAlignment.test_complex_series_frame_alignmentc C s t tj dd¡}ttj d¡}|dkr0t}nd}t |¡ tj d||d W 5 Q R X ttj d¡}t d¡ tj d||d W 5 Q R X t tj dd¡}ttj d¡}t d¡ tj d||d W 5 Q R X t tj dd¡}ttj d¡}|dk}|st}nd}t |¡n}tj d||d |st
|
|d ksFt t |d
|
j } t |j|jd ¡}
|
d |
|
d d } | | kst W 5 Q R X d S )Niè r@ i' r Fr0 r\ rS rB r z_Alignment difference on axis 1 is larger than an order of magnitude on term 'df', by more than z.4gz; performance may suffer.)r r5 rD rF r r ra r# r_ r` Úlenr r3 ÚmessageÚlog10rª rÑ ) rc r( r1 rÉ r ÚseenZis_python_engineZwrnÚwrd Zloggedrh r) r) r* Ú+test_performance_warning_for_poor_alignmentÈ s: ÿz9TestAlignment.test_performance_warning_for_poor_alignmentN)rí rî rï Z index_typesZlhs_index_typesr r! r# rð r% r( r. r3 Úlistr Úfilterwarningsr6 r7 r? rE r) r) r) r* r ò sJ
|
|
þ
|
! ,r c
|
@ sF e Zd Zdd Zdd Zej dddg¡ej dddg¡ej d ej ¡d
|
d Z
|
ej dddg¡ej dddg¡ej d ej ¡d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Z ejj d0d1d2d3 Z d4d5 Z d6d7 Z d8d9 Z!d:d; Z"d<d= Z#d>d? Z$d@dA Z%ej dBdCdDdCdEge& 'g ¡dFg¡dGdH Z(ej dBdCdDdFg¡dIdJ Z)ej dKdCdDdCdEge& 'g ¡dFdCdEig¡dLdM Z*dNdO Z+dPdQ Z,dRdS Z-dTdU Z.dVdW Z/dXdY Z0ej dZd[d\d]g¡d^d_ Z1ej d`dadbg¡dcdd Z2deS )fÚTestOperationsc O s | dd¡d |d< tj||S ©NÚlevelr rB ©Úpopr_ r` ©rc rø rù r) r) r* r` ú s zTestOperations.evalc
|
s4 g |dkrddg fddt jt j D }dd |D }|D ]î}d| d }d
|
| d }d| d }|d kr d } tjt| d tj|||d W 5 Q R X q@td|d|}
|
| j|||d} | |
|
ksÊt t| |d|}
|
| j|d| i||d} | |
|
ksút td|| d |}
|
| j|d| i||d} | |
|
ks@t q@d S )NrS rm rn c s g | ]}| kr|qS r) r) ©r' ry ©Z exclude_arithr) r* Ú
|
<listcomp> s þz8TestOperations.test_simple_arith_ops.<locals>.<listcomp>c s s | ]}|d kr|V qdS )z//Nr) rN r) r) r* r+ s z7TestOperations.test_simple_arith_ops.<locals>.<genexpr>z1 z 1zx z (x + 1)rl z&argument of type 'int' is not iterablerX r\ rB r} r )
|
r r rò r! r] rw r_ r` r< r ) rc r( r1 Z arith_opsÚopsry re r r rd Úexpecr} Úyr) rO r* Útest_simple_arith_opsþ s0
|
|
þ z$TestOperations.test_simple_arith_opsr9 TFr7 ry c C sr | d| d| }t dkrP|dkrPd}tjt|d | |¡ W 5 Q R X d S | |¡}t|}||ksntd S ©Nr rS rT rW rX ©r1 r! r] r^ r` r ©rc r9 r7 ry re rd r rÕ r) r) r* Útest_simple_bool_ops! s
|
z#TestOperations.test_simple_bool_opsc C sr | d| d| }t dkrP|dkrPd}tjt|d | |¡ W 5 Q R X d S | |¡}t|}||ksntd S rU rV rW r) r) r* Ú test_bool_ops_with_constants1 s
|
z+TestOperations.test_bool_ops_with_constantsc C sX t j dddd¡}tt j d¡}d}tjt|d | jd||d d
|
W 5 Q R X d S ) Nr® r- r= é r@ z?N-dimensional objects, where N > 2, are not supported with evalrX úx + y©r} rS ©ru )r5 rD rF r r! r] r^ r` )rc r} rS rd r) r) r* Útest_4d_ndarray_failsA s
|
z$TestOperations.test_4d_ndarray_failsc C s | d¡}|dkstd S )NÚ1rB )r` r )rc r} r) r) r* Ú test_constantH s
|
z TestOperations.test_constantc C s4 t tj dd¡}| jdd|id}t ||¡ d S )Nr@ rC rÉ r] ©r r5 rD rF r` ra r³ )rc rÉ r$ r) r) r* Útest_single_variableL s z#TestOperations.test_single_variablec C s: t tj dd¡}tjtdd | d¡ W 5 Q R X d S )Nr= r® úname 'x' is not definedrX z df[x > 2] > 2)r r5 rD rF r! r] Ú NameErrorr` ©rc rÉ r) r) r* Ú&test_failing_subscript_with_name_errorQ s z5TestOperations.test_failing_subscript_with_name_errorc C sD t tj dd¡}| jdd|id}|d |dk }t ||¡ d S )Nr= r® z(df + 1)[df > 2]rÉ r] rB rC ra rÍ r) r) r* Ú test_lhs_expression_subscriptV s z,TestOperations.test_lhs_expression_subscriptc C s¦ t tj dd¡tdd}d}|j|jk }d}|j|j |j }d}|j|j |j|jdk }|||f}|||f} t|| D ]"\}
|
} t
|
| | j |
|
d |id
|
¡ q~d S ) Nr= r® Úabcrâ z df.a < df.bzdf.a + df.b + df.cz df.a + df.b + df.c[df.b < 0]r rÉ r] ) r r5 rD rF rF r¬ r¯ r: Úzipra r¶ r` ) rc rÉ Úexpr1Zexpec1Zexpr2Zexpec2Zexpr3Zexpec3ÚexprsZexpecsr; rR r) r) r* Útest_attr_expression\ s
|
|
z#TestOperations.test_attr_expressionc C sf t tj dd¡tdd}t tj dd¡}d}d}tjt|d | j|||dd W 5 Q R X d S )
|
Nr= r® rh râ zdf = df2ú%cannot assign without a target objectrX )rÉ r$ r] ) r r5 rD rF rF r! r] r2 r` )rc rÉ r$ rj rd r) r) r* Útest_assignment_failsi s z$TestOperations.test_assignment_failsc C sB t tj dd¡tdd}tjtdd | d¡ W 5 Q R X d S )Nr= rC Úabrâ úinvalid syntaxrX z d c = a + b© r r5 rD rF rF r! r] rå r` re r) r) r* Ú%test_assignment_column_multiple_raiseq s z4TestOperations.test_assignment_column_multiple_raisec C sF t tj dd¡tdd}d}tjt|d | d¡ W 5 Q R X d S )Nr= rC ro râ z5left hand side of an assignment must be a single namerX z d,c = a + brq rè r) r) r* Ú%test_assignment_column_invalid_assignw s z4TestOperations.test_assignment_column_invalid_assignc C sF t tj dd¡tdd}d}tjt|d | d¡ W 5 Q R X d S )Nr= rC ro râ z cannot assign to function callrX z Timestamp("20131001") = a + brq rè r) r) r* Ú3test_assignment_column_invalid_assign_function_call~ s zBTestOperations.test_assignment_column_invalid_assign_function_callc C sT t tj dd¡tdd}| ¡ }|d |d |d< |jddd t ||¡ d S )
|
Nr= rC ro râ r¬ r¯ ú a = a + bT©Úinplace© r r5 rD rF rF Úcopyr` ra r³ ©rc rÉ rh r) r) r* Ú&test_assignment_single_assign_existing s
|
z5TestOperations.test_assignment_single_assign_existingc C sT t tj dd¡tdd}| ¡ }|d |d |d< |jdd d
|
t ||¡ d S ) Nr= rC ro râ r¬ r¯ r: ú c = a + bTrv rx rz r) r) r* Ú!test_assignment_single_assign_new s
|
z0TestOperations.test_assignment_single_assign_newc C s\ t tj dd¡tdd}| ¡ }d}|jddd | ¡ }d|d |d
|
< t ||¡ d S ) Nr= rC ro râ rB z a = 1 + bTrv r¯ r¬ rx )rc rÉ r¬ rh r) r) r* Ú+test_assignment_single_assign_local_overlap s z:TestOperations.test_assignment_single_assign_local_overlapc C sd t tj dd¡tdd}d}|j ¡ }|jddd ||j }t j
|
||jd d
|
|j d ks`t d S ) Nr= rC ro râ rB ru Trv FrË ) r r5 rD rF rF r¬ ry r` r¯ ra r¶ rç r )rc rÉ r¬ Zold_ari r) r) r* Ú"test_assignment_single_assign_name s
|
|
z1TestOperations.test_assignment_single_assign_namec C sT t tj dd¡tdd}|jddd d}tjt|d | d
|
¡ W 5 Q R X d S ) Nr= rC ro râ r| Trv z#can only assign a single expressionrX z c = a = b) r r5 rD rF rF r` r! r] rå rè r) r) r* Ú test_assignment_multiple_raises¨ s
|
z.TestOperations.test_assignment_multiple_raisesc C s\ t tj dd¡tdd}| jdd|i|dd | ¡ }|d |d
|
|d < t ||¡ d S ) Nr= rC ro râ zc = df.a + df.brÉ T)ru Útargetrw r¬ r¯ r: ) r r5 rD rF rF r` ry ra r³ rz r) r) r* Útest_assignment_explicit° s
|
z'TestOperations.test_assignment_explicitc C s: t dgdgd}| d¡}tdg}tj||dd d S )Né iàÿÿÿ©r¬ r¯ za in [11, -32]TFrË )r r` r ra r¶ rÍ r) r) r* Útest_column_in¸ s
|
|
z TestOperations.test_column_inz%Unknown: Omitted test_ in name prior.r c C s` t tj dd¡tdd}|jddd}|d k s4t| ¡ }|d |d |d
|
< t ||¡ d S ) Nr= rC ro râ r| Frv r¬ r¯ r: )
|
r r5 rD rF rF r` r ry ra r³ )rc rÉ Úactualrh r) r) r* Útest_assignment_not_inplaceÁ s z*TestOperations.test_assignment_not_inplacec C sæ t dddgdddgd}| ¡ }|d |d |d
|
< |d
|
|d |d < |jd d d}t ||¡ |d kspt|d d |d< |d d |d< |jdd d}t ||¡ |d ks¶td}tjt|d |jddd W 5 Q R X d S )NrB rC r® r- r= rZ r r¬ r¯ r: Údú$
|
c = a + b
|
d = c + bTrv r; ú$
|
a = a - 1
|
e = a + 2z@Multi-line expressions are only valid if all expressions containrX z(
|
a = b + 2
|
b - 2F) r ry r` ra r³ r r! r] r2 )rc rÉ rh Úanswerrd r) r) r* Útest_multi_line_expressionÍ s0 ü ü üz)TestOperations.test_multi_line_expressionc C s¢ t dddgdddgd}| ¡ }|d |d |d
|
< |d
|
|d |d < |jd d d}t ||¡ |d d |d< |d d |d< |jdd d}t ||¡ d S )NrB rC r® r- r= rZ r r¬ r¯ r: r r Frv r; r )r ry r` ra r³ rz r) r) r* Ú&test_multi_line_expression_not_inplaceò s ü üz5TestOperations.test_multi_line_expression_not_inplacec C sp t dddgdddgd}| ¡ }d}|d | |d
|
< |d
|
| |d < |jd d d}t ||¡ |d ksltd S )NrB rC r® r- r= rZ r r5 r¬ r: r z?
|
c = a * @local_var
|
d = c + @local_var
|
Trv ©r ry r` ra r³ r )rc rÉ rh Z local_varr r) r) r* Ú)test_multi_line_expression_local_variable s û z8TestOperations.test_multi_line_expression_local_variablec C s t dddgdddgd}dd }| ¡ }|d
|
|dd |d < |d |dd |d < |jddd}t ||¡ |d ks|td S )NrB rC r® r- r= rZ r c S s |S r- r) r r) r) r* Ú
|
local_func! s zUTestOperations.test_multi_line_expression_callable_local_variable.<locals>.local_funcr¬ r5 r: r zM
|
c = a * @local_func(1, 7)
|
d = c + @local_func(1, 7)
|
Trv r ©rc rÉ r rh r r) r) r* Ú2test_multi_line_expression_callable_local_variable s û zATestOperations.test_multi_line_expression_callable_local_variablec C s t dddgdddgd}dd }| ¡ }|d
|
|d dd |d < |d |d dd |d< |jddd}t ||¡ |d kstd S )NrB rC r® r- r= rZ r c S s |S r- r) r r) r) r* r 5 s zaTestOperations.test_multi_line_expression_callable_local_variable_with_kwargs.<locals>.local_funcr¬ r5 )r¯ r¬ r: r zU
|
c = a * @local_func(b=7, a=1)
|
d = c + @local_func(b=7, a=1)
|
Trv r r r) r) r* Ú>test_multi_line_expression_callable_local_variable_with_kwargs1 s û zMTestOperations.test_multi_line_expression_callable_local_variable_with_kwargsc C sZ t dddgdddgd}| ¡ }d}tjt|d | d
|
¡ W 5 Q R X t ||¡ d S ) NrB rC r® r- r= rZ r rm rX za = 1)r ry r! r] r2 rÚ ra r³ )rc rÉ Zdf_origrd r) r) r* Útest_assignment_in_queryE s z'TestOperations.test_assignment_in_queryc C sx t dddgdddgd}| ¡ }||d dk }|jd d
|
d t ||¡ i }ddi}| jd |d
|
d t ||¡ d S )NrB rC r® r- r= rZ r r¬ za == 2Trv ú a = 1 + 2©r rw )r ry rÚ ra r³ r` Zassert_dict_equalrz r) r) r* Útest_query_inplaceN s z!TestOperations.test_query_inplaceÚinvalid_targetrB ÚcatrC )rB r® c C sj d}d}t jt|d | j||dd W 5 Q R X t|drft jt|d | j||dd W 5 Q R X d S )Nz)Cannot assign expression output to targetr rX Tr ry F)r! r] r2 r` Úhasattr©rc r rd Ú
|
expressionr) r) r* Útest_cannot_item_assign\ s
|
z&TestOperations.test_cannot_item_assignc C s6 d}d}t jt|d | j||dd W 5 Q R X d S )Nz"Cannot return a copy of the targetr rX Fr )r! r] r2 r` r r) r) r* Útest_cannot_copy_itemh s z$TestOperations.test_cannot_copy_itemr c C sN d}| j ||dddks td}tjt|d | j ||dd W 5 Q R X d S )Nú1 + 2Fr r® z0Cannot operate inplace if there is no assignmentrX T)r` r r! r] r2 )rc r r rd r) r) r* Útest_inplace_no_assignmentp s
|
z)TestOperations.test_inplace_no_assignmentc C sR t jddtddd}|dk }| jdd|id}|dk }t ||¡ t ||¡ d S )NrC Úpr ©r r r r& rÉ r] ©ra r r r` r³ )rc rÉ r; Úrr} r) r) r* Ú*test_basic_period_index_boolean_expressionz s z9TestOperations.test_basic_period_index_boolean_expressionc C sB t jddtddd}| jdd|id}||dk }t ||¡ d S ) NrC r¡ r r¢ zdf[df < 2 + 3]rÉ r] r= r£ ©rc rÉ r¤ r; r) r) r* Ú,test_basic_period_index_subscript_expression s z;TestOperations.test_basic_period_index_subscript_expressionc C sR t jddtddd}| jdd|id}|||dk dk |d }t ||¡ d S )NrC r¡ r r¢ zdf[df[df < 2] < 2] + df * 2rÉ r] r£ r¦ r) r) r* Ú-test_nested_period_index_subscript_expression s z<TestOperations.test_nested_period_index_subscript_expressionc C sV t tj dd¡}tddd|d< | jdd|i||d}|jd k }tj||d
|
d d S ) Nr= r® z1/1/2012)ZperiodsÚdates1zdf.dates1 < 20130101rÉ r Z20130101FrË ) r r5 rD rF r r` r© ra r¶ )rc r( r1 rÉ r rR r) r) r* Útest_date_boolean s ü
|
z TestOperations.test_date_booleanc C s |dkrüt jd||d}|s tt jd||d}|s8tt jd||d}|rPtt jd||d}|shtt jd||d}|stt jd||d}|stt jd ||d}|s°tt jd
|
||d}|sÈtt jd ||d}|ràtt jd ||d}|søtnd }tjt|d t jd||d W 5 Q R X tjt|d t jd||d W 5 Q R X tjt|d t jd||d W 5 Q R X tjt|d t jd||d W 5 Q R X d}tjt|d t jd||d W 5 Q R X tjt|d t jd||d W 5 Q R X d S )NrS z 1 in [1, 2]r\ z 2 in (1, 2)z 3 in (1, 2)z3 not in (1, 2)z[3] not in (1, 2)z[3] in ([3], 2)z[[3]] in [[[3]], 2]z(3,) in [(3,), 2]z(3,) not in [(3,), 2]z[(3,)] in [[(3,)], 2]z 'In' nodes are not implementedrX z[(3,)] in (1, 2, [(3,)])z!'NotIn' nodes are not implementedz[3] not in (1, 2, [[3]]))r_ r` r r! r] r^ )rc r( r1 r rd r) r) r* Útest_simple_in_ops sF z!TestOperations.test_simple_in_opsc C s2 d}d d¡}d}tj|||d}||ks.td S )NrB z * Z!aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar\ )rv r_ r` r )rc r( r1 r¬ r rh r r) r) r* Útest_check_many_exprsË s
|
|
z$TestOperations.test_check_many_exprsr zdf > 2 and df > 3zdf > 2 or df > 3z
|
not df > 2c C sv t tj dd¡}|dkr\d}d|kr*d}tjt|d tj|d|i||d W 5 Q R X ntj|d|i||d d S )
|
Nr= r® rS rW Únotú 'Not' nodes are not implementedrX rÉ )ru r1 r( © r r5 rD rF r! r] r^ r_ r` )rc r r( r1 rÉ rd r) r) r* Útest_fails_and_or_notÒ s$ üüz$TestOperations.test_fails_and_or_notÚcharrr r c C sj t tj dd¡}d| d}|dkrVd}tjt|d tj|||d W 5 Q R X ntj|||d d S ) Nr= r® z(df + 2)[df > 1] > 0 z (df > 0)rS z$cannot evaluate scalar only bool opsrX rÁ r¯ )rc r± r( r1 rÉ re rd r) r) r* Útest_fails_ampersand_pipeñ s z(TestOperations.test_fails_ampersand_pipeN)3rí rî rï r` rT r! r# rð r rñ rX rY r^ r` rb rf rg rl rn rr rs rt r{ r} r~ r r r r
r r r r r r r r r r5 r{ r r r r¥ r§ r¨ rª r« r¬ r° r² r) r) r) r* rH ù st #
|
|
|
%
|
|
&
|
|
/ýþ
|
rH c @ sÄ e Zd Zdd Zejje ddej de ¡dd Z
|
ej de ¡dd Z d
|
d Z d d Zej dejejfejejfejejfejejfejejejejdg¡dd Zdd Zdd ZdS )ÚTestMathc O s | dd¡d |d< tj||S rI rK rM r) r) r* r` ÿ s z TestMath.evalz&Unary ops only implemented for numexprr Úfnc C sh t dtj d¡i}|j}| d}| |¡}tjdd tt||}W 5 Q R X tj ||dd d S )Nr¬ r@ z(a)Úignore©ÚallFrË )
|
r r5 rD rF r¬ r` Úerrstater1 ra r¶ )rc r´ rÉ r¬ r Úgotr´ r) r) r* Útest_unary_functions s
|
|
z TestMath.test_unary_functionsc C sz t tj d¡tj d¡d}|j}|j}| d}| |¡}tjdd tt|||}W 5 Q R X t j
|
||dd d S )Nr@ r z(a, b)rµ r¶ FrË ) r r5 rD rF r¬ r¯ r` r¸ r1 ra r )rc r´ rÉ r¬ r¯ r r¹ r´ r) r) r* Útest_binary_functions s
|
|
z TestMath.test_binary_functionsc C s` t tj d¡tj d¡d}|jd||dd |j}t t |j¡|j ¡}t
|
j ||dd d S )Nr@ r ze = arctan2(sin(a), b)T©r( r1 rw FrË ) r r5 rD rF r` r; Zarctan2Úsinr¬ r¯ ra r¶ ©rc r( r1 rÉ r¹ r´ r) r) r* Útest_df_use_case s üzTestMath.test_df_use_casec C s\ t tj d¡tj d¡d}|jd||dd |j}t |j|j ¡}t j
|
||dd d S )Nr@ r ze = sin(a + b)Tr¼ FrË ) r r5 rD rF r` r; r½ r¬ r¯ ra r¶ r¾ r) r) r* Ú test_df_arithmetic_subexpression) s
|
z)TestMath.test_df_arithmetic_subexpressionzdtype, expect_dtyper c C s t dtj d¡ |¡i}|jj|ks*t|jd||dd |j }t
|
|j¡}|j|jks^t||jksltt j ||dd d S )Nr¬ r@ z
|
b = sin(a)Tr¼ FrË ) r r5 rD rF r¢ r¬ rÆ r r` r¯ r½ ra r¶ )rc rÆ Z expect_dtyper( r1 rÉ r¹ r´ r) r) r* Útest_result_types0 s zTestMath.test_result_typesc C sF t dtj d¡i}d}tjt|d |jd||d W 5 Q R X d S )Nr¬ r@ z#"mysin" is not a supported functionrX zmysin(a)r\ )r r5 rD rF r! r] r2 r` ©rc r( r1 rÉ rd r) r) r* Útest_undefined_funcI s z TestMath.test_undefined_funcc C sF t dtj d¡i}d}tjt|d |jd||d W 5 Q R X d S )Nr¬ r@ z1Function "sin" does not support keyword argumentsrX zsin(x=a)r\ )r r5 rD rF r! r] rw r` r r) r) r* Útest_keyword_argP s zTestMath.test_keyword_argN)rí rî rï r` r! r# r$ r rð r rº r r» r¿ rÀ r5 Úint32rÙ Úint64r÷ r" Z
|
complex128r% rö rÁ rà rÄ r) r) r) r* r³ þ s. ÿ
|
|
|
|
|
|
ûþ
|
|
r³ r@ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestScopec C s$ d}t td tj|||d¡ d S )Nú
|
_var_s * 2rC r\ )ra r Ú_var_sr_ r` )rc r( r1 r; r) r) r* Útest_global_scope\ s
|
ÿzTestScope.test_global_scopec C sD d}t ¡ }tjd|||d t ¡ }| d¡ ||ks@td S )NrB úx + 1r Úlcls)r9 ry r_ r` rL r )rc r( r1 r} rÌ Zlcls2r) r) r* Útest_no_new_localsb s
|
|
|
z TestScope.test_no_new_localsc C s8 d}t ¡ }tjd||d t ¡ }||ks4td S )NrB rË r\ )Úglobalsry r_ r` r )rc r( r1 r} ZgblsZgbls2r) r) r* Útest_no_new_globalsj s
|
|
|
z TestScope.test_no_new_globalsc C s8 d}d}t jt|d tjd||i d W 5 Q R X d S )NrB rc rX rË rt ©r! r] r r_ r` )rc r( r1 r} rd r) r) r* Útest_empty_localsq s zTestScope.test_empty_localsc C s8 d}d}t jt|d tj|||i d W 5 Q R X d S )Nz name '_var_s' is not definedrÈ rX )r( r1 Z global_dictrÐ )rc r( r1 rd r; r) r) r* Útest_empty_globalsx s z TestScope.test_empty_globalsN)rí rî rï rÊ rÍ rÏ rÑ rÒ r) r) r) r* rÇ [ s
|
rÇ c C s8 d} t jt| d tjdddddd W 5 Q R X d S ) Nz Invalid engine 'asdf' passedrX r[ rB rC r\ Úasdf)ru r( ©r! r] ÚKeyErrorr_ r` ©rd r) r) r* Útest_invalid_engine s r× )Ú use_numexprrh ))Tr )FrS c C s< ddl m} t d| ¡ |d }||ks.tW 5 Q R X d S )Nr )Ú _check_engineúcompute.use_numexpr)Z pandas.core.computation.evalrÙ r_ Úoption_contextr )rØ rh rÙ ri r) r) r* Ú test_numexpr_option_respected s
|
rÜ c C sv t dd¡` tddddd d gdddddd gd
|
} | d ¡}td d gdd gd
|
ddgd }t ||¡ W 5 Q R X d S ) NrÚ FTrB rC r® r- r= rZ )r× ÚBz
|
A.isnull())ræ )r_ rÛ r rÚ ra r³ )rÉ ri rh r) r) r* Ú#test_numexpr_option_incompatible_op s ÿ
|
rÞ c C s8 d} t jt| d tjdddddd W 5 Q R X d S ) Nz Invalid parser 'asdf' passedrX r[ rB rC r\ rÓ )ru r1 rÔ rÖ r) r) r* Útest_invalid_parser¤ s rß )rS r rz z dict[str, type[BaseExprVisitor]]Ú_parsersc
|
C sN t | }|d| |}|jD ].}d}tjt|d t|| W 5 Q R X qd S )NrË znodes are not implementedrX )rà Zunsupported_nodesr! r] r^ r1 )r( r1 Z VisitorClassÚinstrQ rd r) r) r* Útest_disallowed_nodes² s
|
râ c C s2 d}t jtdd tj|| |d W 5 Q R X d S )Nzs +rp rX r\ ©r! r] rå r_ r` )r( r1 r; r) r) r* Útest_syntax_error_exprs¾ s rä c C s6 d}d}t jt|d tj|| |d W 5 Q R X d S )Nzs + tzname 's' is not definedrX r\ )r! r] rd r_ r` )r( r1 r; rd r) r) r* Útest_name_error_exprsÄ s rå Úexpressza + @bz@a + bz@a + @bc C sj d\}}|dkr<t jtdd tj|| |d W 5 Q R X n*t jtdd tj|| |d W 5 Q R X d S )N©rB rC rz zThe '@' prefix is onlyrX r\ zThe '@' prefix is notrã )r( r1 ræ r¬ r¯ r) r) r* Ú%test_invalid_local_variable_referenceË s rè c C sd d\}}| dkr@d}t jt|d tjd| |d W 5 Q R X n tjd| |d}||| ks`td S )Nrç r zVariables in expression .+rX zsin + dotted_liner\ )r! r] r r_ r` r )r( r1 r½ Z dotted_linerd r r) r) r* Útest_numexpr_builtin_raises× s ré c C s4 d}t jtdd tjd|| |d W 5 Q R X d S )N)é* g @zResolver of type .+rX r )Z resolversr( r1 )r! r] rw r_ r` )r( r1 Zcannot_resolver) r) r* Útest_bad_resolver_raisesâ s rë c C s. t jtdd tjd| |d W 5 Q R X d S )Nz expr cannot be an empty stringrX Ú r\ ©r! r] r2 r_ r` r\ r) r) r* Útest_empty_string_raisesè s rî c C s. t jtdd tjd| |d W 5 Q R X d S )Nz#only a single expression is allowedrX z 1 + 1; 2 + 2r\ rã r\ r) r) r* Ú$test_more_than_one_expression_raisesî s rï ÚcmprT r9 c
|
C sª t dd ttjji}|| }|| } || }d| d| d}d| d| d}d| d| d } ||| fD ]2}
|
d
|
} tjt| d tj |
|
||d W 5 Q R X qrd S ) Nc S s t j d¡S )Nr@ )r5 rD r² r) r) r) r* r~ ÷ r z0test_bool_ops_fails_on_scalars.<locals>.<lambda>rp r rq r rZ r r[ rÏ rX r\ )
|
Úintr> r5 rD rF r! r] r^ r_ r` ) r7 rð r9 r( r1 Úgenr r r r re rd r) r) r* Ú test_bool_ops_fails_on_scalarsó s
|
|
|
ró Úotherz'x'z...c C sN t ddddgi}| d| ¡}tdddgdd}tr>d |_t ||¡ d S )Nr× r¬ r¯ r: rØ F©rç )r r` r r rç ra r¶ )rô rÉ ri rh r) r) r* Útest_equals_various s rö c C s* d}t j}tj|| |d}||ks&td S )Nzinf + 1r\ )r5 Úinfr_ r` r )r( r1 r rh ri r) r) r* Útest_inf s rø Úcolumnu Temp(°C)u Capacitance(μF)c C sT t tj dd¡|dgd}||| dk }d| d}|j|| d}t ||¡ d S )Nr= rC r¯ râ ú`z` > 5)r( )r r5 rD rF rÚ ra r³ )r( rù rÉ rh Z query_stringri r) r) r* Útest_query_token s
|
rû c C s t ddgddggddgd}||jdk }|jd| |d }t ||¡ |d
|
krd }tjt|d |jd | |d W 5 Q R X n |jd | |d }t ||¡ d S )Nr r@ rB r r Úcountrâ z
|
~(cat > 0)r\ rS r® rX z not (cat > 0))r r rÚ ra r³ r! r] r^ )r( r1 rÉ rh ri rd r) r) r* Útest_negate_lt_eq_le) s rý c C st |dkr&| j tjjtd| d¡ ttjj dddd|dgd }||| d
|
k }|
|
| d ¡}t ||¡ d S ) N)ÚTrueÚFalser÷ ZInfz+GH 47859 DataFrame eval not supported with )r] r r éd )r@ rC r© Zcol1râ rZ z>6) r r r! r# r rÕ r r5 rD r² rÚ ra r³ )r0 rù rÉ rh ri r) r) r* Ú test_eval_no_support_column_name9 s þÿ r c C sÎ t dddgdddgddd gd
|
}|d d
}|d }|jd d d t dddgdddgddd gd
|
}t ||¡ | st ||d ¡ t |d |d ¡ n.tdddgd d}t ||¡ t |d |¡ d S )NrB rC r® r- r= rZ r5 é é )r× rÝ ÚCr× z A = B + CTrv r é é rõ )r r` ra r³ r¶ r )Zusing_copy_on_writerÉ Z result_viewr> rh r) r) r* Útest_set_inplaceN s " " r c @ s0 e Zd Zej ddddddgdg¡dd Zd S )
|
Ú TestValidateÚvaluerB rþ rC r® g @c C s0 d}t jt|d tjd|d W 5 Q R X d S )Nz8For argument "inplace" expected type bool, received typerX z2+2rv rí )rc r rd r) r) r* Útest_validate_bool_argsb s z$TestValidate.test_validate_bool_argsN)rí rî rï r! r# rð r
|
r) r) r) r* r a s r )fÚ
|
__future__r Ú functoolsr Ú itertoolsr r rD r Únumpyr5 r! Z pandas.errorsr r r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr% Zpandas.core.dtypes.commonr r r
|
r rz r_ r r r Zpandas._testingZ_testingra Zpandas.core.computationr r Z pandas.core.computation.enginesr Z pandas.core.computation.exprr r r Z#pandas.core.computation.expressionsr r Zpandas.core.computation.opsr r r r r Z pandas.core.computation.scoper Zfixturer( ZPARSERSr1 r< rF Úranger7 r9 r rH r rú r r rH r³ rF rÉ rÇ r& r× r# rð rÜ rÞ rß ZPyTablesExprVisitorrà Ú__annotations__râ rä rå rè ré rë rî rï rñ r> ró rö rø rû rý Úkeysr r r r) r) r) r* Ú<module> sÐ ôÿ
|
|
|
þ
|
\ Z %
|
þ
|
|
ý
|
|
þþ
|
|
þ
|
|
