U
|
¬ý°dd7 ã @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm Z
|
d dl m Z m Z mZ d dlmZ ejdd Zejdd Zejd d
|
Zejd d Zejd d Zejdd Zejdd Zejdd Zejdd Zejdd Z ejdd Z ej j ej dd G d d d Z!dS ) é N)Úoption_context)Ú DataFrameÚIndexÚSeries)Ú expressionsc C s t tj dd¡tdddS )Né' é ÚABCDÚfloat64©ÚcolumnsÚdtype©r ÚnpÚrandomÚrandnÚlist© r r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_expressions.pyÚ_frame s r c C s t tj dd¡tdddS )Néd r r r
|
r r r r r r Ú_frame2 s r c C s8 t | d ¡ | d d¡| d d¡| d d¡dS © NÚAÚBÚfloat32ÚCÚint64ÚDÚint32)r r r r ©r ÚcopyZastype©r r r r Ú_mixed s
|
üÿr# c C s8 t | d ¡ | d d¡| d d¡| d d¡dS r r ©r r r r Ú_mixed2( s
|
üÿr% c C s t tjjddddtdddS )Né r )r r ©Úsizer r r ©r r r Úrandintr r r r r Ú_integer4 s
|
ÿr+ c C s | t jjddt | ¡d S )Nr é r' )r r r* Úshape)r+ r r r Ú_integer_randint; s r. c C s t tjjddddtdddS )Nr& r )ée r r' r r r r) r r r r Ú _integer2A s
|
ÿr0 c C s | d j ¡ S ©Nr ©Úvaluesr! r" r r r Ú_arrayH s r4 c C s | d j ¡ S r1 r2 r$ r r r Ú_array2M s r5 c C s | d j ¡ S ©Nr r2 )r# r r r Ú _array_mixedR s r7 c C s | d j ¡ S r6 r2 )r% r r r Ú _array_mixed2W s r8 znot using numexpr)Úreasonc @ s e Zd Zejdddd ZeeedddZ ej
|
dd d
|
d d d ddg¡ej
|
dddg¡ej
|
dddddddg¡dd Z ej
|
dd d
|
d d d ddg¡ej
|
dddg¡dd Z d d Zej
|
d ¡ej
|
d d!d"d#d$d%g¡ej
|
d&d'd(g¡d)d* Zej
|
d&d'd(g¡d+d, Zej
|
d-ddg¡ej
|
dd d ddg¡d.d/ Zej
|
d0d1d2d3g¡d4d5 Zej
|
d0d6d7d8g¡d9d: Zej
|
d;ed<d=d>d?gd<d=d>d?ggd@dAdBdCgdDeddgddggd@dCgdDfed<dEd>d?gd<dFd>d?gd<d=d=dGggd@dAdBdCgdDeddgddgddggd@dCgdDfg¡dHdI Zej
|
ddJ¡ej
|
dKdL¡dMdN Zej
|
dOdPdQdRdSg¡ej
|
dTeeeg¡ej
|
dUdVdWg¡dXdY ZdZS )[ÚTestExpressionsT)Zautousec c s t j}d V |t _d S ©N)ÚexprÚ _MIN_ELEMENTS)ÚselfÚ min_elementsr r r Úsave_min_elements^ s z!TestExpressions.save_min_elements)ÚflexÚopnamec s\ |r fdd} |_ n
|
tt }tdd || |}W 5 Q R X t ¡ || |}||fS )Nc s t | |S r; )Úgetattr)ÚxÚy©rB r r Ú<lambda>g ó z)TestExpressions.call_op.<locals>.<lambda>úcompute.use_numexprF)Ú__name__rC Úoperatorr r< Úget_test_result)ÚdfÚotherrA rB ÚopÚexpectedÚresultr rF r Úcall_opd s
|
|
zTestExpressions.call_opÚfixturer+ r0 r. r r r# r% rA FÚarithÚaddÚsubÚmulÚmodÚtruedivÚfloordivc C s¾ | |¡}dt_| ||||¡\}}|dkrFtdd |jjD sFtt ||¡ t
|
t |j D ]X}| |j d d
|f |j d d
|f ||¡\}}|dkr¬|jjdks¬tt ||¡ q`d S )Nr rY c s s | ]}|j d kV qdS )ÚfN)Úkind)Ú.0rD r r r Ú <genexpr> s z6TestExpressions.test_run_arithmetic.<locals>.<genexpr>r[ )Úgetfixturevaluer< r= rR ÚallZdtypesr3 ÚAssertionErrorÚtmÚ assert_equalÚrangeÚlenr Úilocr r\ ) r> ÚrequestrS rA rT rM rQ rP Úir r r Útest_run_arithmetict s
|
0z#TestExpressions.test_run_arithmeticc C sz | |¡}|j}tdd | ¡ d }W 5 Q R X dt_t d¡ | ||||¡\}} t ¡ }
|
|
|
sjt dt
|
| |¡ dS )z²
|
tests solely that the result is the same whether or not numexpr is
|
enabled. Need to test whether the function does the correct thing
|
elsewhere.
|
rI Fr& r Tz Did not use numexpr as expected.N) r_ rJ r r! r< r= Z set_test_moderR rL ra rb rc ) r> rg rS rA Ú comparison_oprM rT rN rQ rP Z used_numexprr r r Útest_run_binary s
|
|
z TestExpressions.test_run_binaryc C sp t j d¡}t j d¡}t tjd ||d¡}|r4tt tjd||d¡}|rPtt tjd||d¡}|sltd S )NiAB r Úevaluateú+)r r r r< Ú_can_use_numexprrK rU ra )r> ÚarrayZarray2rQ r r r Ú test_invalidº s z TestExpressions.test_invalidz>ignore:invalid value encountered in true_divide:RuntimeWarningz opname,op_str)rU rm )rV ú-)rW Ú*)rY ú/)Úpowú**zleft_fix,right_fix)r4 r5 )r7 r8 c sd | |¡ | |¡ fdd}tdd | W 5 Q R X t d¡ | t ¡ | d S )Nc s dkr d S t t} t ¡ < d}t d|t¡ tj| dd}tj| dd}W 5 Q R X t ||¡ t
|
| d¡}|rt d S )Nrt z/invalid value encountered in divide|true_divideÚignoreT©Ú use_numexprFrl ) rC rK ÚwarningsÚcatch_warningsÚfilterwarningsÚRuntimeWarningr< rl rb Úassert_numpy_array_equalrn ra )rO ÚmsgrQ rP ©ÚleftÚop_strrB Úrightr r ÚtestitØ s
|
|
z/TestExpressions.test_binary_ops.<locals>.testitrI Fr& ©r_ r r< Zset_numexpr_threads)r> rg rB r Úleft_fixÚ right_fixr r r r Útest_binary_opsÊ s
|
|
|
z TestExpressions.test_binary_opsc sb | |¡| |¡ fdd}tdd | W 5 Q R X t d¡ | t ¡ | d S )Nc sb d } d } }t j|| dd}t j|| dd}t ||¡ t |||d¡}|r^td S )Nr& Trw Frl )r< rl rb r} rn ra )Zf12Zf22rO rQ rP ©rj r r r r r ù s z3TestExpressions.test_comparison_ops.<locals>.testitrI Fr& r )r> rg rj r
r r r r r Útest_comparison_opsò s
|
|
|
z#TestExpressions.test_comparison_opsÚcondc sV | |¡ fdd}tdd | W 5 Q R X t d¡ | t ¡ | d S )Nc sX t jjt jd} | ¡ t | jjd ¡}t | jjd ¡}t ||¡ d S )N)r r& )
|
r Úemptyr- Zbool_Úfillr< Úwherer3 rb r} )ÚcrQ rP ©r rM r r r s
|
|
z*TestExpressions.test_where.<locals>.testitrI Fr& r )r> rg r rS r r r r Ú
|
test_where s
|
|
zTestExpressions.test_wherez op_str,opname)rs rY )z//rZ )ru rt c C s* t tj d¡dktj d¡dkd}d| d}tt|}t |¡}tj t
|
|d ||| W 5 Q R X tj t
|
|d ||j |j W 5 Q R X tj t
|
|d ||j d W 5 Q R X tj t
|
|d |d|j W 5 Q R X tj t
|
|d |d| W 5 Q R X tj t
|
|d ||d W 5 Q R X d S ) Né
|
ç à?©ÚaÚbz
|
operator 'z!' not implemented for bool dtypes)ÚmatchTF) r r r ÚrandrC rK ÚreÚescapeÚpytestZraisesÚNotImplementedErrorr r )r> r rB rM r~ r[ Úerr_msgr r r Ú!test_bool_ops_raise_on_arithmetic" s &
|
|
z1TestExpressions.test_bool_ops_raise_on_arithmetic)rm rU )rr rW )rq rV c
|
C sÒ d}t tj |¡dktj |¡dkd}dddd}dd d
|
d }tt|}tt||| }|d krjd S tjd ddN t ¡ & |||} |||}
|
t | |
|
¡ W 5 Q R X t ¡ . ||j
|
|j } ||j
|
|j }
|
t | |
|
¡ W 5 Q R X t ¡ * ||j
|
d } ||j
|
d }
|
t | |
|
¡ W 5 Q R X t ¡ * |d|j
|
} |d|j
|
}
|
t | |
|
¡ W 5 Q R X t ¡ & |d|} |d|}
|
t | |
|
¡ W 5 Q R X t ¡ & ||d } ||d }
|
t | |
|
¡ W 5 Q R X W 5 Q R X d S )Nr r r ú|ú&ú^)rm rr rq Úor_Úand_Úxor)r r r rq Té )r? F) r r r r rC rK rb rx Zassert_produces_warningÚassert_frame_equalr r Zassert_series_equal) r> r rB ÚnrM ÚsubsZ sub_funcsr[ ZfeÚrÚer r r Ú test_bool_ops_warn_on_arithmetic> sB &
|
|
|
|
|
|
|
|
|
|
|
|
|
z0TestExpressions.test_bool_ops_warn_on_arithmeticztest_input,expectedr r& r, Zaar r r r )r é r Zbbc C s> |j d d
ddgf |j d d
ddgf ¡}t ||¡ d S )Nr r )ÚlocÚnerb r¥ )r> Z
|
test_inputrP rQ r r r Ú test_bool_ops_column_name_dtypen s .z/TestExpressions.test_bool_ops_column_name_dtype)rU rV rW rX rY rZ Úaxis)r r& c C s |}|dkr |j dd d
f }n|j d d
df }dt_t||}tdd |||d}W 5 Q R X |||d}t ||¡ d S )Nr& r rI F)r¯ )rf r< r= rC r rb r¥ ) r> r¯ rT r rM rN Zop_funcrP rQ r r r Útest_frame_series_axis s
|
z&TestExpressions.test_frame_series_axisrO Ú__mod__Ú__rmod__Ú __floordiv__Ú __rfloordiv__ÚboxÚscalaréûÿÿÿr¤ c C sÄ dt _t dd¡}||}t||}||}tdd ||}W 5 Q R X t ||¡ t|D ]`\} }
|
|t kr~|j
|
| df } n|| } ztt |
|
||}W n t k
|
r° Y q^X | |ks^t q^d S )Nr iÎÿÿÿé2 rI F)r< r= r ZarangerC r rb rc Ú enumerater rf ÚintÚZeroDivisionErrorra ) r> rO rµ r¶ ÚdataÚobjÚmethodrQ rP rh ÚelemZ scalar_resultr r r Ú,test_python_semantics_with_numexpr_installed s"
|
z<TestExpressions.test_python_semantics_with_numexpr_installedN)rJ Ú
|
__module__Ú __qualname__r rS r@ Ú staticmethodÚboolÚstrrR ÚmarkZ parametrizeri rk rp r{ r r r r rª r r® r° r r rÀ r r r r r: \ sÌ
|
|
ùþ ÿùþ ÿ þ ÿ ÿ
|
ÿ
|
ÿ
|
-
|
ÿü
|
þþûùþ
|
ÿ üþ r: )"rK r ry Únumpyr r Zpandasr Zpandas._testingZ_testingrb Zpandas.core.apir r r Zpandas.core.computationr r< rS r r r# r% r+ r. r0 r4 r5 r7 r8 rÆ ZskipifZ USE_NUMEXPRr: r r r r Ú<module> s@
|
|
|
|
|
|
|
|
|
|
|
|
