U
|
ý°dä ã @ s> d dl mZ d dlmZmZ d dlmZ d dlZd dlZd dlZd dl Z
|
d dl Z d dl m mZ d dlZd dlmZmZmZmZ d dlmZ d dlm mZ d dl m Z d dl m!Z!m"Z" d dl#m$Z$m%Z% e j&d d d
|
gd d gd dd Z'G dd dZ(G dd dZ)G dd dZ*G dd dZ+G dd dZ,dd Z-d d Z.d d Z/G d d! d!Z0d"d# Z1d$d% Z2d&d' Z3d(d) Z4e j5 6d*ed+d,gd-d.gge7d/d0d1fed+d2gd-d2gge7d/d0d3fg¡d4d5 Z8d6d7 Z9e j5 6d8ed9d9gged9d9ged9d9ggg¡d:d; Z:d<d= Z;d>d? Z<d@dA Z=dBdC Z>dDdE Z?dFdG Z@dS )Hé )Údeque)ÚdatetimeÚtimezone)ÚEnumN)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeries)Ú expressions)Ú _MIN_ELEMENTSÚNUMEXPR_INSTALLED)Ú_check_mixed_floatÚ_check_mixed_intTi@B ZnumexprÚpython)ZautouseÚparamsÚidsc c s t j}| jt _| jV |t _d S ©N)Úexprr Úparam)Úrequestr © r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_arithmetic.pyÚswitch_numexpr_min_elements$ s r c @ sZ e Zd ZddddZdd ZedddZedd d
|
Zdd d Zdd Z dddZ
|
dS )Ú DummyElementN©Úreturnc C s || _ t |¡| _d S r )ÚvalueÚnpÚdtype)Úselfr r r r r Ú__init__- s zDummyElement.__init__c C s t j| j| jdS )N©r )r Úarrayr r ©r r r r Ú __array__1 s zDummyElement.__array__c C s d| j d| j dS )Nz DummyElement(z, ú))r r r# r r r Ú__str__4 s zDummyElement.__str__c C s t | S r )Ústrr# r r r Ú__repr__7 s zDummyElement.__repr__Fc C s
|
|| _ | S r r! )r r Úcopyr r r Úastype: s zDummyElement.astypec C s t | | j |¡|S r )Útyper Úview)r r r r r r, > s zDummyElement.viewc C s
|
t | jS r )Úboolr )r Úaxisr r r ÚanyA s zDummyElement.any)F)N) Ú__name__Ú
|
__module__Ú __qualname__r r$ r' r& r( r* r, r/ r r r r r , s
|
r c @ sn e Zd Zdd Zdd Zej dej j
|
ddde j ddd d
|
ej j
|
dddej j
|
dddd
|
gej j
|
dddej j
|
dddd
|
ej j
|
ddde j ddd d
|
ge j ddd e j ddd d
|
ej j
|
dddej j
|
dddd
|
gej j
|
ddde j ddd d
|
e j ddd e j ddd d
|
gg¡d d Z ej d ddddddg¡dd Zdd Zdd Zdd Zd d Zd S ) ÚTestFrameComparisonsc C sb t ddddgi}t ddddgi}|dk}t ||¡ |d d¡|d< |dk}t ||¡ d S )NÚAÚfooÚbarÚbazTFÚcategory)r ÚtmÚassert_frame_equalr* )r ÚdfÚexpÚresr r r Ú&test_comparison_with_categorical_dtypeL s z;TestFrameComparisons.test_comparison_with_categorical_dtypec C sD t tj dd¡tdd}d}tjt|d |dk W 5 Q R X d S )Né é ÚABCD©Úcolumnsz+The truth value of a DataFrame is ambiguous©Úmatchr )r r ÚrandomÚrandnÚlistÚpytestÚraisesÚ
|
ValueError)r r; Úmsgr r r Útest_frame_in_list[ s z'TestFrameComparisons.test_frame_in_listz arg, arg2é
|
©ÚsizeÚ20010101©Úperiods©ÚaÚbc s t | t | k}t fdd jD j jd}t ||¡ k}t fdd jD j jd}t ||¡ dddg}d |¡}tjt|d k W 5 Q R X tjt|d k W 5 Q R X tjt|d k W 5 Q R X tjt|d k W 5 Q R X d S )
|
Nc s i | ]}| | | kqS r r ©Ú.0Úcol©ÚxÚyr r Ú
|
<dictcomp> s z@TestFrameComparisons.test_comparison_invalid.<locals>.<dictcomp>©ÚindexrC c s i | ]}| | | kqS r r rW rZ r r r] s z=Invalid comparison between dtype=datetime64\[ns\] and ndarrayzinvalid type promotionzbThe DTypes <class 'numpy.dtype\[.*\]'> and <class 'numpy.dtype\[.*\]'> do not have a common DType.ú|rD ) r rC r_ r9 r: ÚjoinrI rJ Ú TypeError)r ÚargZarg2ÚresultÚexpectedZmsgsrL r rZ r Útest_comparison_invalidb s: 0ý ý û
|
z,TestFrameComparisons.test_comparison_invalidz left, right)ÚgtÚlt)rh rg )ÚgeÚle)rj ri )Úeqrk )Únerl c C s t tjdddtjdddtjjdddtj d¡tt d¡d}tj
|
|j tj t |¡dkd f< tt|}tt|}|d
|
kr°||t d ¡}|t d ¡|}t ||¡ nXd }tjt|d ||t d ¡ W 5 Q R X tjt|d |t d ¡| W 5 Q R X |d
|
kr@||t d¡}|t d¡|}t ||¡ nXd}tjt|d ||t d¡ W 5 Q R X tjt|d |t d¡| W 5 Q R X d S )NrQ rN rR Z20010102i Ê;rO )Zdates1Údates2ZintcolZfloatcolZ stringcolg à?rm )rk rl Z20010109zL'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'Timestamp'rD ZnatzJ'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'NaTType')r ÚpdÚ
|
date_ranger rF ÚrandintrG rH r9 ZrandsÚNaTÚlocÚrandÚlenÚgetattrÚoperatorÚ Timestampr: rI rJ rb ) r ÚleftÚrightr; Zleft_fZright_fre rd rL r r r Útest_timestamp_compare¹ s>
|
ûÿ
|
|
ÿ
|
ÿz+TestFrameComparisons.test_timestamp_comparec C s\ t ddgddgg}t ddgddgg}||k}| ¡ ¡ r@t||k}| ¡ ¡ sXtd S )Nz
|
1989-08-01é é rU rV ÚcÚd)r r/ ÚAssertionErrorÚall)r r; Úotherrd r r r Útest_mixed_comparisonñ s z*TestFrameComparisons.test_mixed_comparisonc C s^ t t d¡ d¡}t ddgddgddgg}|dk}t ||¡ |ddgk}t ||¡ d S )Nr? ©é r| FT©r| r| r| )r r ÚarangeÚreshaper9 r: )r r; re rd r r r Ú test_df_boolean_comparison_errorþ s z5TestFrameComparisons.test_df_boolean_comparison_errorc C s@ t tj dd¡tddddgd}| d ¡}| ¡ ¡ r<td S )Né r r4 ÚBÚCr^ )r r rF rG ÚrangeÚ__eq__r/ r ©r r; rd r r r Ú test_df_float_none_comparison s "
|
z2TestFrameComparisons.test_df_float_none_comparisonc C s¼ t ddddddg}|jdk}t || |jdd
d d
f ¡ t || |jdd
d d
f ¡ |jdk}t || |jdd
d d
f ¡ t || |jdd
d d
f ¡ d S )Nr{ r5 rT r| r6 r )r rU r9 r: rr rV )r r; Zmask_aZmask_br r r Útest_df_string_comparison s
|
"$
|
"z.TestFrameComparisons.test_df_string_comparisonN)r0 r1 r2 r> rM rI ÚmarkÚ parametrizer rF rp rn ro rf rz r r r r r r r r r3 I sd þþû þ þû þþû þ þûáþ-
|
*úþ
|
- r3 c @ s¶ e Zd Zej dddddddg¡dd Zej d
|
eje g¡d d Z
|
d d Z dd Z dd Z dd Zej dddddddg¡dd Zej dddddddg¡dd Zdd Zd S ) ÚTestFrameFlexComparisonsÚoprk rl rg rh ri rj c C sH t j dd¡}t j dd¡}t|}t|}t |jd ¡}| |¡j ¡ sPt |
|
|¡j ¡ rdt t ||}t t |}t |||||¡ |jdd
dd
f ¡ } || }
|
||| j|j|jd} t |
|
| ¡ t ||j|||j¡ t |d||d¡ d} t |t j||t j¡ tjt| d || W 5 Q R X d S ) Né r )r{ r r{ r^ r ú$Unable to coerce to Series/DataFramerD )r rF rG r ÚonesÚshaperk Úvaluesr r rl r/ ru rv r9 r: rr r) Úreindexr_ rC ÚnanrI rJ rK ) r r ÚdataZ
|
other_datar; r Úndim_5ÚfÚoZpart_oÚrsZxprL r r r Útest_bool_flex_frame s(
|
|
z-TestFrameFlexComparisons.test_bool_flex_frameÚboxc C s t j dd¡}t|}|t j d¡}|t j d¡}|j|dd}| |¡}|j|dd}| |¡} t ||t|k¡ t || ¡ t || ¡ t ||j |¡j ¡ t || t
|
|¡¡ t ||jt|dd¡ t ||jt
|
|dd¡ |j |dd}
|
| |¡} |j |dd} | |¡} t | |t|k¡ t | | ¡ t |
|
| ¡ t |
|
|j |¡j ¡ |j |dd}| |¡}|j|dd}| |¡}t ||t|k¡ t || ¡ t || ¡ t ||j |¡j ¡ tt j d¡}tt j d¡}d S )Nr r r ©r. )r rF rG r rk rl r9 r: r ÚTrH rg rj ri rh )r r¢ r r; Zidx_serZcol_serZidx_eqZcol_eqZidx_neZcol_neZidx_gtZcol_gtZidx_leZcol_leZidx_geZcol_geZidx_ltZcol_ltr r r Útest_bool_flex_series= sB
|
|
|
|
|
|
z.TestFrameFlexComparisons.test_bool_flex_seriesc C s² t tj dd¡}tj|jd< | |¡}|jd r6t| |¡}|jd sNt| |¡}|jd rft|
|
|¡}|jd r~t| |¡}|jd rt| |¡}|jd r®td S )Nr r ©r r ) r r rF rG r rr rk r rl rg rh ri rj )r r; r r r r Útest_bool_flex_frame_nah s
|
|
|
|
|
|
z0TestFrameFlexComparisons.test_bool_flex_frame_nac C sp t t jddt jg¡}t dt jdd g¡}td|i}td|i}d ddg¡}tjt|d | |¡ W 5 Q R X tjt|d |d |d ¡ W 5 Q R X tjt|d |j |j k W 5 Q R X |
|
|¡}|j ¡ sÞt t dt jd g¡}td|i}tjt|d | d¡ W 5 Q R X tjt|d |d d¡ W 5 Q R X tjt|d |j dk W 5 Q R X d S )
|
Nr{ r? y @é rU r` z9'>' not supported between instances of '.*' and 'complex'z unorderable types: .*complex\(\)rD ) r r" r r ra rI rJ rb rg r rl r r ) r ÚarrZarr2r; Údf2rL r Zarr3Údf3r r r Ú"test_bool_flex_frame_complex_dtypey s2 þÿ
|
z;TestFrameFlexComparisons.test_bool_flex_frame_complex_dtypec C sV t ddtjdgi}t ddt ¡ dgi}| |¡}t ddddgi}t ||¡ d S )NrY r5 r6 FT)r r r r Únowrl r9 r: )r Údf1rª rd r< r r r Ú!test_bool_flex_frame_object_dtype s
|
|
z:TestFrameFlexComparisons.test_bool_flex_frame_object_dtypec C s t tjg}|tjk}|jd ¡ dks,t| tj¡}|jd ¡ dksNt|tjk}|jd ¡ dksnt| tj¡}|jd ¡ dkstd S )Nr¦ FT)r rn rq ÚilocÚitemr rk rl r r r r Útest_flex_comparison_nat¨ s
|
|
z1TestFrameFlexComparisons.test_flex_comparison_natÚopnamec C sV t dddgdddgd}d}t|||j ¡ }t |tdgt t ¡gdd ¡ d S )
|
Nr{ r| r ç ð?ç @ç @rZ Úcount©r_ Úname)
|
r ru ÚdtypesÚ value_countsr9 Úassert_series_equalr r r r- )r r³ r; Úconstrd r r r Ú&test_df_flex_cmp_constant_return_typesº s ÿz?TestFrameFlexComparisons.test_df_flex_cmp_constant_return_typesc C sd t dddgdddgd}d}|jd d
}t|||j ¡ }t |tdgt t
|
¡gd d
|
¡ d S ) Nr{ r| r r´ rµ r¶ rZ r r· r¸ ) r r° ru rº r» r9 r¼ r r r r- )r r³ r; r½ Úemptyrd r r r Ú,test_df_flex_cmp_constant_return_types_emptyÅ s ÿzETestFrameFlexComparisons.test_df_flex_cmp_constant_return_types_emptyc C s t j dddg¡}t||d}tddg}|j|dd}tddgddgd}t ||¡ tddgdd gd
|
}|j|dd}t ||¡ d S ) Nr{ r| r ©r4 r r r£ Fr4 r ©r_ )rn Z IntervalIndexZ from_breaksr r rk r9 r: )r Úiir; Úserr= re Zser2Úres2r r r Ú-test_df_flex_cmp_ea_dtype_with_ndarray_seriesÑ s zFTestFrameFlexComparisons.test_df_flex_cmp_ea_dtype_with_ndarray_seriesN)r0 r1 r2 rI r r r¡ r r" r r¥ r§ r¬ r¯ r² r¾ rÀ rÆ r r r r r s
|
|
*'
|
|
|
r c @ sH e Zd Zdd Zejje ddej dddg¡dd Z d
|
d Z
|
d d Z dd Z ej ddddg¡dd Z ej dedd¡dd Zdd Zej dd d d d g¡d!d" Zd#d$ Zej d%d&d'g¡d(d) Zd*d+ Zd,d- Zd.d/ Zej ddddd0g¡d1d2 Zej d3d4d5g¡d6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Z d5S )@ÚTestFrameFlexArithmeticc st t d¡}t|td jdd}t fdd jD }t ||¡ jjdd}t ||¡ d S )Nr rÁ r r£ c s i | ]}| | qS r r rW ©r; rÄ r r r] í s z?TestFrameFlexArithmetic.test_floordiv_axis0.<locals>.<dictcomp>) r r r r ÚfloordivrC r9 r: r )r r© rd re Úresult2r rÈ r Útest_floordiv_axis0å s
|
z+TestFrameFlexArithmetic.test_floordiv_axis0znumexpr not installed)Úreasonr³ rÉ Úpowc s¢ t t|t td ¡ td d d¡d }t| d d< d t |dd}t fdd jD }t ||¡ t |j
|
dd}t ||¡ d S )
|
Néd r{ éÿÿÿÿr´ r r r£ c s i | ]}| | qS r r rW ©r; r rÄ r r r] s zLTestFrameFlexArithmetic.test_floordiv_axis0_numexpr_path.<locals>.<dictcomp>) ru rv r r r r r rC r9 r: r )r r³ r© rd re rÊ r rÐ r Ú test_floordiv_axis0_numexpr_pathô s
|
" z8TestFrameFlexArithmetic.test_floordiv_axis0_numexpr_pathc C sf t jddd}t jddd}t|}t||d}|j|dd}t|| || d}t ||¡ d S )Nú
|
2016-01-01rN rR Ú1©r r{ r r£ )rn ro Ztimedelta_ranger r Úaddr9 r: )r ÚdtiZtdiZtserr; rd re r r r Útest_df_add_td64_columnwise s z3TestFrameFlexArithmetic.test_df_add_td64_columnwisec C s t jddd}tdddgdd}t||d }t||d }t jd
|
d ¡ }|j||d }ttd ddgdd|d d }t ||¡ d S )NrÒ r rR z1 Dayrq z2 Daysútimedelta64[ns]r! rÁ r{ )Údays©Ú
|
fill_valuez
|
2016-01-02z
|
2016-01-03z
|
2016-01-05zdatetime64[ns]r| ) rn ro r r Ú TimedeltaZto_timedelta64rÕ r9 r: )r rÖ rÄ r; r Úfillrd re r r r Ú$test_df_add_flex_filled_mixed_dtypes s ÿüÿz<TestFrameFlexArithmetic.test_df_add_flex_filled_mixed_dtypesc s| | fdd}t | d| }||d| }t ||¡ t | d| }||d| }t ||¡ t|dd id d S )Nc s2 d¡r"tt dd¡|| S tt | |S )NZ__rÚ__)Ú
|
startswithru rv ÚreplacerZ ©r r r r + s
|
z8TestFrameFlexArithmetic.test_arith_flex_frame.<locals>.fr| r r! )ru r9 r: r )r Úall_arithmetic_operatorsÚ float_frameÚmixed_float_framer rd re r râ r Útest_arith_flex_frame% s z-TestFrameFlexArithmetic.test_arith_flex_framer Ú__add__Ú__sub__Ú__mul__c
|
C sö t t|}t ||d| }||d| }d } |dkrBdd d} n|dkrRdd i} tjrr|dkrrd| d j| d< t ||¡ t|| d t ||d| }||d| }t ||¡ t|dd id t ||d| }||d| }t ||¡ d S )
|
Nr| )rè Zuint64)r r )rç ré r r r4 r! ) ru rv r Ú USE_NUMEXPRr r9 r: r r )
|
r r Z int_frameÚmixed_int_framerå r r rd re r r r r Útest_arith_flex_frame_mixed; s&
|
z3TestFrameFlexArithmetic.test_arith_flex_frame_mixedÚdimr r? c C sB |}t d| ¡}d}tjt|d t||| W 5 Q R X d S )N)r{ r rD )r r rI rJ rK ru )r rã rä rí r r© rL r r r Útest_arith_flex_frame_raised s
|
z3TestFrameFlexArithmetic.test_arith_flex_frame_raisec C sÄ | d¡}t ||d ¡ | |d d
¡}t ||tj ¡ |d d
|¡}t ||tj ¡ tjtdd |j |jd dd W 5 Q R X tjtdd |j |jd ddd W 5 Q R X d S ) Nr{ r rÛ rD r rÚ r_ ©r. rÛ ) rÕ r9 r: r r rI rJ ÚNotImplementedErrorr° )r rä Z const_addrd r r r Ú test_arith_flex_frame_cornero s
|
z4TestFrameFlexArithmetic.test_arith_flex_frame_cornerrÕ ÚsubÚmulÚmodc C sb |}| d¡}|d }t||}tt|}t |||||¡ t ||dd||j|j¡ d S )NrU Útwor r£ )Úxsru rv r9 r: r¤ )r Ú simple_framer r; ÚrowrY r r r r Útest_arith_flex_series_ops s
|
|
|
z2TestFrameFlexArithmetic.test_arith_flex_series_opsc C sh |}| d¡}|d }t |j|d d|| ¡ t | |¡|| ¡ t |j|dd|j| j¡ d S )NrU rõ r£ r )rö r9 r: rÕ Údivr¤ )r r÷ r; rø rY r r r Útest_arith_flex_series s
|
z.TestFrameFlexArithmetic.test_arith_flex_seriesr Úint64Úfloat64c C sX t t d¡ d¡|d}t tjtjgddgddgg}|j|d dd }t ||¡ d S )
|
Nr? r r! r´ ç ø?g ô?r r_ r£ ) r r r r r Úinfrú r9 r: )r r r; re rd r r r Ú#test_arith_flex_series_broadcasting s z;TestFrameFlexArithmetic.test_arith_flex_series_broadcastingc C s t g td}tddgd}tddgddggddgd}tjtd d
|
|j|d d W 5 Q R X tjtd d
|
|j|d d dd W 5 Q R X d S )Nr! r4 r rB r{ r| r r@ rÛ rD ÚErÚ rï )r Úobjectr rI rJ rð rÕ rò )r Zser_len0Zdf_len0r; r r r Ú test_arith_flex_zero_len_raises s z7TestFrameFlexArithmetic.test_arith_flex_zero_len_raisesc C s^ t jddt jdddgdd}td|itd d
|
}| d¡ d ¡}|jd dd }t ||¡ d S ) Nr r{ r r@ r Úfloatr! r5 r? r r| rÚ ) r r" r r r ÚfillnarÕ r9 r: )r Zdatr; r< r= r r r Ú test_flex_add_scalar_fill_value s
|
z7TestFrameFlexArithmetic.test_flex_add_scalar_fill_valuec C sr t dddddgdddddgd}t dddgdddgd}t dddddgdddddgd}| |¡}t ||¡ d S )Nr{ r| r r@ r r r )r rò r9 r: )r r® rª re rd r r r Ú'test_sub_alignment_with_duplicate_index¶ s
|
|
z?TestFrameFlexArithmetic.test_sub_alignment_with_duplicate_indexÚ __truediv__c C sj t t d¡tj d¡d}t|||}ddg|_ddg|_t|||}t ||¡ t | |j
|
d S )NrN rÁ r4 ) r r r rF rs ru rC r9 r: r' rº )r r r; re rd r r r Ú&test_arithmetic_with_duplicate_columns¾ s
|
|
z>TestFrameFlexArithmetic.test_arithmetic_with_duplicate_columnsÚlevelr Nc C s t dddgdddgd}|j d¡|_t dddgdddgd}|j ddg¡|_|j||d }t dddgdddgd}|j ddg¡|_t ||¡ d S )
|
Nr r{ r| r rÁ ZL1))r4 r )r4 ÚDZL2)r
|
)r rC Z set_namesrÕ r9 r: )r r
|
r® rª rd re r r r Útest_broadcast_multiindexÊ s z1TestFrameFlexArithmetic.test_broadcast_multiindexc C s¾ t dddgdddgdtjdgdgd ddggd
|
d d gd d}tdgtjdgdggd d
|
gd d}t dddgdddgdtjdgdgd ddggd
|
d d gd d}|j|d d}t ||¡ d S )Nr{ r| r r@ r ©iÚ iä rU rV r Úscenrô Úid©Únamesr çÙ?çffffffö?ç333333@ç333333 @ç@g@r£ ©r r Ú from_productr rÕ r9 r: ©r r; Úseriesre rd r r r Ú test_frame_multiindex_operationsÙ s( ÿþþ ÿþz8TestFrameFlexArithmetic.test_frame_multiindex_operationsc C sº t dgdgdtjdgdggddgdd }td
|
d d gtjdgdgd ddggdddgdd }t dddgdddgdtjdgdgd ddggdddgdd }|j|d d}t ||¡ d S )Nr{ r r rU rV r rô r r ç $@g 4@g >@r r| r g &@é g ?@g *@g 7@g @@r£ r r r r r Ú<test_frame_multiindex_operations_series_index_to_frame_indexñ s( þ ÿþ ÿþzTTestFrameFlexArithmetic.test_frame_multiindex_operations_series_index_to_frame_indexc C s¶ t dddgdddgdtjdgdgd ddggd
|
d d gd d}tdgtjdgdggd d
|
gd d}t tjtjdtjdddddtjfgd
|
d d gd d}|j|d d}t ||¡ d S )Nr{ r| r r@ r r rU rV r r rô r r r r r} ©rU rV r ©rU rV r{ )rU rV r| r£ )
|
r r r r r r Ú from_tuplesrÕ r9 r: r r r r Ú)test_frame_multiindex_operations_no_align s0 ÿþþ
|
üùþ zATestFrameFlexArithmetic.test_frame_multiindex_operations_no_alignc C s® t dddgdddgdtjddd gd
|
d d gd d}tdgtjdgdggd d
|
gd d}t ddtjgddtjgdtjddd gd
|
d d gd d}|j|dd}t ||¡ d S )Nr{ r| r r@ r r r r )rU r} r| r rô r r r r rV rU r r r r r r£ )
|
r r r! r r r r rÕ r9 r: r r r r Ú+test_frame_multiindex_operations_part_align& s4 ýúþ þýúþ zCTestFrameFlexArithmetic.test_frame_multiindex_operations_part_align) r0 r1 r2 rË rI r Zskipifr 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Ç ä s8
|
(
|
|
|
|
|
|
rÇ c @ s. e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j dddg¡dd Z e
|
j dddgde ddg¡eddeddgg¡dd Zdd Zd d Zd d Ze
|
j d!d"d#d$d%d&d'e d(d)¡d*fe d(d)¡d+fg¡e
|
j j d,ejejejej ej ej gd-d. d/d0d1 Z d2d3 Z d4d5 Z!d6S )7ÚTestFrameArithmeticc C sJ t ddgdd}tddgddgg}|| }t||d}t ||¡ d S ) Nrq rØ r! r{ r| r r@ rÔ ©r r r9 r: )r rÄ r; rd re r r r Útest_td64_op_nat_castingI s
|
z,TestFrameArithmetic.test_td64_op_nat_castingc C s¨ t d¡ dd¡}t|ddgdddgd }|d
|
gd d
f }|jd
|
|jd
|
fksTttdd gd dgdd gg|j|j|jd }|| }t
|
||¡ || }t
|
||¡ d S )Nr? r r| TFr4 r r ©rC r_ r{ r@ r ©rC r_ r © r r r r r r rC r_ r r9 r: )r r© r; Úrowlikere rd r r r Ú'test_df_add_2d_array_rowlike_broadcastsS s ú z;TestFrameArithmetic.test_df_add_2d_array_rowlike_broadcastsc C s¨ t d¡ dd¡}t|ddgdddgd }|d d
d
|
gf }|j|jd d
|
fksTttd
|
dgd dgd dgg|j|j|jd}|| }t
|
||¡ || }t
|
||¡ d S )Nr? r r| TFr4 r r r' r{ r r é rN r( r) )r r© r; Úcollikere rd r r r Ú'test_df_add_2d_array_collike_broadcastsh s ú z;TestFrameArithmetic.test_df_add_2d_array_collike_broadcastsc C sä |}|r|dkrt |¡ t d¡ dd¡}t|ddgddd gd
|
}|d gd d
f }|jd |jd fksntt|j d ||
|
¡ t|j d ||
|
¡ t|j d ||
|
¡ g}t||j |j d
|
} t|||}
|
t |
|
| ¡ d S ) N©Ú__rmod__Ú __rfloordiv__r? r r| TFr4 r r r' r{ )ÚtdÚ&mark_array_manager_not_yet_implementedr r r r r r ru rr ÚsqueezerC r_ r9 r: ) r r rã Úusing_array_managerr³ r© r; r* Úexvalsre rd r r r Ú)test_df_arith_2d_array_rowlike_broadcasts} s
|
ýz=TestFrameArithmetic.test_df_arith_2d_array_rowlike_broadcastsc C sò |}|r|dkrt |¡ t d¡ dd¡}t|ddgddd gd
|
}|d d
d gf }|j|jd d fksntt|d || ¡ t|d || ¡ d }d } |dkrÀtj
|
dd | ¡ D } t||j |j | d}
|
t|||} t | |
|
¡ d S )Nr/ r? r r| TFr4 r r r' r{ r )TFc s s | ] }|j V qd S r )r ©rX r[ r r r Ú <genexpr>± s zPTestFrameArithmetic.test_df_arith_2d_array_collike_broadcasts.<locals>.<genexpr>r( )r2 r3 r r r r r r ru r4 Z common_typer rC r_ r9 r: ) r r rã r5 r³ r© r; r- r6 r re rd r r r Ú)test_df_arith_2d_array_collike_broadcasts s
|
þz=TestFrameArithmetic.test_df_arith_2d_array_collike_broadcastsc C sh t ddgddgg}|d }|j dd ¡}|dk ¡ s<td| }|j dd ¡}|dk ¡ sdtd S )NFTr{ c S s | j S r ©Úkind©r[ r r r Ú<lambda>À ó z:TestFrameArithmetic.test_df_bool_mul_int.<locals>.<lambda>Úic S s | j S r r; r= r r r r> Ä r? )r rº Úapplyr r )r r; rd Úkindsr r r Útest_df_bool_mul_int¸ s z(TestFrameArithmetic.test_df_bool_mul_intc C sL t dddgdddgd}|| }t dd d
|
gdd d gd}t ||¡ d S ) NrU rV r} r{ r| r rÁ ZaaZbbÚccr@ r? ©r r9 r: )r rx rd re r r r Útest_arith_mixedÇ s z$TestFrameArithmetic.test_arith_mixedrY r4 r c C sB t ddgddgd}||d| }||| d}t ||¡ d S )Ngñ?gffffff
|
@g @g333333ÀrÁ r{ )r r9 r¼ )r Zall_arithmetic_functionsrY r; rd re r r r Útest_arith_getitem_commuteÎ s z.TestFrameArithmetic.test_arith_getitem_commuter r{ r| ©r{ r| r c C sD t ddgddgd}t ddgddgd}|| }t ||¡ d S )Nr{ rÁ r| r rE )r r r; re rd r r r Ú&test_arith_alignment_non_pandas_objectÕ s z:TestFrameArithmetic.test_arith_alignment_non_pandas_objectc C s6 t tjdddd dd¡dddgd d
|
d gd }| d ¡j}t |j| |j|jd }t || |¡ t |jj
|
| j
|
|j|jd }t |j |dd|¡ t |d }t |j| |j|jd }t || |¡ t |jj
|
| j
|
|j|jd }t |j |dd|¡ tj j|j }t |j| |j|jd }t | |¡|¡ d S )Nr{ rN Úf8r! r Úonerõ ÚthreerU rV r} r' r^ r r£ r_ )r r r r rö r r_ rC r9 r: r¤ rÕ rH rF rs r )r r; Zval1ÚaddedZval2Zval3r r r Ú test_arith_non_pandas_objectß s$ ý z0TestFrameArithmetic.test_arith_non_pandas_objectc sn |t t jddd¡}ddg}t|g|d}d t| }t fdd |D g|d}t ||¡ d S )
|
Nç rµ )ÚstartÚendr{ r| rB rN c s g | ]}t | qS r )ru )rX Ún©Únumr r r Ú
|
<listcomp> s zVTestFrameArithmetic.test_operations_with_interval_categories_index.<locals>.<listcomp>)rn ZCategoricalIndexZinterval_ranger ru r9 r: )r rã Úindr r; rd re r rS r Ú.test_operations_with_interval_categories_indexù s zBTestFrameArithmetic.test_operations_with_interval_categories_indexc C s t t d¡t d¡gt d¡t d¡gdddgd}|dg }|| }t t d ¡t d ¡gtjtjgdddgd}t ||¡ d S )
|
NZ2019Z2020Z2018Z2021)r5 r6 r5 r6 rB r )r rn rw rÜ r r r9 r: )r r; rª rd re r r r Ú test_frame_with_frame_reindex s þû
|
þz1TestFrameArithmetic.test_frame_with_frame_reindexz value, dtype)r{ Úi8)r´ rJ )ì rJ )y ð?Ú
|
complex128)rZ r[ )Tr- é Únsú<m8[ns]ú<M8[ns]r c C s | j S r )r0 r= r r r r> . r? z TestFrameArithmetic.<lambda>)r c
|
C s( t jdft jdft jdft jdfh}t||}td|j|jgi|jd}t jdft j dft jdft jdft jdft jdft jdft j
|
dft j dfh } ||f| kr\d }
|
|dkr¸|t jksÊ|dkrÐ|t jkrÐd } nN|dkrÞd} n@|t j
|
krd} |dkr t j r |d kr t }
|
nd
|
|j d } tjt| d ( t |
|
¡ |||j W 5 Q R X W 5 Q R X nÈ||f|krè|t jt jfkr¼t j r|d krt }
|
nd }
|
t |
|
¡ |||j W 5 Q R X n*d } tjt| d |||j W 5 Q R X n<t d ¡ |||jj} |||j} W 5 Q R X t | | ¡ d S )Nr- r4 r! r_ r^ r[ z3ufunc 'remainder' not supported for the input typesz-numpy boolean subtract, the `-` operator, is r zcannot perform __z7__ with this index type: (DatetimeArray|TimedeltaArray)rD z+operator '.*' not implemented for .* dtypes)rv ÚtruedivrÍ rÕ ró r r r r rô rò r rê Ú UserWarningr0 rI rJ rb r9 Úassert_produces_warningrð rº r¼ )r r r r r r ÚskipÚelemr; ÚinvalidÚwarnrL rd re r r r Útest_binop_other sn ü
|
÷ ÿÿ ÿþý ÿ " z$TestFrameArithmetic.test_binop_otherc C s t tddgtddgg¡}t tddgddtddgg¡}tddgddgg|d}tddgddgg|d}|| }tddgddgg|d}t ||¡ d S ) Nr{ r| r r@ ÚInt8r! rB r ©r Z from_arraysr r r9 r: ©r ÚmidxZmidx2rx ry rd re r r r Ú*test_arithmetic_midx_cols_different_dtypesv s "z>TestFrameArithmetic.test_arithmetic_midx_cols_different_dtypesc C s t tddgtddgg¡}t tddgddtddgg¡}tddgddgg|d}tddgddgg|d}|| }tddgddgg|d}t ||¡ d S ) Nr{ r| r r@ rh r! rB rÏ ri rj r r r Ú:test_arithmetic_midx_cols_different_dtypes_different_order s "zNTestFrameArithmetic.test_arithmetic_midx_cols_different_dtypes_different_orderN)"r0 r1 r2 r& r+ r. r7 r: rC rF rI r r rG r r" r r rI rN rW rX Z timedelta64Z
|
datetime64rv rÕ rò ró r` rô rÍ rg rl rm r r r r r$ H sV
|
|
(ÿ
|
øþ úö F
|
r$ c C s´ t tj d¡ dd¡ddgd} ttjd}| | }t | jtj | j d}t
|
||¡ t j tdd | |k W 5 Q R X t | j d
|
¡| j d}t j tdd ||k W 5 Q R X d S ) Nr? r r| r4 r rB r! z not alignedrD zM8[ns])r r rF rG r r rý r r rC r9 r: rI rJ rK r, )r; rÄ rd re rª r r r Ú,test_frame_with_zero_len_series_corner_cases s rn c C sB t ddgtjd} tddgddgd}| | }| }t ||¡ d S )Nr4 r ©rC r r{ r| r )r r rý r r9 r: )r; rÄ rd re r r r Ú,test_zero_len_frame_with_series_corner_cases s
|
rp c C sF dt ddtjgi} t| }|jdd}t dddg}t ||¡ d S )NZOner4 g333333ó?r{ r£ r )r r r r Úsumr9 r¼ )r r; rd re r r r Ú+test_frame_single_columns_object_sum_axis_1ª s ÿ rr c @ sÈ e Zd Zdd Zdd Zej dej ej
|
ej ej g¡dd Z ej dd d
|
g¡ej d ¡d d Zej ddddddg¡dd Zdd Zdd Zdd Zdd Zd d Zd d Zej d!ejejejejejej g¡d"d# Z d$d% Z d&d' Z d(d) Z d*d+ Z!d,d- Z"ej ddd.dd/d0dd1d2ddd3g ¡d4d5 Z#ej d6d7d8d9gd:e$j%d7d8d9ge$j&d;e'd7d<g¡d=d> Z(ej d6d7d8gd?e$ %d7d8g¡e'd7d9g¡d@dA Z)dBdC Z*dDdE Z+dFdG Z,dHdI Z-dJS )KÚTestFrameArithmeticUnsortedc C sp t jddddd}ttj t|¡|dgd}| d¡}|| }|jj t
|
j ksRt || }|jj t
|
j kslt d S ) Nz1/1/2011rN ÚHú
|
US/Eastern)rS ÚfreqÚtzrU r^ z Europe/Moscow) rn ro r r rF rG rt Ú
|
tz_convertr_ rw r Úutcr )r Úrngr; Z df_moscowrd r r r Ú*test_frame_add_tz_mismatch_converts_to_utc¼ s
|
zFTestFrameArithmeticUnsorted.test_frame_add_tz_mismatch_converts_to_utcc C s t jdddd}ttj t|d¡|d}||d d d
}|| }tj|jdd d
< t
|
||¡ |d d d
}|| tj t|¡¡ }t
|
||¡ d S ) Nz1/1/2000z1/1/2010r4 )rv r r r| r{ ) rn Z period_ranger r rF rG rt r r° r9 r: ZtakeZ permutation)r rz Útsrd re Zhalfr r r Útest_align_frameÇ s z,TestFrameArithmeticUnsorted.test_align_framer c C sÖ t dddd gddddgdtd}| tj¡}||d}||d t¡}d |t |¡< t ||¡ |||}||| t¡}d |t |¡< t ||¡ ||| d ¡}t ||¡ || d ¡|}tj ||d
|
d d S ) Nr| ç @é{ r{ r r@ )Úcol1Zcol2r! r¨ F)Z check_dtype)
|
r r r r r r* ÚcomZisnar9 r: )r r r; Zfilledrd re r r r Útest_operators_none_as_naÔ s" ÿ
|
|
z5TestFrameArithmeticUnsorted.test_operators_none_as_nazop,res)r F)Ú__ne__Tz ignore:elementwise:FutureWarningc C s* t ||d}t| ¡ ¡ |ks&td S )Nr5 )ru r- r r )r r r= rä rd r r r Ú%test_logical_typeerror_with_non_validï s zATestFrameArithmeticUnsorted.test_logical_typeerror_with_non_validrÕ rò ró rú r` c s( t jtddddgdddggdd d
|
gd }tt d ¡ d d¡|dddgd ¡ tj t
|
t |d d krnd S t dddgdddg}t
|
||d
|
dd}t fdd| ¡ D ¡ ¡ }t ||¡ t ddgddg}t
|
||d dd}t fdd| ¡ D ¡ ¡ ¡ }t ||¡ d S )NÚabcrK rõ rL r{ r| r ÚfirstÚsecondÚthirdr éQ é Zvalue1Zvalue2Zvalue3r^ r´ r g Y@r )r
|
r. c s: g | ]2\}} j d d
d d
|f d d
f |qS r ©rr ©rX r@ Úv©r; ÚidxZopar r rU s zETestFrameArithmeticUnsorted.test_binary_ops_align.<locals>.<listcomp>c s4 g | ],\}} j d d
|f d d
f |qS r r r r r r rU s )r r rH r r r r Z
|
sort_indexrn Z
|
IndexSliceru rv r ÚconcatÚitemsr9 r: Z reindex_like)r r r_ r[ rd re r r r Útest_binary_ops_align÷ s4 þý
|
ÿ ÿÿz1TestFrameArithmeticUnsorted.test_binary_ops_alignc C s2 t ddgddgg¡}ttjddd|d}td d
|
d }| ¡ }d d g|j_| ¡ }d |j _
|
|j |d d d}|j |d d d}|j |d d d}|j |d d d} |j |d d d}
|
|j |d d d} ttj d d
|
d d
|
gd d
|
d d
|
ggdd|d} ||fD ]} t | | ¡ qðd d g| j_|| |
|
| fD ]} t | | ¡ qd S )Nr4 r rU rV )r| r@ rü r! rB r{ r| rT Zlvl0Zlvl1)r. r
|
)r r r r r r r) rC r r_ r¹ ró r" r9 r: )r rk r; Úsrª Ús2Zres1rÅ Zres3Zres4Zres5Zres6r< r= r r r Ú&test_binary_ops_align_series_dataframe s, ÿ zBTestFrameArithmeticUnsorted.test_binary_ops_align_series_dataframec C s t jdddgdd}| d¡d d
}| d¡d d
}td
|
d dgi|d }td
|
d d gi|d }td
|
tjd tjgi|d }t || |¡ d S ) Nz
|
2011-01-01z
|
2011-01-02z
|
2011-01-03ÚUTC)rw z
|
Asia/Tokyor| ru r{ r4 r r )rn Z DatetimeIndexrx r r r r9 r: )r ÚbaseZidx1Zidx2r® rª r< r r r Ú test_add_with_dti_mismatched_tzs= s z<TestFrameArithmeticUnsorted.test_add_with_dti_mismatched_tzsc C s | |jd d d
¡}|d= tj|jd |jd
df< || }|d ¡ j}|d d ¡ }t |d ¡ |j| ¡ tj|j|j |¡ < t |d |j|d j ¡ t
|
|d |j¡d d
¡ ¡ sÎt t
|
|d ¡ ¡ sät || }t |j|j¡ || } t
|
|d ¡ ¡ s t t
|
| d ¡ ¡ s4t |t }
|
t
|
|
|
j¡ ¡ sTt t | } t
|
| j¡ ¡ stt t t } | jst |j |jd d d
d} t | | |d ¡ || }t|d d
|
|| }t|d d
|
|| }t|dd id
|
|| }t|d d
|
d S ) Nr| r r@ r r4 r rÏ rB rý r! )r r_ r r rr Zdropnar) r9 r¼ ÚisinÚisnanr r Úassert_index_equalr r r¿ rC r: r )r rä rå rë Z
|
frame_copyrM Zindexerr< Z
|
self_addedZ added_revZ
|
plus_emptyZ
|
empty_plusZ empty_emptyÚreverser r r Útest_combineFrameG s@ &
|
|
z-TestFrameArithmeticUnsorted.test_combineFramec
|
C s\ | |jd ¡}|| }| ¡ D ] \}}t ||| || ¡ q | ¡ }d|d< t|}|| } | ¡ D ] \}}t | | ||| ¡ qjd| kstt | d ¡
|
¡ s®t|| }t
|
|j |j k¡sÌt|| d¡ }t|dd id || d¡ }t|dd id |d| d ¡ }t|d d
|
d d d d |d| d ¡ }t|d d
|
d d d d d S ) Nr r{ r Úfloat32r r! Úfloat16rÎ rü rý )r4 r r r Úint32)rö r_ r r9 r¼ Úto_dictr r r r r rº r r* r r )
|
r rä rå rë r rM Úkeyr Z larger_seriesZ larger_addedr r r Útest_combine_series s8 ÿ ÿz/TestFrameArithmeticUnsorted.test_combine_seriesc C sx |d }|j |dd}| ¡ D ]`\}}|| }tj|| |dd || j|ksTt|j|jkrp|jdks~tq |jd ks tq |d d
}|j |dd}t |j|j¡ |d d
} |j | dd}
|
t ||
|
¡ |j |d d
dd}t t
|
j |j|j d} t || ¡ |d d
j |dd}t t
|
j |j|j d} t || ¡ |d d
j g d
|
} | j|dd}t|t|ksttd S ) Nr4 r_ r£ F)Z check_nameséûÿÿÿr r^ r{ rB )rÕ r r9 r¼ r¹ r r r_ r: r r r rC r ró rt ) r Zdatetime_framer| rM r¢ rY rd Z smaller_frameZ smaller_addedZ
|
smaller_tsZsmaller_added2re Úframer r r Útest_combine_timeseries¬ s@ ÿ ÿ z3TestFrameArithmeticUnsorted.test_combine_timeseriesc C s |d }t |j|jd ¡ |d }| ¡ D ] \}}t |j|| jd ¡ q,t|dd id t d }|j t j¡s|tt |j
|
dkstd S )Nr| r r! r ) r9 Úassert_numpy_array_equalr r r r r_ Úequalsr rt rC )r rä rå rd r} r r r r Útest_combineFuncÚ s
|
z,TestFrameArithmeticUnsorted.test_combineFuncÚfuncc C sð t ¡ }t ¡ }| d¡}t |jd ¡}|||}t |j||j|j¡ d} tj t
|
t | ¡d ||| W 5 Q R X |||}
|
t |
|
j||j|j¡ ||d} t | j||jd¡ d} tj t
|
| d |||d d
W 5 Q R X d S )NrU )r{ r{ r{ zNUnable to coerce to Series/DataFrame, dimension must be <= 2: (30, 4, 1, 1, 1)rD r zQCan only compare identically-labeled \(both index and columns\) DataFrame objectsr| ) r9 ZmakeTimeDataFramerö r r r r§ r rI rJ rK ÚreÚescape) r r÷ rä rª r® rª rø r rd rL rÊ Úresult3r r r Útest_comparisonsè s*
|
|
ÿ
|
ÿ
|
ÿz,TestFrameArithmeticUnsorted.test_comparisonsc C sH t dd dD }tt|}d}tjt|d ||d W 5 Q R X d S )Nc S s i | ]}|d dddqS )r5 r6 r7 )r[ r\ Úzr r8 r r r r] s zZTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raises.<locals>.<dictcomp>)rU rV r} z;'[<>]=?' not supported between instances of 'str' and 'int'rD r )r ru rv rI rJ rb )r Zcompare_operators_no_eq_ner; r rL r r r Ú*test_strings_to_numbers_comparisons_raises s ÿ
|
zFTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raisesc C sr t ¡ }tj|j|jd df< tjdd |jdk }W 5 Q R X tjdd |dk j}W 5 Q R X t ||¡ d S )Nr r4 Úignore)re Úraise) r9 Z makeDataFramer r rr r_ Zerrstater r§ )r Z
|
missing_dfre rd r r r Ú'test_comparison_protected_from_errstate s zCTestFrameArithmeticUnsorted.test_comparison_protected_from_errstatec C s t t d¡ d¡}t ddg¡}t ddg¡}|j}dddg}t|}t ddgddgddgg}||k}t ||¡ |j
|
|k}t ||j
|
¡ d} d}
|
d} t j t| d ||k W 5 Q R X t j t| d ||k W 5 Q R X ||k}t ||¡ |j
|
|k}t ||j
|
¡ t j t|
|
d ||k W 5 Q R X t j t| d |j
|
|k W 5 Q R X t ddgddgddgg}||k}t ||¡ t j t| d ||k W 5 Q R X t j t| d ||k W 5 Q R X ||k}t ||¡ |j
|
|k}t ||j
|
¡ t j t|
|
d ||k W 5 Q R X |j
|
j|jks$tt t d¡ d¡td
|
td d }|j|_|j|_t j t| d ||k W 5 Q R X t j t| d ||k W 5 Q R X d S ) Nr? r r| FTz5Unable to coerce to Series, length must be 2: given 3ú,Unable to coerce to DataFrame, shape must bez4operands could not be broadcast together with shapesrD ZABr
r' )r r r r r" Z
|
atleast_2dr¤ Útupler9 r: r r§ rI rJ rK r r rH r_ rC ) r r; rV Zb_rZb_cÚlstÚtupre rd Zmsg1dZmsg2dZmsg2dbr r r Útest_boolean_comparison sf
|
|
|
|
ÿz3TestFrameArithmeticUnsorted.test_boolean_comparisonc C s t d}tt dt| ¡ dt|¡|tdd}d|jd d
dd
f ¡ }t d}t d}| ¡ }|| | j |d }|| |7 < |j |d }| ¡ }|| || j |d } || || 7 < |j |d }
|
t
|
||¡ t
|
|| ¡ t
|
||
|
¡ | ¡ }|| | j |d }|| |8 < |j |d }| ¡ }|| || j |d } || || 8 < |j |d }
|
t
|
||¡ t
|
|| ¡ t
|
||
|
¡ d S )
|
NZabcdefgrN rÏ r' rÎ r{ ZbedcfZbcdefrB ) rH r r r rt r r r° r) r r9 r: ) r rC ZX_origÚZZblock1ÚsubsÚXZresult1rÊ r Zresult4r r r Útest_inplace_ops_alignmentn s> ý z6TestFrameArithmeticUnsorted.test_inplace_ops_alignmentc C s$ t dddg}ttjjdddd dd¡}| ¡ }|}|d7 }t ||¡ t |d |¡ ||ksht |j
|
|j
|
ksxt | ¡ }|}|d7 }t ||¡ t |d |¡ ||ks´t |j
|
|j
|
ksÄt | ¡ }|}|d 7 }t ||¡ t |d |¡ | ¡ }|}|d 7 }t ||¡ t |d |¡ ||ks2t |j
|
|j
|
ksDt tjjdddd}t| ¡ d
|
d }| ¡ }|}|d d7 < t| ¡ d d
|
d }t ||¡ t ||¡ |j
|
|j
|
ksÄt | ¡ }|}|d d 7 < t| ¡ d d
|
d }t ||¡ t ||¡ |j
|
|j
|
ks t d S ) Nr{ r| r r r rN rO rÏ rþ r5 rÁ r4 ) r r r rF rp r r) r9 r¼ r Z_mgrr: ) r Zs_origÚdf_origr r r; rª r© re r r r Útest_inplace_ops_identity sX z5TestFrameArithmeticUnsorted.test_inplace_ops_identityÚandrÉ rô ÚorrÍ Úxorc C s¤ |dkr d S t dddgdddgd}d}|d kr@d
|
d d
|
g|d < | ¡ }d | d}d| d}t||| t|||}t ||¡ t|}t||ks td S )Nrú r´ rµ r¶ r{ r| r rT )r¿ rÀ rÁ TFrU Z__irß )r r) ru r9 r: r r )r r r; ZoperandZdf_copyZiopre r r r Útest_inplace_ops_identity2× s z6TestFrameArithmeticUnsorted.test_inplace_ops_identity2Úvalr{ r| r )r{ r| r r! r@ c C s¨ dddg}dddg}t tj dd¡||d}tjjj}t |||d |jd
|
}t
|
|||d d |¡ t d d d gd d d gdddgd |jd
|
}t
|
|||dd |¡ d S )Nr4 r r r» ÚYr¹ r r^ )r» rÄ r¹ r r_ r{ r| rC ) r r rF rG rn ÚcoreÚopsÚalign_method_FRAMEr_ r9 r: )r rà r_ rC r; Úalignre r r r Útest_alignment_non_pandasý s
|
|
|
|
ÿz5TestFrameArithmeticUnsorted.test_alignment_non_pandasrH c C s dddg}dddg}t tj dd¡||d}tjjj}d }tj t
|
|d
|
|||d W 5 Q R X tj t
|
|d
|
|||d W 5 Q R X d S ) Nr4 r r r» rÄ r¹ r r^ z5Unable to coerce to Series, length must be 3: given 2rD r_ rC ) r r rF rG rn rÅ rÆ rÇ rI rJ rK )r rà r_ rC r; rÈ rL r r r Ú)test_alignment_non_pandas_length_mismatch s
|
|
|
zETestFrameArithmeticUnsorted.test_alignment_non_pandas_length_mismatchc C sr dddg}dddg}t tj dd¡||d}tjjj}t d d
|
dgd d d gdddgg¡}t
|
|||dd t ||j |j d¡ t
|
|||dd t ||j |j d¡ d}t d d
|
dgd d d gg¡}t jt|d |||d W 5 Q R X t jt|d |||d W 5 Q R X t d¡}t d¡}t jt|d |||d W 5 Q R X t jt|d |||d W 5 Q R X d S )Nr4 r r r» rÄ r¹ r r^ r{ r| r@ r r? r¨ r r, r_ rC r´ rD )r r r zGUnable to coerce to Series/DataFrame, dimension must be <= 2: (3, 3, 3))r r rF rG rn rÅ rÆ rÇ r" r9 r: r_ rC rI rJ rK Úzerosr« r¬ )r r_ rC r; rÈ rà rL r r r Ú'test_alignment_non_pandas_index_columns$ s6
|
|
|
"þþ
|
ÿzCTestFrameArithmeticUnsorted.test_alignment_non_pandas_index_columnsc C sF t ddgdd gd}|d }t d ¡ t||| W 5 Q R X d S )NrO rÁ r )r r9 rb ru ©r rã r; rV r r r Útest_no_warningF s z+TestFrameArithmeticUnsorted.test_no_warningc C sL t ddgdd gd}|d }tjtdd t|||d W 5 Q R X d S )NrO rÁ r z takes 2 positional argumentsrD r )r rI rJ rb ru rÍ r r r Útest_dunder_methods_binaryL s z6TestFrameArithmeticUnsorted.test_dunder_methods_binaryc C sj t jddd dd¡}t jdtd}t|}| ¡ |d< | t¡}|| ¡ }|| ¡ }t
|
||¡ d S )NrÎ rý r! rN )rN r{ z0.X) r r r r Úintr r4 r* r Zmeanr9 r: )r r» rÄ r® rª rd re r r r Útest_align_int_fill_bugS s
|
z3TestFrameArithmeticUnsorted.test_align_int_fill_bugN).r0 r1 r2 r{ r} rI r r rv rÕ rò ró r` r Úfilterwarningsr r r r r r£ r¦ r© rk rl rh rg ri rj r® r° r³ r¸ r¼ r¾ r r r" rü r rÉ rÊ rÌ rÎ rÏ rÑ r r r r rs » sv ÿ
|
|
|
&
|
=(.þ
|
N09õþ
|
üþ
|
(
|
"rs c C sP t ddddgi} t dddgd}| | }t dtjdtjgi}t ||¡ d S )Nr4 r r{ r| r r´ ©r r r r9 r: )rx ry rd re r r r Útest_pow_with_realignmentb s
|
rÔ c C st t dtjtjtjgi} t ddddgi}t ddddgi}| | }t ||¡ | d |d }t ||d ¡ d S )Nr4 r r´ )r r r r9 r: r¼ )rx ry re rd r r r Útest_pow_nan_with_zerom s rÕ c C s t tj ddd¡dddgd} tdd d
|
gdddgd }| d ¡| d ¡ d d
|
¡ }t || jdd}| d¡}|| }t
|
||¡ || d¡ }t
|
||¡ d S )Nr rÎ )rN r rU rV r} rB r{ r| r r rü rÏ ZInt64ro ) r r rF rp r Zto_numpyr rC r* r9 r: )r; rÄ re Zdf_eard r r r Ú&test_dataframe_series_extension_dtypesz s
|
rÖ c s t j ddd¡} t| }| ¡ dddd¡}t j|jddddgf< | ¡ d di¡}t j|jdd gf< | ¡ dddd
|
¡}t j|jdt d d ¡f< | ¡ dddd ¡}t j|jdt d d¡f< ||f||f||ffD ]6\ }t fdd j D }t
|
||¡ qÚd S )Nr iè )rÎ rN r )r{ r r¨ r{ r r¨ r )r| r r@ r| )r@ r r? r@ c s i | ]}| | | qS r r )rX r@ ©rx ry r r r] s z6test_dataframe_blockwise_slicelike.<locals>.<dictcomp>) r rF rp r r) r* r r° r rC r9 r: )r© r® rª r« Zdf4Zdf5r= re r r× r Ú"test_dataframe_blockwise_slicelike s rØ z df, col_dtyper´ rµ g @r~ ÚabrB rý rV r c C sT t dtjgdtjggtdd}| d|i¡}| tdgtdd }t ||¡ d S ) NrO r¶ rÙ rB rV g ð¿rU r )r r r rH r* r r9 r: )r; Z col_dtypere rd r r r Ú/test_dataframe_operation_with_non_numeric_types£ s rÚ c C s` t dggdgd} t dddggdddgd}| | }t tjddggdddgd}t ||¡ d S )Nr r )r rC r rB rÓ )r® rª rd re r r r Ú"test_arith_reindex_with_duplicates² s
|
rÛ Úto_addr{ c C sr t ddgddgd}dt| d d}tjt|d || W 5 Q R X tjt|d | | W 5 Q R X d S )Nr{ r| rZ zUnable to coerce list of r z to Series/DataFramerD )r r+ rI rJ rK )rÜ r; rL r r r Ú"test_arith_list_of_arraylike_raise» s rÝ c C s t ddddgi}| ¡ }|d }|j}|d7 }| rP|j|k sBtt ||¡ n,|j|ks^tt ddddgi}t ||¡ d S )Nr4 r{ r| r r@ )r r) Z_valuesr r9 r: )Zusing_copy_on_writer; r½ r Úvalsre r r r Ú%test_inplace_arithmetic_series_updateÇ s rß c C s t dggdgtjdgdggddgdd} t dggdgtdgddd}t dggdgtjdgdggddgdd}| | }t ||¡ dS ) zP
|
Regression test for: https://github.com/pandas-dev/pandas/issues/33765
|
r{ rU r rV r r^ ©r¹ N)r r r r r9 r: )r® rª re rd r r r Ú!test_arithemetic_multiindex_alignÙ s ý ýrá c C sF t dtdtd} | d }t t d¡tdtd}t ||¡ d S )NTrÙ Zcdr´ r
)r rH r r r9 r: )r; rd re r r r Útest_bool_frame_mult_floatì s râ c C sV t ddd g| d}t dddg| d}tddd g| d}| ¡ | ¡ }t ||¡ d S )Nr{ r| r! r r )r r Zto_framer9 r: )Zany_int_ea_dtypeZseries1Zseries2re rd r r r Útest_frame_sub_nullable_intô s
|
rã c s G fdddt G fdddt ddddgd d
|
d gd } | d } ddd d gd ddgd }t ||¡ | | }t ||¡ d S )Nc s, e Zd ZefddZe fddZdS )zEtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedSeriesc s S r r r# ©ÚSubclassedSeriesr r Ú _constructor s zRtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedSeries._constructorc s S r r r# )ÚSubclassedDataFramer r Ú_constructor_expanddim s z\test_frame_op_subclass_nonclass_constructor.<locals>.SubclassedSeries._constructor_expanddimN)r0 r1 r2 Úpropertyræ rè r ©rç rå r r rå s rå c sD e Zd ZdgZdd fdd Zedd Zefdd Z ZS )
|
zHtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFrameÚ my_extra_dataNr c s || _ t j|| d S r )rë Úsuperr )r rë ÚargsÚkwargs©Ú __class__r r r s zQtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFrame.__init__c S s t t| | j¡S r )Ú functoolsÚpartialr+ rë r# r r r ræ s zUtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFrame._constructorc s S r r r# rä r r Ú_constructor_sliced s z\test_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFrame._constructor_sliced) r0 r1 r2 Ú _metadatar ré ræ ró Ú __classcell__r rä rï r rç s
|
rç Z some_datar{ r| r r@ r r? rÁ r rN é r% )Zsdfrd re r rê r Ú+test_frame_op_subclass_nonclass_constructorý s r÷ c C sj t dd} t| jdddgi}t| jdddgi}|| j || j k}tdddg| jd}t ||¡ d S )NÚColsz col1 col2r{ r| r Trà )r r r r r9 r¼ )rø Zq1Zq2rd re r r r Útest_enum_column_equality! s
|
rù )AÚ collectionsr r r Úenumr rñ rv r« Únumpyr rI Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr2 Zpandasrn r r r r Zpandas._testingZ_testingr9 Zpandas.core.commonrÅ Úcommonr Zpandas.core.computationr
|
r Z#pandas.core.computation.expressionsr r Zpandas.tests.frame.commonr r Zfixturer r r3 r rÇ r$ rn rp rr rs rÔ rÕ rÖ rØ r r rH rÚ rÛ rÝ rß rá râ rã r÷ rù r r r r Ú<module> sr
|
U H f E
|
, þþ
|
.
|
$
|
