zmc
2023-12-22 9fdbf60165db0400c2e8e6be2dc6e88138ac719a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
U
®ý°dã@snddlZddlZddlZddlmmZddlZ    ddlm
Z
m Z ddl m ZGdd„dƒZGdd„dƒZdS)éN)ÚSeriesÚisnac@s4eZdZdd„Zej ddddddg¡d    d
„ƒZdS) Ú TestSeriesCovcCst | |¡| ¡d¡t |dd… |dd…¡|dd… ¡d¡t |ddd… |ddd…¡¡stt‚|dd… ¡}tj|dd…<t    | |¡ƒs¤t‚t    |dd…j|dd…ddƒsÊt‚|dd… 
|j ¡}|dd… 
|j ¡}t    |j|ddƒs
t‚dS)Nééééé
é ©Z min_periods) ÚtmÚassert_almost_equalÚcovZstdÚnpÚisnanÚAssertionErrorÚcopyÚnanrÚreindexÚindex)ÚselfÚdatetime_seriesÚcpÚts1Úts2©rú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_cov_corr.pyÚtest_covs 
ÿþ(&zTestSeriesCov.test_covÚ    test_ddofNrrrécCsbtj d¡}tj d¡}t|ƒ}t|ƒ}|j||d}tj|||ddd}t ||¡s^t‚dS)Nr    )Zddofrr)rÚrandomZrandrrÚmathÚiscloser)rrZ    np_array1Z    np_array2Ús1Ús2ÚresultÚexpectedrrrÚ test_cov_ddof,s  zTestSeriesCov.test_cov_ddof)Ú__name__Ú
__module__Ú __qualname__rÚpytestÚmarkZ parametrizer'rrrrrsrc@s8eZdZejdd„ƒZejdd„ƒZdd„Zdd„Zd    S)
ÚTestSeriesCorrc Cs6ddlm}t | |¡d¡t |dd… |dd…¡d¡t|dd…j|dd…ddƒsft‚|dd… |j¡}|dd… |j¡}t|j|ddƒs¤t‚t     
|ddd… |ddd…¡¡sÌt‚|dd    …  ¡}t    j |dd…<t| |¡ƒsüt‚t  ¡}t  ¡}| |¡}| ||¡\}    }
t ||    ¡dS)
Nr©Ústatsrrrr
r rr    )Úscipyr/r r ÚcorrrrrrrrrrÚmakeTimeSeriesZpearsonr) rrr/rrrÚAÚBr%r&Ú_rrrÚ    test_corr;s  "&(
zTestSeriesCorr.test_corrc Csôddlm}t ¡}t ¡}|dd…|dd…<|j|dd}| ||¡d}t ||¡|j|dd}| ||¡d}t ||¡tdd    d
d d d dddd    g
ƒ}tddddddddddg
ƒ}d}d}t |j|dd|¡t |j|dd|¡dS)Nrr.réûÿÿÿZkendall©ÚmethodZspearmang‰|U6ÅÆì¿g™~좦%î?gõÉab¹†ð¿gÝd    çŠî¿g+¸~œè?g    ù$ Wv°¿gàBG@¿ÆÀgü‚ÄÍÚ?g¯^EF4ð¿g»~¶@èã¿g§üsù¿g]’˜Úã?g\‚ßÈ4‘¿gñ\ÆêÂò?gˆ¦’$öð¿gH=ú!+º¿gIѱ  ì¿gE½Œb™ì?g)óêw¼¤Û?g•äÊêg»â?)    r0r/r r2r1Z
kendalltaur Z    spearmanrr)rr/r3r4r%r&ZkexpZsexprrrÚtest_corr_rankYsP   öÿöÿzTestSeriesCorr.test_corr_rankc    CsPttj d¡ƒ}ttj d¡ƒ}d}tjt|d|j|ddW5QRXdS)Nr    zGmethod must be either 'pearson', 'spearman', 'kendall', or a callable, )ÚmatchZ____r8)rrr Zrandnr+ZraisesÚ
ValueErrorr1)rr#r$ÚmsgrrrÚtest_corr_invalid_method‹s
z'TestSeriesCorr.test_corr_invalid_methodcCsòdd„}tdddddgƒ}tdddddgƒ}d}t |j||d    |¡t |j||d    d
¡t |dd …j|dd…|d    d
¡t |ddd…j|ddd…|d    ¡s®t‚t ||g¡}t d
dd œdd
d œg¡}t |     ¡j|d    |¡dS) NcSs||k ¡rdSdS)Nçð?g)Úall)ÚaÚbrrrÚ<lambda>–óz:TestSeriesCorr.test_corr_callable_method.<locals>.<lambda>rrrérrr8r?r)rr)
rr r r1rrrÚpdZ    DataFrameZ    transpose)rrZmy_corrr#r$r&ZdfrrrÚtest_corr_callable_method“s& ÿÿ ÿz(TestSeriesCorr.test_corr_callable_methodN)    r(r)r*ÚtdZskip_if_no_scipyr6r:r>rGrrrrr-:s 
 
1r-)r!Únumpyrr+Zpandas.util._test_decoratorsÚutilZ_test_decoratorsrHZpandasrFrrZpandas._testingZ_testingr rr-rrrrÚ<module>s *