U
|
ý°d´B ã @ s d dl Z d dlZd dlZd dlm mZ d dlZ d dlm
|
Z
|
m Z m Z d dl mZ G dd dZG dd dZG dd dZdS ) é N)Ú DataFrameÚSeriesÚisnac @ s e Zd Zdd Zej ddddddg¡d d
|
Zej d e dddg¡e
|
d d dg¡g¡dd Z ej dddg¡dd Z dS )ÚTestDataFrameCovc C sæ | ¡ }|j t|d}t ||¡ |j t|d d}t|j ¡ sJt| ¡ }t j
|
|j d d
|j d¡f< t j
|
|j dd
|j d¡f< |j t|d d}| ¡ }t j
|
|jd< t j
|
|jd < t ||¡ | ¡ }|d |d
|
¡}t |d d
|
|¡ tjtd d | ¡ W 5 Q R X |j d d}|jd d
ddd
|
dgf ¡ }t ||¡ tt ddd¡}| ¡ }tt |jj¡ d¡|j |j d}t ||¡ t j
|
|jd< | ¡ }tt |jdd
j¡ d¡|j |j d}t ||¡ d S )N)Ú min_periodsé é ÚAé
|
ÚBé ©r r ©r r ÚCú!could not convert string to float©ÚmatchT©Ú numeric_onlyÚDç ç ð?)r r ©ÚindexÚcolumnsr )ÚcovÚlenÚtmÚassert_frame_equalr ÚvaluesÚallÚAssertionErrorÚcopyÚnpÚnanÚilocr Zget_locÚlocÚassert_almost_equalÚpytestÚraisesÚ
|
ValueErrorr ZlinspaceÚTÚreshape)ÚselfÚ float_frameÚfloat_string_frameÚexpectedÚresultÚframeÚdf© r4 ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_cov_corr.pyÚtest_cov sJ ÿ ýzTestDataFrameCov.test_covÚ test_ddofNr r é é c C sZ t j d¡}t j d¡}t||d}|j|d}t j|||d}t|}t ||¡ d S )Nr
|
©r r )Zddof)r# ÚrandomZrandr r r r )r- r7 Z np_array1Z np_array2r3 r1 Z expected_npr0 r4 r4 r5 Ú test_cov_ddofB s z TestDataFrameCov.test_cov_ddofÚ other_columnr ç @ç @c C s^ t t ddd g¡|d}| ¡ }t ddgddgg¡}t |ddgddgd}t ||¡ d S ) Nr r8 ©ÚaÚbç à?r rA rB ©r r )r ÚpdÚarrayr r# r r )r- r= Údatar1 Zarrr0 r4 r4 r5 Útest_cov_nullable_integerM s
|
z*TestDataFrameCov.test_cov_nullable_integerr TFc C sp t ddgddgd}t ddgdgd}|rF|j|d }t ||¡ n&tjtd
|
d |j|d W 5 Q R X d S ) Nr r ÚxÚy)rA ÚcrC rA r r r r )r r r r r( r) r* )r- r r3 r0 r1 r4 r4 r5 Útest_cov_numeric_onlyX s z&TestDataFrameCov.test_cov_numeric_only) Ú__name__Ú
|
__module__Ú __qualname__r6 r( ÚmarkÚ parametrizer< rE rF r# rH rL r4 r4 r4 r5 r s 0
|
|
ÿ
|
r c @ sÈ e Zd Zej ddddg¡ejdd Zdd Z ejej d dddg¡d
|
d Z
|
ej d ddg¡d d Z ejej d dddg¡dd Z ej dddg¡dd Z dd Zdd Zejej de dddg¡e ddd g¡g¡ej d e dddg¡e d d d g¡e d d ejg¡g¡ej ddddg¡d!d" Zd#d$ Zej d%dd&d'd(g¡d)d* Zd+d, Zejej ddddg¡d-d. Zejej d dddg¡ej d/d0d1g¡d2d3 Zd S )4ÚTestDataFrameCorrÚmethodZpearsonÚkendallÚspearmanc C s t j|j|jd d
df< t j|j|jdd
df< |d dd
|j|jd d
df< |j|d}|d j|d |d}t |d d |¡ d S )Nr r r
|
r é ©rS r )r# r$ r& r Úcorrr r' )r- r. rS Úcorrelsr0 r4 r4 r5 Útest_corr_scipy_methodj s $ z(TestDataFrameCorr.test_corr_scipy_methodc C s\ t jtdd | ¡ W 5 Q R X |jdd}|jd d
ddddgf ¡ }t ||¡ d S ) Nr r Tr r r r r )r( r) r* rX r& r r )r- r/ r1 r0 r4 r4 r5 Útest_corr_non_numericw s
|
z'TestDataFrameCorr.test_corr_non_numericÚmethc C s´ t dddtjtjtjgtjtjtjdddgtjtjtjtjtjtjgd}| |¡}t|jd shtt|jd szt|jd dkst|jd dkstt|jd s°td S ) Nr g ø?©r r r r r )r r )r r )r r )r r# r$ rX r r& r! ©r- r\ r3 Úrsr4 r4 r5 Útest_corr_nooverlap~ s ýÿ
|
z%TestDataFrameCorr.test_corr_nooverlapc C sR t dddtjtjtjgtjtjtjdddgd}| |¡}t|j ¡ sNtd S )Nr r )r r# r$ rX r r r r! r^ r4 r4 r5 Útest_corr_constant s þÿ
|
z$TestDataFrameCorr.test_corr_constantc C sp t ddgddgd}t t d¡ddgddgd }tjdd
|
t d t¡ | |¡}W 5 Q R X t ||¡ d S ) NTFr r r@ ©r8 r8 rA rB r )ÚrecordÚignore)
|
r r# ÚonesÚwarningsÚcatch_warningsÚ simplefilterÚRuntimeWarningrX r r )r- r\ r3 r0 r1 r4 r4 r5 Útest_corr_int_and_boolean s z+TestDataFrameCorr.test_corr_int_and_booleanr rX c C sR t tj d¡ dd¡tdd}t|| }|j|jk s<t |j
|
|j¡sNt d S )Né( r
|
é Zabcd©r ) r r# r; Úrandnr, ÚlistÚgetattrr r r! Úequals)r- rS r3 r1 r4 r4 r5 Ú&test_corr_cov_independent_index_column« s z8TestDataFrameCorr.test_corr_cov_independent_index_columnc C s@ t tjjdd}d}tjt|d |jdd W 5 Q R X d S )N©r
|
r8 ©ÚsizezGmethod must be either 'pearson', 'spearman', 'kendall', or a callable, r Z____rW )r r# r; Únormalr( r) r* rX )r- r3 Úmsgr4 r4 r5 Útest_corr_invalid_method³ s z*TestDataFrameCorr.test_corr_invalid_methodc C s2 t ddddgddddgd}| ¡ | ¡ d S )Nr r8 r9 rl r@ )r r rX )r- r3 r4 r4 r5 Ú test_corr_intº s z TestDataFrameCorr.test_corr_intÚnullable_columnr r8 r9 Nr= r r> r? c C sF t ||d}|j|d}t t d¡ddgddgd}t ||¡ d S )Nr@ rW rb rA rB rD )r rX r# re r r )r- rz r= rS rG r1 r0 r4 r4 r5 Útest_corr_nullable_integerÁ s z,TestDataFrameCorr.test_corr_nullable_integerc C s² t dtdi}tdd d d
|d< |d }t|jjdksBt|jdd}|rpd|jd < |jd
|
d ks®tn>d|j d < |jd
|
dkst|d |kst|j d dks®td S ) Nr r
|
éÿÿÿÿr r8 Tr éc r )r r )r r )
|
r Úranger Z_mgrZarraysr! rX r% r& r )r- Zusing_copy_on_writer3 ZserÚ_r4 r4 r5 Útest_corr_item_cacheÑ s
|
|
z&TestDataFrameCorr.test_corr_item_cacheÚlengthrV éÈ iÐ c C sX t |ddgg ddgd}| ¡ }t tjtjgtjtjgdddgd}t ||¡ d S )NçÙ?g¹?r r rm r ©r ©r rX r# r$ r r )r- r r3 r1 r0 r4 r4 r5 Ú test_corr_for_constant_columnsæ s ÿz0TestDataFrameCorr.test_corr_for_constant_columnsc C sP t dddgdddgd}| ¡ }t ddgddgdddgd}t ||¡ d S ) Ng#B ¡Ç;g#B ¡×;g±mÉxµá;r r r r r ©r rX r r ©r- r3 r1 r0 r4 r4 r5 Ú test_calc_corr_small_numbersð s ÿ z.TestDataFrameCorr.test_calc_corr_small_numbersc C sZ t ddgddgd}|j|dd}t tjtjgtjtjgdddgd}t ||¡ d S ) Nr r8 r r9 )rS r r r r r
)r- rS r3 r1 r0 r4 r4 r5 Ú)test_corr_min_periods_greater_than_lengthù s ÿz;TestDataFrameCorr.test_corr_min_periods_greater_than_lengthr TFc C s t ddgddgddgd}t t d¡ddgddgd }|rX|j||d
|
}t ||¡ n(tjtd d |j||d
|
W 5 Q R X d S ) Nr r rI rJ ©rA rB rK rb rA rB r r r r ) r r# re rX r r r( r) r* )r- r\ r r3 r0 r1 r4 r4 r5 Útest_corr_numeric_only s z(TestDataFrameCorr.test_corr_numeric_only)rM rN rO r( rP rQ ÚtdÚskip_if_no_scipyrZ r[ r` ra rj rr rx ry rE rF r# r$ r{ r r r r r r4 r4 r4 r5 rR g sL
|
|
ÿ.þ
|
rR c @ s e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
dd Z dd Z dd Z dd Zejdd Zejdd Zejdd Zd S ) ÚTestDataFrameCorrWithc C s® |}t tj t|¡|jd}|j|dd}|j|jd d d
|jd d d
dd
d}|d= |j |dd}t
|
|d |d |d ¡¡ |j |d d}t
|
||jj |jdd¡ |j |dd
|
d }t
|
|d |d |d ¡¡ d|ksît|j |d d
|
d }|jd |jkstd d dddg}ddddg} ttj dd¡|| d}
|
ttj dd¡|d d
| d} |
|
j | d d} |d d
D ](} t
|
| | |
|
j| | j| ¡¡ qd S )Nr r ©Úaxisr| r
|
rD r r r T)r ÚdroprA rB rK ÚdÚeZoneZtwoZthreeZfourr rl r )r r# r; rn r r ÚaddZreindexr Úcorrwithr r' rX Úassert_series_equalr+ r! r r& )r- Údatetime_framerA ZnoiserB ZcolcorrZrowcorrZdroppedr r Údf1Údf2rY Úrowr4 r4 r5 Ú test_corrwith s* . z#TestDataFrameCorrWith.test_corrwithc C s t ¡ }t ¡ }ddddg}d|d< d|d< tjtdd | |¡ W 5 Q R X |j|d
|
d }|jd d
|f |jd d
|f ¡}t ||¡ tjtd d |j|d d W 5 Q R X |j|d d
|
d}|jd d
|f j|jd d
|f d d}t ||¡ d S )Nr r r r ZfooÚobjÚbarzCould not convertr Tr zunsupported operand typer r )r r )r ZmakeTimeDataFramer( r) Ú TypeErrorr r& r )r- r r Úcolsr1 r0 r4 r4 r5 Útest_corrwith_with_objects5 s & *z0TestDataFrameCorrWith.test_corrwith_with_objectsc C s. | |d ¡}| |d j¡}t ||¡ d S )Nr )r ÚapplyrX r r )r- r r1 r0 r4 r4 r5 Útest_corrwith_seriesI s z*TestDataFrameCorrWith.test_corrwith_seriesc C sr t t d¡dgd}t t d¡d dgd}| |¡d }t |d |d ¡d d }t ||¡ |dk sntd S )Ni' rA rm r8 r r )r r# Úaranger Zcorrcoefr r' r! )r- r r Zc1Úc2r4 r4 r5 Ú test_corrwith_matches_corrcoefO s z4TestDataFrameCorrWith.test_corrwith_matches_corrcoefr TFc C s° t ddddgddddgddd d
|
gd }td dddg}|r|j||d }|d |¡|d |¡g}t|ddgd}t ||¡ n(tjtdd |j||d W 5 Q R X d S )Nr rl r9 r8 é é rA rB rK r r r r )rG r z4unsupported operand type\(s\) for /: 'str' and 'int'r ) r r r rX r r r( r) r )r- r r3 Úsr1 Zcorrsr0 r4 r4 r5 Útest_corrwith_mixed_dtypesX s "ÿ þz0TestDataFrameCorrWith.test_corrwith_mixed_dtypesc C sl t tjjddddgd}t tjjdddddgd}|j|dd j ¡ }|j |j¡ ¡ }t ||¡ d S )
|
Nrs rt rA rB rm ©r
|
r9 rK T©r )
|
r r# r; r r Ú sort_valuesr Ú intersectionr Úassert_index_equal©r- r r r1 r0 r4 r4 r5 Ú test_corrwith_index_intersectionk s
|
z6TestDataFrameCorrWith.test_corrwith_index_intersectionc C sl t tjjddddgd}t tjjdddddgd}|j|dd j ¡ }|j |j¡ ¡ }t ||¡ d S )
|
Nrs rt rA rB rm r« rK Fr¬ )
|
r r# r; r r r r Úunionr r¯ r° r4 r4 r5 Útest_corrwith_index_unions s
|
z/TestDataFrameCorrWith.test_corrwith_index_unionc C sn t t t d¡gd ¡j}| ¡ }tj||d fdd}| |¡}t t
|
d¡ddddgd}t ||¡ d S ) Nr
|
r9 r r r rl r8 r ) r r# Zvstackr¤ r+ r" rE Úconcatr r re r r r° r4 r4 r5 Útest_corrwith_dup_cols{ s
|
z,TestDataFrameCorrWith.test_corrwith_dup_colsc C sN t ddgddgg}| ¡ }t ddgddgd}tj|d |d dd d S ) NgÉ?r r g ð¿r: r gÔFFõg<)Zatolr r r4 r4 r5 Ú!test_corr_numerical_instabilities
s z7TestDataFrameCorrWith.test_corr_numerical_instabilitiesc C sF t tjjdd}|j|d dd}tt t|¡}t ||¡ d S )N©éd r9 rt r8 rU rW © r r# r; r r re r r r r r4 r4 r5 Útest_corrwith_spearman s z,TestDataFrameCorrWith.test_corrwith_spearmanc C sF t tjjdd}|j|d dd}tt t|¡}t ||¡ d S )Nr· rt r8 rT rW r¹ r r4 r4 r5 Útest_corrwith_kendall s z+TestDataFrameCorrWith.test_corrwith_kendallc C sÊ t dtjddgddddgdddd gd
|
}|d d g }|d j|jd d}tdddgdd d gd}t ||¡ t ddddgddddgd}tddddg}| |¡}tddgdd gd}t ||¡ d S )Nr r¨ r FTr
|
rl é r9 r] r r rU rW r r r r r g( µ¦yâ?)r r# r$ r r r r r )r- r r r1 r0 Zdf_boolZser_boolr4 r4 r5 Ú%test_corrwith_spearman_with_tied_data s"
|
|
ýÿ ÿ
|
z;TestDataFrameCorrWith.test_corrwith_spearman_with_tied_dataN)rM rN rO r r¡ r£ r¦ r( rP rQ rª r± r³ rµ r¶ r r rº r» r½ r4 r4 r4 r5 r s
|
|
|
|
r )rf Únumpyr# r( Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr ZpandasrE r r r Zpandas._testingZ_testingr r rR r r4 r4 r4 r5 Ú<module> s V .
|
