U
|
¬ý°d"Ó ã @ s U d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dl Z
|
d dl Z d dl Z d dl mZmZmZmZmZmZ d dlmZ d dlmZ d dlmZ d d l m Z m Z e j eeej gd
|
d d Z!d d Z"dd Z#dddddddddddg Z$ed d d gZ%d e&d < e% 'd d! e$D ¡ e% 'd"d! e$D ¡ G d#d$ d$Z(G d%d& d&Z)G d'd( d(Z*G d)d* d*Z+G d+d, d,Z,G d-d. d.Z-G d/d0 d0Z.G d1d2 d2Z/d3d4 Z0d5d6 Z1e j2 3d7d d8d9gd d9d:gfg¡d;d< Z4d=d> Z5e j2 3d?d8d9d@g¡e j2 3dAd d8d9d@g¡dBdC Z6dDdE Z7dS )Fé )Ú annotations)Úabc)Ú timedelta)ÚDecimalN)ÚIndexÚ
|
RangeIndexÚSeriesÚ TimedeltaÚTimedeltaIndexÚarray)Úops)Ú expressions)Úassert_invalid_addsub_typeÚassert_invalid_comparison)Úparamsc C s | j S )zL
|
Fixture to test behavior for Index, Series and tm.to_array classes
|
)Úparam)Úrequest© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arithmetic/test_numeric.pyÚbox_pandas_1d_array s r c C s8 t t | ¡¡ ¡ r4t t | ¡¡ ¡ s,t|d9 }|S )z_
|
Helper to adjust the expected result if we are dividing by -0.0
|
as opposed to 0.0
|
éÿÿÿÿ)ÚnpZsignbitr ÚanyÚallÚAssertionError)ÚzeroÚexpectedr r r Úadjust_negative_zero( s r c C s |t jtjfkrt | ¡n| }|t jtjfkr8t |¡n|}|||}| ||¡}t|trx|j
|
| j ¡sx|j d ¡|_ t ||¡ d S ©N)r ÚrpowÚoperatorÚpowr ÚabsÚcombineÚ
|
isinstancer ÚindexÚequalsÚ
|
_with_freqÚtmÚassert_series_equal)ÚseriesÚotherÚopÚleftÚrightZcython_or_numpyÚpythonr r r Ú
|
compare_op7 s
|
r0 Úi1Úi2Úi4Úi8Úu1Úu2Zu4Úu8Úf2Zf4Úf8é
|
é( zlist[Index | Series]Úleftsc C s g | ]}t d ddg|dqS )r: é é ©Údtype)r ©Ú.0r@ r r r Ú
|
<listcomp>G s rC c C s& g | ] }|d krt dddg|dqS )r8 r: r= r> r? )r rA r r r rC H s c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestNumericComparisonsc C s° t d¡tdddgk}dtdddgk}t ||¡ tdddgt d¡k }tdddgdk }t ||¡ t dddg¡d tdddgk}dtdddgk}t ||¡ d S )Nr é é é ç )r Úfloat64r r( r) r )ÚselfÚresultr r r r Ú(test_operator_series_comparison_zerorankO s "z?TestNumericComparisons.test_operator_series_comparison_zerorankc C s. |}t td¡}t ||¡}t||| d S ©Né )r r Úranger( Ú box_expectedr )rJ Úbox_with_arrayÚ fixed_now_tsÚtsÚobjr r r Ú test_df_numeric_cmp_dt64_raises[ s z6TestNumericComparisons.test_df_numeric_cmp_dt64_raisesc C sL t tj d¡dd}t tj d¡}t d¡|_t || d|| ¡ d S )NrN r ©Únamez
|
2000-01-01rE ) r r ÚrandomÚrandnÚpdÚ TimestamprW r( r) )rJ ÚaÚbr r r Útest_compare_invalidc s z+TestNumericComparisons.test_compare_invalidc C s¬ |}|t k r|ntj}ttj d¡}tj||dd}|dk}ttjdt d}tj||dd}t
|
||¡ |dk}t
|
|| ¡ d}t j t |d |dk W 5 Q R X d S )Ni F©Z transposer\ r? z0Invalid comparison between dtype=float64 and str©Úmatch)r r Zndarrayr rX rY r( rP ÚzerosÚboolÚ assert_equalÚpytestÚraisesÚ TypeError)rJ rQ ÚboxZxboxrT rK r Úmsgr r r Ú$test_numeric_cmp_string_numexpr_pathk s z;TestNumericComparisons.test_numeric_cmp_string_numexpr_pathN)Ú__name__Ú
|
__module__Ú __qualname__rL rU r^ rj r r r r rD N s rD c @ sî e Zd Zej dejee g¡ejjde
|
dd ddd Z ej dejee g¡ejjde
|
dd dd d
|
Z ejjd e d d e d d ¡ e d d ¡ e d d ¡ d¡e d d ¡ d¡gdd ddd Zejjd e d d e d d ¡ e d d ¡ gdd dej dejejg¡dd Zdd Zejjde dde dd ¡ e dd ¡ e dd ¡ d ¡e d ¡e d d ¡ej d ¡ej d ¡ejd!d"d#e d!¡e d!¡ ¡ ejd!d$d# ¡ e d!¡ ¡ e d d%¡ej ge dd&d' Z!d(S ))Ú.TestNumericArraylikeArithmeticWithDatetimeLikeÚbox_clsr- c C s t | jt| j S r ©Útyperk Ústrr@ ©Úxr r r Ú<lambda> ó z7TestNumericArraylikeArithmeticWithDatetimeLike.<lambda>©Úidsc C s t jdddgdd}||}tdddg|jd}t|tsB|tkrJt|}|j|jksZt|| }t ||¡ || }t ||¡ d S ) NrE rF rG úm8[s]r? Z10sZ40sZ90s) r r r
|
r@ r$ r r r( rd )rJ r- ro r. r rK r r r Útest_mul_td64arr s z?TestNumericArraylikeArithmeticWithDatetimeLike.test_mul_td64arrc C s t | jt| j S r rp rs r r r ru rv c C sÒ t jdddgdd}||}tdddg|jd}t|tsB|tkrJt|}|j|jksZt|| }t ||¡ || }t ||¡ d }t j
|
t |d
|
|| W 5 Q R X d }t j
|
t |d
|
|| W 5 Q R X d S ) Nr: r; éZ ry r? Z1sZ2sZ3sz5ufunc '(true_)?divide' cannot use operands with typesr` z3ufunc 'floor_divide' cannot use operands with types) r r r
|
r@ r$ r r r( rd re rf rg )rJ r- ro r. r rK ri r r r Útest_div_td64arr s z?TestNumericArraylikeArithmeticWithDatetimeLike.test_div_td64arrÚ scalar_tdrE ©Údaysztimedelta64[s]ztimedelta64[ms]c C s
|
t | jS r ©rq rk rs r r r ru à rv c
|
C s |}|}t dd tt|D }t|tjr@|j}| |¡}nt|t krV| d¡}t
|
||¡}t
|
||¡}|| }t
|
||¡ || } t
|
| |¡ d S )Nc S s g | ]}t |d qS )r~ )r )rB Únr r r rC É s z`TestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_mul_tdscalar.<locals>.<listcomp>úm8[us]) r
|
rO Úlenr$ r Ú timedelta64r@ Úastyperq r r( rP rd )
|
rJ r} Ú numeric_idxrQ rh r% r r@ rK Zcommuter r r Ú test_numeric_arr_mul_tdscalarº s
|
zLTestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_mul_tdscalarc C s
|
t | jS r r rs r r r ru à rv r@ c
|
C s |}t d¡jt jdd}|j|dd}tj||dd}| d¡ d¡}t|tkr^| d¡}tj||dd}|| } t | |¡ || } t | |¡ d S )Ni N F©Úcopyr_ ztimedelta64[D]ztimedelta64[ns]ztimedelta64[us])
|
r Úaranger
Úint64r( rP Úviewrq r rd )
|
rJ r@ r} rQ rh Zarr_i8ÚarrrT r rK r r r Ú*test_numeric_arr_mul_tdscalar_numexpr_pathÙ s
|
zYTestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_mul_tdscalar_numexpr_pathc
|
C s¾ |}|dd
}t ddg}t|tjrR|j}|t d¡k rFt d¡}| |¡}nt|tkrh| d¡}t ||¡}t ||¡}|| }t
|
||¡ d} t j t | d || W 5 Q R X d S ) NrE rG z3 Daysz36 Hoursry r ú$cannot use operands with types dtyper` )r
|
r$ r r r@ r
rq r r( rP rd re rf rg )
|
rJ Z
|
three_daysr rQ rh r% r r@ rK ri r r r Ú test_numeric_arr_rdiv_tdscalarù s"
|
|
zMTestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_rdiv_tdscalarr+ é )Úhourszm8[h]ÚNaTÚDrG r z
|
2021-01-01z
|
Asia/Tokyo)ÚtzÚUTCÚnsc C s> |}t ||¡}d dddddddd d
|
d g
|
¡}t||| d S ) Nú|úunsupported operand typez3Addition/subtraction of integers and integer-arraysz Instead of adding/subtractingr z;Concatenation operation is not implemented for NumPy arraysz+Cannot (add|subtract) NaT (to|from) ndarrayzBoperand type\(s\) all returned NotImplemented from __array_ufunc__z(can only perform ops with numeric valuesz*cannot subtract DatetimeArray from ndarrayz.Cannot add or subtract Timedelta from integers)r( rP Újoinr )rJ r r+ rQ rh r- ri r r r Ú&test_add_sub_datetimedeltalike_invalid s ôÿzUTestNumericArraylikeArithmeticWithDatetimeLike.test_add_sub_datetimedeltalike_invalidN)"rk rl rm re ÚmarkÚ parametrizer r r r r< rz r| r Zto_timedelta64Zto_pytimedeltar
r r rI r r r rZ ÚoffsetsZMinuteZSecondr[ Ú to_pydatetimeZ to_datetime64Z
|
datetime64r Úreprr r r r r rn
sr ÿ ÿ û÷
|
ýù
|
|
|
|
ïërn c @ sö e Zd Zdd Zdd Zdd Zdd Zej d e
|
j e
|
j g¡d
|
d Z ej d ejejejg¡d d Zej d ejejejg¡dd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Z d#d$ Z d%d& Z d'd( Z d)d* Z d+S ),ÚTestDivisionByZeroc C sv |}t tjtjtjtjtjgtjd}t||}|| }t ||¡ t| d¡t
|
|¡ d¡ }t |t|¡ d S ©Nr? r4 © r r ÚnanÚinfrI r r( Úassert_index_equalr r
r r) ©rJ r r Úidxr Z expected2rK Ú
|
ser_compatr r r Ú test_div_zeroG s "
|
z TestDivisionByZero.test_div_zeroc C sv |}t tjtjtjtjtjgtjd}t||}|| }t ||¡ t| d¡t
|
|¡ d¡ }t |t|¡ d S r¢ r£ r§ r r r Útest_floordiv_zeroT s "
|
z%TestDivisionByZero.test_floordiv_zeroc C sl |}t tjtjtjtjtjgtjd}|| }t ||¡ t| d¡t |¡ d¡ }t |t|¡ d S r¢ )
|
r r r¤ rI r( r¦ r r
r r) )rJ r r r¨ r rK r© r r r Ú test_mod_zeroa s " z TestDivisionByZero.test_mod_zeroc C s |}t tjtjtjtjtjgtjd}t tjtjtjtjtjgtjd}t||}t||}t |d |¡ t |d |¡ d S )Nr? r rE ) r r r¤ r¥ rI r Údivmodr( r¦ )rJ r r r¨ ZexleftZexrightrK r r r Útest_divmod_zeroj s ""
|
|
z#TestDivisionByZero.test_divmod_zeror, c C sd |j tjkrd S |d }ttj tj tj tjtjgtjd}t||}|||}t ||¡ d S )NrG r? )
|
r@ r Úuint64r r¥ r¤ rI r r( r¦ )rJ r r r, r¨ r rK r r r Útest_div_negative_zerou s (
|
|
z)TestDivisionByZero.test_div_negative_zeroÚdtype1c C sÔ |}t ddddgdd |¡}t ddddgdd |¡}tjd d
|
$ t |j tj¡|j d d d }W 5 Q R X tj|jdd
< |jd krª|jdkrªt j
|
rª|dkrª| d¡}|| }t ||¡ | || ¡rÐtd S )NrG é rN é ÚfirstrV r ÚsecondÚignore©r rI ©r@ rW r Úfloat32)r r
r ÚerrstateÚvaluesrI r¥ Úilocr@ ÚexprZ USE_NUMEXPRr( r) r& r ) rJ Zswitch_numexpr_min_elementsr± Úany_real_numpy_dtypeÚdtype2r´ rµ r rK r r r Útest_ser_div_ser s ý
|
z#TestDivisionByZero.test_ser_div_serc C sÌ |}t ddg |¡}t ddg |¡}|| || f}t|}|d tj¡|d< tj|d d< t||}t |d |d ¡ t |d |d ¡ t|j |}t |d |d ¡ t |d |d ¡ d S )NrE r rF )
|
r r
Úlistr rI r¥ r r( r) r» )rJ r± r¾ r¿ r- r. r rK r r r Útest_ser_divmod_zero¢ s
|
z'TestDivisionByZero.test_ser_divmod_zeroc C s t tjdg}t tjdg}|| || f}t||}t |d |d ¡ t |d |d ¡ t|j|}t |d |d ¡ t |d |d ¡ d S )Nç ð?ç @r rE )r r r¥ r r( r) r» )rJ r- r. r rK r r r Útest_ser_divmod_inf¹ s
|
z&TestDivisionByZero.test_ser_divmod_infc C sz t dgd ¡}t j d¡}tdgd }|t| }t ||¡ t|| }t ||¡ t|t| }t ||¡ d S )Nr rN rH )r r rX rY r r( r) )rJ Z
|
zero_arrayÚdatar rK r r r Útest_rdiv_zero_compatÈ s z(TestDivisionByZero.test_rdiv_zero_compatc C sD t dddgdd}t tj tjtjgdd}|d }t ||¡ d S ©Nr r rE r´ rV ©r r r¥ r¤ r( r) ©rJ Úserr rK r r r Útest_div_zero_inf_signs× s z*TestDivisionByZero.test_div_zero_inf_signsc C s> t dddgdd}t dtjdgdd}d| }t ||¡ d S )Nr r rE r´ rV rH )r r r¤ r( r) rÊ r r r Útest_rdiv_zeroß s z!TestDivisionByZero.test_rdiv_zeroc C sD t dddgdd}|d }t tj tjtjgdd}t ||¡ d S rÈ rÉ )rJ rË rK r r r r Útest_floordiv_divç s z$TestDivisionByZero.test_floordiv_divc C sn t ddddgddddgd¡}|| }tddddg}ttjtjtjdg}t ||d¡}t ||¡ d S ) NrG r² rN r³ r ©r´ rµ rà rE )rZ Ú DataFramer r r¤ r( Úassert_frame_equal)rJ ÚdfrK r´ rµ r r r r Útest_df_div_zero_dfï s z&TestDivisionByZero.test_df_div_zero_dfc C s¤ t ddddgddddgd¡}tddddg}ttjtjtjdg}t ||d¡}tjd d
|
|j d ¡|j }W 5 Q R X t j||j|j d }t
|
||¡ d S ) NrG r² rN r³ r rÏ rà rE r¶ r· Úfloat©r% Úcolumns) rZ rÐ r r r¤ rº r» r
r% rÖ r( rÑ )rJ rÒ r´ rµ r r rK r r r Útest_df_div_zero_arrayù s z)TestDivisionByZero.test_df_div_zero_arrayc C sª t ddddgddddgd¡}|d }t jtj|j|jd}tj|jdd
df< t ||¡ tj
|
d d
|
|j d ¡d }W 5 Q R X t j||j|jd}t ||¡ d S ) NrG r² rN r³ r rÏ rÕ rE r¶ r· rI ) rZ rÐ r r¥ r% rÖ r¤ r¼ r( rÑ rº r» r
©rJ rÒ rK r r Úresult2r r r Útest_df_div_zero_int s z'TestDivisionByZero.test_df_div_zero_intc C sJ t tj dd¡¡}|d }|| }|| }| d¡ | d¡¡rFtd S ©Nr: rN r ©rZ rÐ r rX rY Zfillnar& r ©rJ rÒ rË ÚresZres2r r r Ú(test_df_div_zero_series_does_not_commute s
|
z;TestDivisionByZero.test_df_div_zero_series_does_not_commutec C sî t ddddgddddgd¡}tddddg}|s>| d¡}ttjtjtjdg}t ||d¡}|| }t ||¡ t jddddgddddgddd }tddddgd
|
d }ttjtjtjdg}t ||d¡}|| }t ||¡ d S ) NrG r² rN r³ r rÏ rI Fr r r? )rZ rÐ r r
r r¤ r( rÑ )rJ Zusing_array_managerrÒ r´ rµ r rK r r r Útest_df_mod_zero_df s
|
$z&TestDivisionByZero.test_df_mod_zero_dfc C s¸ t ddddgddddgd¡}tddddgdd}ttjtjtjdg}t ||d¡}tjd d
|
|j|j }W 5 Q R X t j||j|jdd }tj|j dd
d f< t
|
||¡ d S ) NrG r² rN r³ r rÏ rI r? r¶ r· )r% rÖ r@ rE ) rZ rÐ r r r¤ rº r» r% rÖ r¼ r( rÑ )rJ rÒ r´ rµ r r rÙ r r r Útest_df_mod_zero_array8 s z)TestDivisionByZero.test_df_mod_zero_arrayc C s t ddddgddddgd¡}|d }t jtj|j|jd}t ||¡ tjdd |j
|
d
|
¡d }W 5 Q R X t j||j|jd}t ||¡ d S ) NrG r² rN r³ r rÏ rÕ r¶ r· rI ) rZ rÐ r r¤ r% rÖ r( rÑ rº r» r
rØ r r r Útest_df_mod_zero_intI s z'TestDivisionByZero.test_df_mod_zero_intc C sJ t tj dd¡¡}|d }|| }|| }| d¡ | d¡¡rFtd S rÛ rÜ rÝ r r r Ú(test_df_mod_zero_series_does_not_commuteW s
|
z;TestDivisionByZero.test_df_mod_zero_series_does_not_commuteN)!rk rl rm rª r« r¬ r® re r r r ÚtruedivÚfloordivr° r r rI r¯ rÀ r rÅ rÇ rÌ rÍ rÎ rÓ r× rÚ rß rà rá râ rã r r r r r¡ F s.
|
|
|
|
r¡ c @ sò e Zd Zdd Zdd Zdd Zej de j
|
e j e j g¡dd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zej de je jg¡dd Zdd Zdd Zd d Zd d Zej d e j d!d"d#d$e j g¡d%d& Z d'd( Z d)d* Z!d+d, Z"d-S ).ÚTestMultiplicationDivisionc C sl |}t tdg}t tdg}t ||¡}t ||¡}|td }t ||¡ |td }t ||¡ d S )Nr: rN rF )r r r( rP rd )rJ rQ rh rË r rK r r r Útest_divide_decimale s z.TestMultiplicationDivision.test_divide_decimalc C sb t ddgdd}t ddgdd}t dtj g}| |¡}tj||dd || }t ||¡ d S )
|
NrE r r´ rV g{®Gáz¿g{®Gáz¿rµ F©Z check_names)r r r¥ Údivr( r) )rJ r´ rµ r rK r r r Útest_div_equiv_binopu s
|
z/TestMultiplicationDivision.test_div_equiv_binopc C sH |}|d }| d¡}t ||¡ |d }t|jd }t ||¡ d S )NrE rI rF )r
r( r¦ r r» )rJ r r¨ rK r r r r Ú test_div_int s
|
z'TestMultiplicationDivision.test_div_intr, c C s* |}t ||¡}||d}t ||¡ d S ©NrE )r( rP rd )rJ r, r rQ r¨ rK r r r Útest_mul_int_identity s
|
z0TestMultiplicationDivision.test_mul_int_identityc C sd |}|| }|t jddd }t ||d ¡ |jt jkr>dnd}|t jd|d }t ||¡ d S )NrN r r? r¯ )r r r( r¦ r@ r¯ r )rJ r r¨ ÚdidxrK Ú arr_dtyper r r Útest_mul_int_array s z-TestMultiplicationDivision.test_mul_int_arrayc C sJ |}|| }|j tjkr dnd}|ttjd|d }t |t|¡ d S )Nr¯ r rN r? )r@ r r¯ r r r( r) )rJ r r¨ rî rï rK r r r Útest_mul_int_series s
|
z.TestMultiplicationDivision.test_mul_int_seriesc C sB |}t jddd}|t|d }t||d }t ||¡ d S )NrN rI r? g¹?)r r r r( r) )rJ r r¨ Zrng5rK r r r r Útest_mul_float_series¨ s
|
z0TestMultiplicationDivision.test_mul_float_seriesc C s |}|| }t ||d ¡ d S ©NrF ©r( r¦ )rJ r r¨ rK r r r Útest_mul_index° s z)TestMultiplicationDivision.test_mul_indexc C s8 |}d}t jt|d |tjddd W 5 Q R X d S )Nz,cannot perform __rmul__ with this index typer` Ú20130101rN ©Zperiods)re rf rg rZ Ú
|
date_range©rJ r r¨ ri r r r Útest_mul_datelike_raises¶ s z3TestMultiplicationDivision.test_mul_datelike_raisesc C sb |}d}t jt|d ||dd
W 5 Q R X t jt|d |t ddg¡ W 5 Q R X d S )Nz(operands could not be broadcast togetherr` r rG rE rF )re rf Ú
|
ValueErrorr r rù r r r Ú test_mul_size_mismatch_raises¼ s z8TestMultiplicationDivision.test_mul_size_mismatch_raisesc C sJ |}|}t ||jd}t ||¡}t ||¡}||d}t ||¡ d S )NrÄ ©r r» r( rP rd )rJ r, r rQ rh r¨ r rK r r r Útest_pow_floatÄ s
|
z)TestMultiplicationDivision.test_pow_floatc C sF |}|}t |jd }t ||¡}t ||¡}|d }t ||¡ d S ró rý )rJ r rQ rh r¨ r rK r r r Ú test_moduloÑ s z&TestMultiplicationDivision.test_moduloc C sj |}t |d}tjdd t |jd\}}W 5 Q R X t|t|f}t||D ]\}}t ||¡ qPd S )NrF r¶ r· )r r rº r» r Úzipr( r¦ ) rJ r r¨ rK ré Úmodr ÚrÚer r r Útest_divmod_scalarÝ s
|
z-TestMultiplicationDivision.test_divmod_scalarc
|
C s |}t j|jj|jjdd }t||}t jdd t|j|\}}W 5 Q R X t|t|f}t||D ]\}} t
|
|| ¡ qjd S ©Nr? rF r¶ r· ) r Úonesr» Úshaper@ r rº r r r( r¦ ©
|
rJ r r¨ r+ rK ré r r r r r r r Útest_divmod_ndarrayè s
|
z.TestMultiplicationDivision.test_divmod_ndarrayc
|
C s |}t j|jj|jjdd }t|t|}t jdd t|j|\}}W 5 Q R X t|t|f}t||D ]\}} t
|
|| ¡ qnd S r ) r r r» r r@ r r rº r r( r) r r r r Útest_divmod_seriesô s z-TestMultiplicationDivision.test_divmod_seriesr+ é iéÿÿÿgX9´Èv¾@g
ëQ¸ Àc C s t j dd¡}dd }||}t |t |¡ ||| ¡ t t |¡| ||| ¡ t |t |¡ ||| ¡ t t |¡| ||| ¡ d S )NrN rG c S s t j| tddddgdS )NZABCDEZjimZjoeZjolierÕ )rZ rÐ rÁ rs r r r ru s ÿz?TestMultiplicationDivision.test_ops_np_scalar.<locals>.<lambda>)r rX rY r( rÑ r )rJ r+ ÚvalsÚfrÒ r r r Útest_ops_np_scalar s z-TestMultiplicationDivision.test_ops_np_scalarc C s t ¡ }d|_t d|i¡}t j|| ||d dd t j|| ||d dd t j||k ||d k dd t j|| ||d dd d S )NrS ÚAFrè )r( ÚmakeTimeSeriesrW rZ rÐ r) )rJ rS rÒ r r r Útest_operators_frame s z/TestMultiplicationDivision.test_operators_framec C s( t jdd t ddddgddddgd¡}|d |d
|
}t|d j|d
|
j d d }t j|jdd
< t ||¡ |d d }tt j|j
|
d d }t ||¡ | d ¡}|d |d
|
}t|d j|d
|
j }t ||¡ | d ¡}|d |d
|
}|d
|
|d }| |¡rt W 5 Q R X d S )Nr¶ r· rG r² rN r³ r rÏ r´ rµ rI r? ©r% rW )r rº rZ rÐ r r» r¤ r¼ r( r) r% r
r& r )rJ ÚprK r rÙ r r r Ú test_modulo2 s"
|
|
z'TestMultiplicationDivision.test_modulo2c C sn t jddX tddg}|d }tt jt jg}t ||¡ d| }tt jdg}t ||¡ W 5 Q R X d S )Nr¶ r· r rE rH )r rº r r¤ r( r) )rJ ÚsrK r r r r Útest_modulo_zero_int4 s z/TestMultiplicationDivision.test_modulo_zero_intN)#rk rl rm rç rê rë re r r r Úmulr Zrmulrå rí rð rñ rò rõ rú rü r! r rþ rÿ r r r
|
r r¤ r¥ r r r r r r r r ræ a s,
|
|
|
|
ræ c @ sL e Zd Zej dedddgedddedddgeddded d
|
ej ej ged ddfedddgedddeddddged ddedd d ej ged ddfg¡dd Z
|
ej de j ddddgiedde j ddddgiedde j dd d
|
ej ej gied dfe j ddddgiedde j dddddgied de j ddd d ej gied dfg¡dd Z dd Zdd Zdd Zdd Zdd Zej d d ¡d d Zejjd!d"d# d$d# d%d# gd&d'd(gd)d*d+ Zejjd!d,d# d-d# gd.d(gd)d/d0 Zejjd!d1d# d2d# d3d# gd&d'd(gd)d4d5 Zd6d7 Zd8S )9ÚTestAdditionSubtractionzfirst, second, expectedrE rF rG ÚABCrt r ZABDç @ç @ÚABCDr² rN c C s$ t || |¡ t || |¡ d S r )r( r) ©rJ r´ rµ r r r r Útest_add_seriesF s z'TestAdditionSubtraction.test_add_series©r% c C s$ t || |¡ t || |¡ d S r )r( rÑ r r r r Útest_add_framesZ s z'TestAdditionSubtraction.test_add_framesc C sØ t t dd¡}d| }| dd ¡}t ||¡ t d|i¡}d| }t d| dd ¡i¡}t ||¡ t ¡ }d|_ |
|
¡ }d d
|
d g¡}t j t|d || W 5 Q R X t j t|d || W 5 Q R X d S ) NrN r: Úfoo_c S s d| S ©Nr! r rs r r r ru s rv zDTestAdditionSubtraction.test_series_frame_radd_bug.<locals>.<lambda>r c S s d| S r" r rs r r r ru x rv rS r r zConcatenation operationr` )r r( Z rands_arrayÚmapr) rZ rÐ rÑ r rW r r re rf rg ) rJ rR r rK r ÚframerS Zfix_nowri r r r Útest_series_frame_radd_bugo s( ýÿz2TestAdditionSubtraction.test_series_frame_radd_bugc C s t tj d¡}||j ¡ }||j }t ||¡ t tj dddtj dddd}||j ¡ }||j }t ||¡ d}t
|
j t |d ||j ¡ }W 5 Q R X tjtj dd¡tj dddd}t d¡|d < |d |j ¡ |d
|
< |d |j |d < tj|d |d
|
d d d S )NrN rö r÷ r z.cannot subtract PeriodArray from DatetimeArrayr` rF Z20130102Údater rK Frè )r r rX rY r% Z to_seriesr( r) rZ rø re rf rg Z to_periodrÐ r[ )rJ rË r rK ri rÒ r r r Útest_datetime64_with_index s,
|
þ
|
ÿz2TestAdditionSubtraction.test_datetime64_with_indexc C sð |}t j d¡}t|t |j¡d}|| }|| }| ¡ D ]P\}}| ¡ D ]>\} }
|
|| | d } t |
|
¡s||
|
| kstqLt | ¡sLtqLq<| ¡ D ]T\}}| ¡ D ]B\} }
|
|| | || } t |
|
¡sÚ|
|
| ksètq¦t | ¡s¦tq¦qd S )Nr² r rF )r rX r r rÖ ÚitemsÚisnanr ) rJ Ú float_framer$ ÚgarbageZ colSeriesZidSumZ seriesSumÚcolr* r¨ ÚvalZorigValr r r Útest_frame_operators s"
|
|
z,TestAdditionSubtraction.test_frame_operatorsc C s6 t j|ddddgd}|| }|d }t ||¡ d S )Nr ÚCÚBr )rÖ rF )rZ rÐ r( rÑ )rJ r* Zframe2Úaddedr r r r Ú test_frame_operators_col_alignÆ s z6TestAdditionSubtraction.test_frame_operators_col_alignc C s: t ddd dgi¡}t || t ddtjdgi¡¡ d S )Nr\ r] ZaaZbb)rZ rÐ r( rÑ r r¤ )rJ rÒ r r r Ú test_frame_operators_none_to_nanÌ s z8TestAdditionSubtraction.test_frame_operators_none_to_nanr@ )rÔ r c C sZ t j|dt jdg|dt jdg|dg}|D ]&}|| |¡sDtt || |¡ q.d S )Nr? r )rÖ r@ r ©r% r@ )rZ rÐ r& r r( rÑ )rJ r@ ÚframesrÒ r r r Ú test_frame_operators_empty_likeÐ s
|
ýz7TestAdditionSubtraction.test_frame_operators_empty_likeÚfuncc C s | d S ró r rs r r r ru Þ rv z TestAdditionSubtraction.<lambda>c C s | d d d
S ró r rs r r r ru Þ rv c C s dS rM r rs r r r ru Þ rv ÚmultiplyÚsliceZconstantrw c C s* |}t ¡ d¡}||}t||| d S ©NrS ©r( r Úrenamer0 )rJ Zall_arithmetic_functionsr7 r, r* r+ r r r Ú test_series_operators_arithmeticÜ s z8TestAdditionSubtraction.test_series_operators_arithmeticc C s | d S rì r rs r r r ru è rv c C s dS rM r rs r r r ru è rv Úaddc C s* |}t ¡ d¡}||}t||| d S r: r; )rJ Z comparison_opr7 r, r* r+ r r r Ú test_series_operators_compareç s z5TestAdditionSubtraction.test_series_operators_comparec C s | d S ró r rs r r r ru ò rv c C s | d d d
S ró r rs r r r ru ò rv c C s dS rM r rs r r r ru ò rv c
|
C sê t ¡ d¡}||}t||}t|tjrbt|t|krbg }|D ]}| |¡ | t j
|
¡ qDn|}t |¡}t j dd t|j t |¡}W 5 Q R X t||D ]@\}} t t |¡| ¡ |j|jksÎtt |j|j d ¡¡ q¤d S )NrS r¶ r· )r( r r< r r$ r ÚIterabler Úappendr r¤ Zasarrayrº r» r Zassert_almost_equalrW r r¦ r% r' )
|
rJ r7 r* r+ ÚresultsZother_npr Ú expectedsrK r r r r Ú test_divmodð s
|
|
|
z#TestAdditionSubtraction.test_divmodc C s| t ¡ d¡}|d }t||}ttjgt| |jdd}ttj gt| |jdd}t
|
|d |¡ t
|
|d |¡ d S )NrS r r rE ) r( r r< r r r r¥ r r% r¤ r) )rJ Ztserr+ rK Úexp1Úexp2r r r Útest_series_divmod_zero s
|
z/TestAdditionSubtraction.test_series_divmod_zeroN)rk rl rm re r r r rÁ r r¤ r rZ rÐ r r% r' r. r2 r3 r6 r= r? rD rG r r r r r B sh ýýúþ
|
ý ýúþ
|
|
ý
|
ÿ
|
ý
|
r c @ s² e Zd Zej deeeg¡ej de j
|
e j e j g¡dd Z ej deeg¡ej de j
|
e j e j g¡dd Zej deeg¡ej de j
|
e j e j g¡dd Zd d
|
Zd S ) ÚTestUFuncCompatÚholderr@ c C s |t kr t nt}|tkrD|tjkr4t d| d¡ tdddd}n|tjd|ddd}t |¡}|t tjd|d¡dd}t
|
||¡ d S )Nzdtype z not relevant for RangeIndexr rN ÚfoorV r? ) r r r r r re Úskipr Úsinr( rd )rJ rI r@ rh r¨ rK r r r r Útest_ufunc_compat! s
|
|
z!TestUFuncCompat.test_ufunc_compatc C s: |dddddg|dd}|t kr$t nt}t |¡}|jdkrFt||sJttt tjdddddgtjd ¡dd
|
}t
|
||¡}t ||¡ t |d ¡}|jdkrªt||s®ttd d dd dgtjdd}t
|
||¡}t ||¡ |d }|jdkrþt||sttdddddgtjdd}t
|
||¡}t ||¡ |d }|jdkrTt||sXttddd d dgtjdd}t
|
||¡}t ||¡ |d }|jdkrªt||s®ttd d dddgtjdd}t
|
||¡}t ||¡ |d }|jdkr t||sttd d dd dgtjdd}t
|
||¡}t ||¡ d S )NrE rF rG r² rN rt r¸ r9 r? rV rÄ g à?rà g ø?g @r r g @g @g @g ð¿rH ) r r r Úsqrtr@ r$ r r rI r( rP rd Údivide)rJ rI r@ r¨ rh rK Úexpr r r Útest_ufunc_coercions1 s@
|
( z$TestUFuncCompat.test_ufunc_coercionsc C s |dddg|dd}|t kr t nt}t |¡}t|ts<ttdddgtjdd}tddd gtjdd}t |d
|
t
|
||¡¡ t |d t
|
||¡¡ d S ) NrE rF rG rt r¸ rH rà rÄ r r ) r r r Úmodfr$ Útupler rI r( rd rP )rJ rI r@ rT rh rK rE rF r r r Ú!test_ufunc_multiple_return_values] s
|
z1TestUFuncCompat.test_ufunc_multiple_return_valuesc C sX t dddgdddgdd}tj |ddgd¡ t dddgdddgdd}t ||¡ d S ) Nr rE rF rG rt r r: é )r r r> Úatr( r) )rJ r r r r r Ú test_ufunc_atj s z TestUFuncCompat.test_ufunc_atN)rk rl rm re r r r r r r r r¯ rI rM rQ rT rW r r r r rH s * rH c @ sj e Zd Zej ddeg¡dd Zej ddeg¡dd Zej de j
|
e j e j e j e jg¡dd ZdS )
|
ÚTestObjectDtypeEquivalencer@ Nc C sv |}t dddg|d}t tjtjtjg|d}t ||¡}t ||¡}tj| }t ||¡ |tj }t ||¡ d S )NrE rF rG r? )r r r¤ r( rP rd ©rJ r@ rQ rh rË r rK r r r Ú test_numarr_with_dtype_add_nant s
|
|
z9TestObjectDtypeEquivalence.test_numarr_with_dtype_add_nanc C sl |}t dddg|d}t dddg|d}t ||¡}t ||¡}d| }t ||¡ |d }t ||¡ d S )NrE rF rG r? r² )r r( rP rd rY r r r Ú test_numarr_with_dtype_add_int s z9TestObjectDtypeEquivalence.test_numarr_with_dtype_add_intr, c C sL t tj d¡t d¡td}|d|}|d| t¡}t | t¡|¡ d S )Nr: r4 rà )
|
r r rX rY r Úobjectr
rÔ r( r) )rJ r, r rK r r r r Ú test_operators_reverse_object s
|
z8TestObjectDtypeEquivalence.test_operators_reverse_object)rk rl rm re r r r\ rZ r[ r r> Úsubr rä rå r] r r r r rX q s
|
|
þrX c @ sè e Zd Zej dejejej ej
|
ej g¡ej de ddde ddde ddde d d
|
d g¡ej d e ddde ddde ddde d d
|
d g¡d d Z ej dejejej ej
|
ej g¡ej de ddde ddde ddde d d
|
d g¡ej dd ddg¡dd Zej de ddde dddg¡ej d e ddde dddg¡dd Zej de ddde dddg¡ej dddg¡dd Zej dejejej ej
|
ej ejejg¡dd Zdd Zej de dd dde dd dfe d d d d e d!d"dfe dd ddee dd djd fe dd#dd$ee dd#djd$ fe dd%e dfe dd&dd'e dddfe d
|
dd(d&e d)d dfe d*d+d'de dfg¡d,d- Zej d.ejejg¡ej d/ddd g¡d0d1 Zd2S )3Ú TestNumericArithmeticUnsortedr, Úidx1r r: rE r= rF iöÿÿÿrN éûÿÿÿr Úidx2c C sL | d¡}| d¡}|||}|t| ¡ t| ¡ }tj||dd d S ©NrJ ÚbarÚequiv©Úexact)Ú_renamer Úto_numpyr( r¦ )rJ r, r` rb rK r r r r Útest_binops_index£ s
|
|
|
|
z/TestNumericArithmeticUnsorted.test_binops_indexr¨ Úscalarc C s0 |||}|t | ¡ |}tj||dd d S ©Nre rf )r ri r( r¦ )rJ r, r¨ rk rK r r r r Útest_binops_index_scalarÆ s
|
z6TestNumericArithmeticUnsorted.test_binops_index_scalarc C sL | d¡}| d¡}t||}tt| ¡ t| ¡ }tj||dd d S rc )rh r! r ri r( r¦ )rJ r` rb rK r r r r Útest_binops_index_powß s
|
|
|
|
z3TestNumericArithmeticUnsorted.test_binops_index_powc C s0 t ||}t t| ¡ |}tj||dd d S rl )r! r ri r( r¦ )rJ r¨ rk rK r r r r Ú test_binops_index_scalar_powê s
|
z:TestNumericArithmeticUnsorted.test_binops_index_scalar_powc C s~ t d}ttj d¡}|t||}|||}t ||¡ t tj dd¡¡}|t ||g¡|}|||}t ||¡ d S )NrN rF )
|
r r r rX rY r( r) rZ rÐ rÑ )rJ r, r% r+ r rK r r r Ú$test_arithmetic_with_frame_or_seriesô s
|
|
zBTestNumericArithmeticUnsorted.test_arithmetic_with_frame_or_seriesc C sJ t ddd}|d }t ddd}tj||dd |d }t ddd}tj||dd |d }t d d
|
d}tj||dd |d }t dd d d ¡}tj||dd |d }t dddd }tj||dd |d }|}tj||dd || }t|j|j }tj||dd t ddd}|d }t|jd }tjt|j|dd d S )Nr r: rF r= r² Trf rU éþÿÿÿr³ rN rE rI éè )r r( r¦ r
r r» Ú_values)rJ r¨ rK r r r r Útest_numeric_compat2 s4 z2TestNumericArithmeticUnsorted.test_numeric_compat2zidx, div, expectedrr iô iÿÿÿi7ÿÿÿéýÿÿÿé! éC éd rÄ é2 r² rG iúÿÿÿrq iÿÿÿi8ÿÿÿc C s t j|| |dd d S )NTrf rô )rJ r¨ ré r r r r Ú test_numeric_compat2_floordiv8 s z;TestNumericArithmeticUnsorted.test_numeric_compat2_floordivr@ Údeltac C s ||}t dddg|d}|| }t |j| |d}t ||¡ || }t |j| |d}t ||¡ t || d| ¡ t || d| ¡ || jrtd S )Nr: é rU r? rF r )r r» r( r¦ Úemptyr )rJ r@ r{ r% rK r r r r Útest_addsub_arithmeticR s z4TestNumericArithmeticUnsorted.test_addsub_arithmeticN) rk rl rm re r r r r> r^ r rå rä r rj rm rn ro r! r rp rt r rs rz r r rI r~ r r r r r_ s¬ ûþ
|
|
|
|
|
üþ
|
|
|
|
üþ ûþ
|
|
|
|
|
üþ ùþ
|
)
|
ý
|
ýïþ
|
r_ c C s~ t dddgdd dgd¡} t jddddgiddd gd
|
}| j|dd }t tjd d d gd tjd tjgd¡}t ||¡ d S )Nr rE rF gñ?)r r0 r g333333ó?gÍÌÌÌÌÌô?rG r ©Z
|
fill_valuerà rH )rZ rÐ Z rfloordivr r¥ r¤ r( rÑ )rÒ r+ rK r r r r Útest_fill_value_inf_maskingf s ÿr c C s t jt d¡tjddddgd tjgd t d¡dtdtdd } t jtj dd¡td
|
td d }t d ¡ | j
|
|d d W 5 Q R X d S )Nr: rE rF rG r² )r r0 r/ r Z
|
abcdefghijr rÕ Z
|
abcdefghjkZABCXr r ) rZ rÐ r r r¤ rÁ rX rY r( Zassert_produces_warningré )Zpdf1Zpdf2r r r Útest_dataframe_div_silencedt s
|
üø
|
ÿ r zdata, expected_datarE rF r² c
|
C s| t |dd}| |}||| }||| }t|| fkr>t}nt|| fkrPt}nt }||dd} t || ¡ t || ¡ d S )NZInt64r? )r r r r( rd )
|
r Z box_1d_arrayrÆ Z expected_datar Ú containerr- r. Úclsr r r r Ú test_integer_array_add_list_like s r c C s t jdtj d¡it jjddgdddggddgd d
|
} | ¡ }|j dd¡|_| | }t jd gd dg| jd }t
|
||¡ d S ) Nr\ é r] r rE rF ZlevAZlevB)Únamesr rH )rÖ r% ) rZ rÐ r rX rY Z
|
MultiIndexZ from_productr r% Z swaplevelr( rÑ )rÒ Zdf2rK r r r r Ú"test_sub_multiindex_swapped_levels¡ s ÿþr ÚpowerrN Ú string_sizec C sX t td| ¡}tj|t jd}d| }||k}t t j|jtd¡}t
|
||¡ d S )Nr: r? ú ) r r rO rZ rÐ r rb r rc r( rÑ )r r r\ r. r- rK r r r r Útest_empty_str_comparison° s r c C sp t dddg} t dddgdd}| | }t dddgdd}t ||¡ | | }t ddd gdd}t ||¡ d S )
|
NrE rF rG ZUInt64r? r
ZFloat64r r )r r( r) )Zseries1Zseries2rK r r r r Ú test_series_add_sub_with_UInt64½ s r )8Ú
|
__future__r Ú collectionsr Údatetimer Údecimalr r Únumpyr re ZpandasrZ r r r r r
|
r Zpandas._testingZ_testingr( Z pandas.corer Zpandas.core.computationr r½ Z pandas.tests.arithmetic.commonr r ZfixtureZto_arrayr r r0 Z_ldtypesr< Ú__annotations__ÚextendrD rn r¡ ræ r rH rX r_ r r r r r r r r r r r r Ú<module> s^
|
7 B b ^R/ Gþ
|
|
