U
|
ý°dd ã @ sp d dl Zd dlZd dlm mZ d dlZd dlm Z m
|
Z
|
m Z m Z d dl mZ e ¡ dd ZG dd dZdS )é N)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesc s& G fdddt ddddgiS )Nc s e Zd Ze fddZdS )z2gpd_style_subclass_df.<locals>.SubclassedDataFramec s S ©N© ©Úself©ÚSubclassedDataFramer úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_subclass.pyÚ _constructor s z?gpd_style_subclass_df.<locals>.SubclassedDataFrame._constructorN)Ú__name__Ú
|
__module__Ú __qualname__Úpropertyr r r
|
r r r s r Úaé é é )r r r r
|
r Úgpd_style_subclass_df s r c @ s$ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zej d+d,d-g¡d.d/ Zd0d1 Z d2d3 Z e j d4d5 Z d6d7 Z!d8d9 Z"d:d; Z#d<d= Z$d>d? Z%d@dA Z&dBdC Z'dDS )EÚTestDataFrameSubclassingc s G fdddt G fdddt tdtdd} |}t| sRt|j}t|sft| ¡ dksvt|dd
}t| st| ¡ dks tt d
|
d g¡} d dgd dgg|d}t|d sØtt ddg¡} d dgd dgg|d}t|d std S )Nc s$ e Zd Ze fddZdd ZdS )zQTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeriesc s S r r r ©Ú CustomSeriesr r r s z^TestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeries._constructorc S s dS ©NÚOKr r r r r Úcustom_series_function$ s zhTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeries.custom_series_functionN)r r r r r r r r r r r s r c sB e Zd ZdZdd fdd ZefddZZdd Z Z S )
|
zTTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFramez
|
Subclasses pandas DF, fills DF with simulation results, adds some
|
custom plotting functions.
|
N)Úreturnc s t j|| d S r )ÚsuperÚ__init__)r ÚargsÚkw©Ú __class__r r r - s z]TestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame.__init__c s S r r r )ÚCustomDataFramer r r 0 s zaTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame._constructorc S s dS r r r r r r Úcustom_frame_function6 s zjTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame.custom_frame_function)
|
r r r Ú__doc__r r r Z_constructor_slicedr% Ú __classcell__r ©r$ r r" r r$ ' s r$ é
|
)Úcol1Zcol2r r é )ÚAr, )r, ÚBr r r ©Úcolumnsr, )r, Ú )r- r0 )
|
r r ÚrangeÚ
|
isinstanceÚAssertionErrorr* r r% r Ú from_tuples)r ÚdataZcdfZ
|
cdf_seriesZcdf_rowsZmcolZ cdf_multiZ
|
cdf_multi2r r( r Ú"test_frame_subclassing_and_slicing s" z;TestDataFrameSubclassing.test_frame_subclassing_and_slicingc C sè t jdddgdddgddddgd}d |_|jd ks:t|d
|
g jd ksNt|jddgd d
f jd ksnt|jd dgd d
f jd kst|jd d
d d
f jd ks®tt |¡}t ||¡ |j|jksÔt|j|jksätd S ) Nr r r )ÚXÚYr ÚbÚc©ÚindexZXXXr7 r ) Útmr Ztestattrr3 ÚlocÚilocZround_trip_pickleÚassert_frame_equalÚ _metadata)r ÚdfZ unpickledr r r Útest_dataframe_metadataR s ÿ
|
z0TestDataFrameSubclassing.test_dataframe_metadatac C sê t jdddgdddgddd gd
|
d d d gd}|jd d
df }t jdddgtddd}t ||¡ t|t jsvt|jd d
df }t jdddgtddd}t ||¡ t|t js¾t|jd d
df }t jddd gtddd}t ||¡ t|t jst|jd d d
f }t jdddgtdd d}t ||¡ t|t jsRt|jdd d
f }t jdddgtdd d}t ||¡ t|t jst|jd d d
f }t jddd gtdd d}t ||¡ t|t jsætd S )Nr r r é r+ é é é é )r7 r8 ÚZr r9 r: r; r7 Úabc©r< Únamer8 rI ZXYZ) r= r r> ÚSubclassedSeriesÚlistÚassert_series_equalr2 r3 r? ©r rB ÚresÚexpr r r Útest_indexing_slicedf s8 ÿ z-TestDataFrameSubclassing.test_indexing_slicedc C s6 G dd dt }tjtdd | j W 5 Q R X d S )Nc @ s e Zd Zedd ZdS )zFTestDataFrameSubclassing.test_subclass_attr_err_propagation.<locals>.Ac S s | j S r )Z i_dont_existr r r r Ú nonexistence s zSTestDataFrameSubclassing.test_subclass_attr_err_propagation.<locals>.A.nonexistenceN)r r r r rT r r r r r, s r, z.*i_dont_exist.*)Úmatch)r ÚpytestZraisesÚAttributeErrorrT )r r, r r r Ú"test_subclass_attr_err_propagation s z;TestDataFrameSubclassing.test_subclass_attr_err_propagationc C sP t jdddgdddgdtdd}t jdddgdddgd td
|
d}|j|d d \}}t jdtjdtjdgdtjdtjdgdtd d}t jddtjdtjgddtjdtjgd td d}t|t jsÒtt ||¡ t|t jsîtt ||¡ |j |j
|
¡\}}t|t j s tt ||j ¡ t|t j s>tt ||j
|
¡ d S )Nr r r+ ©r r9 ÚACEr; r rD )r: ÚdÚABDr ©ÚaxisÚABCDE) r= r rN ÚalignÚnpÚnanr2 r3 r@ r r: rM rO )r Údf1Údf2Úres1Úres2Úexp1Úexp2r r r Útest_subclass_align s4 ÿ ÿ$þ$þ z,TestDataFrameSubclassing.test_subclass_alignc C s* t jdddgdddgdtdd}t jdddgtd d
|
d }|j|d d \}}t jdtjdtjdgdtjdtjdgdtdd}t jddtjdtjgtdd
|
d }t|t js¶tt ||¡ t|t jsÒtt
|
||¡ | |¡\}}t|t jsütt
|
||¡ t|t jstt ||¡ d S )Nr r r+ rY rZ r; r rD r\ ÚxrK r r] r_ ) r= r rN rM r` ra rb r2 r3 r@ rO )r rB Úsre rf rg rh r r r Ú test_subclass_align_combinations° s* $$þ ÿ z9TestDataFrameSubclassing.test_subclass_align_combinationsc C sH t ddgi¡}| ¡ D ]*\}}t|t js0tt ||j| ¡ qd S )Nr r )r= r Ziterrowsr2 rM r3 rO r> )r rB ÚiÚrowr r r Útest_subclass_iterrowsÌ s z/TestDataFrameSubclassing.test_subclass_iterrowsc
|
C sx t jdddgdddgddd ggd
|
d d gd ddgd}| ¡ }t jddddddddd g tdtdgd}t ||¡ d S )Nr r r rD r+ rE rF rG rH r r9 r: r7 r8 rI ©r< r/ Z aaabbbcccZ XYZXYZXYZr; )r= r ÚstackrM rN rO rP r r r Útest_subclass_stackÓ s ý ÿz,TestDataFrameSubclassing.test_subclass_stackc
|
C s¤ t jddddgddddgd d
|
d d gd dddggtjtttdtdddgdtjtttdtdddgdd}t jddgddgddgddgd d gd
|
d gd dgddggtjtttdtd td dddgdtd d gdd d}| ¡ }t ||¡ | d¡}t ||¡ t jddgddgddgddgd d
|
gd d gd dgddggtjtttdtd td!dddgdtd"d#gdd d}| d¡}t ||¡ d S )$Nr) é é é é é é é é é é é! é( é) é* é+ ÚAABBÚcdcdÚaaaÚccc©ÚnamesÚWWXXÚyzyzÚwwwÚyyyrp ÚAAAABBBBÚccddccddÚyzyzyzyzÚWr7 ©rL ÚWXWXWXWXÚyÚz© r= r r r4 rN Úzipr rq r@ ©r rB rR rQ r r r Útest_subclass_stack_multiâ sb * ÿ ÿû
|
ø
|
þñ
|
ø
|
þñ
|
z2TestDataFrameSubclassing.test_subclass_stack_multic
|
C s¤ t jddddgddddgd d
|
d d gd dddggtjtttdtdddgdtjtttdtdddgdd}t jddgddgddgddgd d gd
|
d gd dgddggtjtttdtd td dddgdtd d gdd d}| ¡ }t ||¡ | d¡}t ||¡ t jd!d"gddgd#d$gddgd%d&gd d gd'd(gddggtjtttdtd td)dddgdtd*d+gdd d}| d¡}t ||¡ d S ),Nr) rs ç (@ç *@rv rw ç 6@ç 7@rz r{ ç @@ç @@r~ r ç E@ç E@r r r r
r r r r r rp r r r r r7 r g $@g &@g 4@ç 5@g >@g ?@g D@g D@r r r r r r r r Ú test_subclass_stack_multi_mixed sj
|
|
|
|
ü ÿ ÿöø
|
þñ
|
ø
|
þñ
|
z8TestDataFrameSubclassing.test_subclass_stack_multi_mixedc
|
C sx t jdddgdddgddd ggd
|
d d gd ddgd}| ¡ }t jddddddddd g tdtdgd}t ||¡ d S )Nr r r rD r+ rE rF rG rH r r9 r: r7 r8 rI rp Z XXXYYYZZZZ abcabcabcr; )r= r ÚunstackrM rN rO rP r r r Útest_subclass_unstackY s ý ÿz.TestDataFrameSubclassing.test_subclass_unstackc
|
C s t jddddgddddgd d
|
d d gd dddggtjtttdtdddgdtjtttdtdddgdd}t jddddddddgd d d
|
dd dd dggtdd gdd tjtttd td td dddgdd}| ¡ }t ||¡ | d¡}t ||¡ t jdd dd
|
dd dd gdd ddddddggtd!d"gdd tjtttd td td#dddgdd}| d¡}t ||¡ d S )$Nr) rs rt ru rv rw rx ry rz r{ r| r} r~ r r r r r r r
r r r r r rp r, r- r ÚWWWWXXXXÚyyzzyyzzÚcdcdcdcdr: r[ ÚABABABAB© r= r r r4 rN r r r¢ r@ r r r r Útest_subclass_unstack_multih sB * ÿ ÿû
|
&þý
|
&þý
|
z4TestDataFrameSubclassing.test_subclass_unstack_multic
|
C s t jddddgddddgd d
|
d d gd dddggtjtttdtdddgdtjtttdtdddgdd}t jddddddddgd d d
|
dd dd dggtdd gdd tjtttd td td dddgdd}| ¡ }t ||¡ | d¡}t ||¡ t jdd dd
|
dd dd gdd ddddddggtd!d"gdd tjtttd td td#dddgdd}| d¡}t ||¡ d S )$Nr) rs r r rv rw r r rz r{ r r r~ r r r r r r r
r r r r r rp r, r- r r¤ r¥ r¦ r: r[ r§ r¨ r r r r Ú!test_subclass_unstack_multi_mixed sR
|
|
|
|
ü ÿ ÿöþþú
|
þþú
|
z:TestDataFrameSubclassing.test_subclass_unstack_multi_mixedc
|
C s t ddddddgddddddgddddddgd ¡}|jd
|
d d d }t dddd dddd d¡}d\|j_|j_t ||¡ d S )Nr, r- ÚCÚOneÚTwog ð?g @g @)r< r/ Úvaluesr< r/ r® ©r, r- r« )r¬ r rp )r= r Zpivotr< rL r/ r@ )r rB ZpivotedÚexpectedr r r Útest_subclass_pivotÁ s ýÿ
|
|
þÿz,TestDataFrameSubclassing.test_subclass_pivotc C s t ddgddgddgddgd ¡}tj|d
|
d gd }t jddd dgddd dgddddgddddggd
|
d ddgd}t ||¡ d S )NÚJohnÚMaryÚDoeÚBoç @ç @é é )ÚfirstÚlastÚheightÚweightrº r» )Zid_varsr¼ r½ ÚvariableÚvaluer. )r= r ÚpdZmeltr@ )r ZcheeseZmeltedr° r r r Útest_subclassed_meltØ s" üÿ
|
|
|
|
ü
|
ù
|
z-TestDataFrameSubclassing.test_subclassed_meltc
|
C sú t j d¡ t j d¡}t ddddddd dd
|
d d dd dddtttd|d¡}|j |d< |
|
¡ |
|
¡ dddddd gd
|
d d d ddgddddddgddddddgd}t |¡}| ddg¡dddg }t j |ddgddd }t ||¡ d S ) Né{ r r r9 r: )r r r r[ ÚeÚfg @g333333ó?gffffffæ?g @gÍÌÌÌÌÌô?g¹?)ZA1970ZA1980ZB1970ZB1980r7 Úidi² i¼ r r r )r7 r, r- ÚyearrÅ rÆ r7 r, r- )rm Új)ra ÚrandomÚseedÚrandnr= r Údictr r1 r< ÚtolistZ set_indexrÀ Z wide_to_longr@ )r rj rB Zexp_datar° Z
|
long_framer r r Ú test_subclassed_wide_to_longñ s*
|
|
|
|
ûÿ
|
|
û
|
z5TestDataFrameSubclassing.test_subclassed_wide_to_longc sÐ dd dd t jddddgd d
|
dd gddd d gd d
|
d dggddddgd}| fdd¡ |j fdddd t jdddd gd d
|
ddgddd d gd d
|
d dggddddgd}|jfdddd}t|t jsÚtt ||¡ t ddd gddd gddd gddd gg¡}|jd d dd}t|t js4tt ||¡ |jd d dd d }t|t jsftt ||¡ t ddd gddd gddd gddd gg¡}|jd!d dd}t|t jrÀtt ||¡ d S )"Nc S s t | tjstd S r )r2 r= rM r3 ©rn r r r Úcheck_row_subclass s zJTestDataFrameSubclassing.test_subclassed_apply.<locals>.check_row_subclassc S s | d dkr | d d7 < | S )Nr¾ r¼ r¿ g à?r rÎ r r r Ústretch s z?TestDataFrameSubclassing.test_subclassed_apply.<locals>.stretchr² r´ r¼ r¶ r³ rµ r· r½ r¸ r¹ rº r» r¾ r¿ r. c s | S r r ©rj ©rÏ r r Ú<lambda># ó z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>c s | S r r rÑ rÒ r r rÓ $ rÔ r r] g @c s | S r r rÑ )rÐ r r rÓ 0 rÔ r r c S s t dddg¡S ©Nr r r )r= rM rÑ r r r rÓ 6 rÔ c S s
|
dddgS rÕ r rÑ r r r rÓ : rÔ Úexpand)r^ Z result_typec S s
|
dddgS rÕ r rÑ r r r rÓ @ rÔ )r= r Úapplyr2 r3 r@ rM rO )r rB r° Úresultr )rÏ rÐ r Útest_subclassed_apply sD
|
|
|
|
ü
|
ù
|
|
|
|
|
ü
|
ù
|
* *z.TestDataFrameSubclassing.test_subclassed_applyc C sD t dddgdddgddd gd
|
¡}t|| }t|t js@td S © Nr r r rD r+ rE rF rG rH r¯ ©r= r Úgetattrr2 rM r3 )r Zall_reductionsrB rØ r r r Útest_subclassed_reductionsD s $ z3TestDataFrameSubclassing.test_subclassed_reductionsc C s> t dddddgdtjdddgdd d d dgd
|
¡}| ¡ }t|t jsJtt d d d gd ddgddd gd¡}| ¡ }t|t jstt jddddgddddgdd d dgd d d d!ggtj t
|
t t
|
d"t
|
d#d$d%gd&tj t
|
t t
|
d't
|
d(d)d*gd&d+}| ¡ }t|t jstt ¡ }| ¡ }t|t js:td S ),Nr² ZMylaZLewisg 8@r r} é FT)ZPersonZAgeÚSingler r r r+ rE rF rG r¯ r) rs rt ru rv rw rx ry rz r{ r| r~ r r r r r r r
r r r r r rp ) r= r ra rb Úcountr2 rM r3 r r4 rN r ©r rB rØ r r r Útest_subclassed_countK s6 ýÿ$* ÿ ÿû z.TestDataFrameSubclassing.test_subclassed_countc C sB t jddgddgdddgd}| ddg¡}t|t js>td S )Nr rD r )Znum_legsZ num_wingsZfalconZdogr; )r= r Úisinr2 r3 rá r r r Ú test_isinj s ÿz"TestDataFrameSubclassing.test_isinc C s` t dddgdddgddd gd
|
¡}| ¡ }t|t js<tt ¡ }| ¡ }t|t js\td S rÚ )r= r Z
|
duplicatedr2 rM r3 rá r r r Útest_duplicatedq s $z(TestDataFrameSubclassing.test_duplicatedÚ
|
idx_methodÚidxmaxÚidxminc C sD t dddgdddgddd gd
|
¡}t|| }t|t js@td S rÚ rÛ )r ræ rB rØ r r r Útest_idxz s $ z!TestDataFrameSubclassing.test_idxc C s t ddddgddddgg¡}t ddddg¡}| |¡}t|t jsJtt ddddgddddgg¡}t ddddg¡}| |¡}t|t jstd S )Nr r éþÿÿÿéÿÿÿÿr )r= r rM Údotr2 r3 )r rB rk rØ r r r Útest_dot s
|
|
z!TestDataFrameSubclassing.test_dotc C s\ t dddgdddgddd gd
|
¡}| ¡ }t|t js<t|jd d }t|t jsXtd S ) Nr r r rD r+ rE rF rG rH r¯ Fr; )r= r Z memory_usager2 rM r3 rá r r r Útest_memory_usage s
|
$ z*TestDataFrameSubclassing.test_memory_usagec C s| dddddg}dddd g}t jtj d
|
d ¡||d }t jtj d d ¡|d d
|d }|j|d ddd}t|t jsxtd S )Nr r9 r: r[ rà ZoneZtwoZthreeZfourr+ rD rp r TZkendall)r^ ZdropÚmethod) r= r ra rÈ rÊ Zcorrwithr2 rM r3 )r r< r/ rc rd Zcorrelsr r r Ú test_corrwith s ÿ
|
ÿz&TestDataFrameSubclassing.test_corrwithc C s¬ d}t jd|dd}tjtjtjtjgtjtjtjgtjtjtjgd|d}| |dd
¡}t|tjspt| |d ¡}t|tj st| d¡}t|tj s¨td S ) Nr z1/1/1990Z53s)ZperiodsÚfreqr¯ r; rê z
|
1989-12-31)
|
rÀ Z
|
date_ranger= r ra rb Zasofr2 r3 rM )r ÚNÚrngrB rØ r r r Ú test_asof¡ s ýú
|
z"TestDataFrameSubclassing.test_asofc C s@ t dddgdddgddd gd
|
¡}| ¡ }t|t js<td S rÚ )r= r rè r2 rM r3 rá r r r Ú test_idxmin_preserves_subclass¶ s $z7TestDataFrameSubclassing.test_idxmin_preserves_subclassc C s@ t dddgdddgddd gd
|
¡}| ¡ }t|t js<td S rÚ )r= r rç r2 rM r3 rá r r r Ú test_idxmax_preserves_subclass½ s $z7TestDataFrameSubclassing.test_idxmax_preserves_subclassc C sZ t dddgdddgddd gd
|
¡}| ¡ }t|t js<t| ¡ }t|t|sVtd S rÚ )r= r Zconvert_dtypesr2 r3 Útype)r r rB rØ r r r Ú&test_convert_dtypes_preserves_subclassÄ s
|
$z?TestDataFrameSubclassing.test_convert_dtypes_preserves_subclassc C sP t dddgdddgddd gd
|
¡}| tjtjtjd
|
¡}t|t jsLtd S rÚ ) r= r Zastypera Úint64Úint32Úfloat64r2 r3 rá r r r Ú test_astype_preserves_subclassÍ s $z7TestDataFrameSubclassing.test_astype_preserves_subclassc C sF t ddddgi}t ddddgi¡}| |¡s4t| |¡sBtd S )Nr r r r )r r= r Úequalsr3 )r rc rd r r r Útest_equals_subclassÔ s z-TestDataFrameSubclassing.test_equals_subclassc C sZ t ddddgi¡}|jddgdd}t ddddgi¡}t|t jsJtt ||¡ d S )Nr, r r r Zffill)rï )r= r Úreplacer2 r3 r@ )r rB rØ r° r r r Útest_replace_list_methodÜ s
|
z1TestDataFrameSubclassing.test_replace_list_methodN)(r r r r6 rC rS rX ri rl ro rr r r¡ r£ r© rª r± rÁ rÍ rÙ rÝ râ rä rå rV ÚmarkZ parametrizeré rí rî ÚtdZskip_if_no_scipyrð rô rõ rö rø rü rþ r r r r r r sD 7#
|
9>'2 6
|
|
r )Únumpyra rV Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr ZpandasrÀ r r r r Zpandas._testingZ_testingr= Zfixturer r r r r r Ú<module> s
|
|
