h
? S SK rS SKrS SKJr S SKJrJrJrJ r J
r
S SK J r
S SKJr \R " \" SS/SS/SS/S S//SS /S
9\" SS/SS/SS/S S//SS/S
9\" SS/SS/SS/S S//S S /S
9\" SS/SS/SS/S S//S S /S
9\" SS/SS/SS/S S//S
S/S
9\" SS/SS/SS/S S//S S/S
9\" SS/S S/SS/SS //SS
/S
9\" SS/SS/SS/S S //S S /S
9\" SS/SS/SS/S S //S S/S
9/ S9S 5 r\R S 5 r\R S 5 rS rS rS r\R. R1 SSS /5 S 5 r\R. R1 S S S/5 S 5 r\R. R1 S \" S!5 5 S" 5 r S# r S$ r S% r \R. R1 S&S' S( /5 S) 5 r \R. R1 S&S* S+ /5 S, 5 r! " S- S.5 r"g)/ N)IS64) DataFrameIndex
MultiIndexSeries
date_range) safe_sort columnsC ?g @g @g @g @X)paramsc U R $ )z Pairwise frames test_pairwise)param)requests eC:\Users\psant\Downloads\Noiseback\school\venv\Lib\site-packages\pandas/tests/window/test_pairwise.pypairwise_framesr s == c 2 [ SS/SS/SS/SS//SS/S9$ )z'Pairwise target frame for test_pairwiser
r r r
r r r r r r pairwise_target_framer # s. q!fq!fq!fq!f51v FFr c 2 [ / SQ/ SQ/ SQ/ SQ// SQS9$ )z&Pairwise other frame for test_pairwise)Nr r )Nr r
)N r
)Nr r )YZr r r r r r pairwise_other_framer$ ) s |\<@ r c > U nU[ R R S5 R [ U5 5 - nUR SSS9R
U5 n[ R " UR S [ R " USS USS 5 S 5 g Nr
2 window min_periodsi)r r )
nprandom default_rngstandard_normallenrollingcovtmassert_almost_equalilocseriesABresults r test_rolling_covr< 2 s A BII ! !! $ 4 4SV < U nU[ R R S5 R [ U5 5 - nUR SSS9R
U5 n[ R " UR S [ R " USS USS 5 S 5 g r& ) r- r. r/ r0 r1 r2 corrr4 r5 r6 corrcoefr7 s r test_rolling_corrr@ : s A BII ! !! $ 4 4SV < r4 r5 r6 )abr; s r !test_rolling_corr_bias_correctionrP B s
"BJJ'z,PR/S A
A FFAbqE
VVAcrF
YYc!f!Y
4
9
9!
c 0 [ UR SSS9U 5 " 5 nUR [ S 5 S4S4 nUR R S5 Ul [ US R SSS9U 5 " US 5 n[ R " X#SS9 g )NrI r
r) r F check_names)getattrr2 locslicerH droplevelr4 assert_series_equal)rQ framer; expecteds r test_rolling_pairwise_cov_corrr\ O s
U]]"!] [ TR SS9T5 " U 5 $ )NrI r_ )rU r2 )xr] r8 s r (test_flex_binary_frame..^ s b(A6 J1 Mr r
rH r )rU r2 applyr4 assert_frame_equalrL r r- r. r/ r0 shaperH r ) r] rZ resres2expframe2res3kr8 s ` @r test_flex_binary_framern X s$
1XF
&...+V
4U
;C 5===+V 4V llF 5===+V 4V isnaall)r* srq s r $test_rolling_corr_with_zero_variancerv q s_ rxx| A 299R= !E 99F9 # ( (u ( 5 : : < @ @ B BB Br c [ [ R " SS/SS/SS/SS/S S
/S S //5 5 n U S
R SSS9R U S 5 n[ S U 5 5 ( d e[ [ R R S5 R
S5 5 n U S
R SSS9R U S 5 n[ S U 5 5 ( d eg )NgF$?g)?gT5?g?'N?gFr1?g+ C-?gR `?g1:b%?g9T!X4!?gXS_?gS= ?g\+Lt?r r
T)centerr c 3 | # U H2 n[ R " [ R " U5 5 S :* v M4 g7fr Nr- abs
nan_to_num.0ra s r #test_corr_sanity.. ' :crvvbmmA&'1,c :<r
) r
c 3 | # U H2 n[ R " [ R " U5 5 S :* v M4 g7frz r{ r~ s r r r r r )r r- arrayr2 r> rt r. r/ )dfrh s r test_corr_sanityr z s
Z(Y'Z(Z(Z(Z(
B
Q%--$-
'
,
,RU
3C :c: : :: : 299((+227; r4 rY r s r test_rolling_corr_diff_lengthr s +B Aq!f %B
ZZqaZ
0
5
5b
9F'(H6,
Y
/C
ZZqaZ
0
5
5c
:F6,r fc > U R SSS9R U SS9$ NrI r
r) Tpairwiser2 r3 ra s r rb rb ! 199BA96::1t:Lr c > U R SSS9R U SS9$ r r2 r> r s r rb rb ! 199BA96;;A;Mr c $ [ SS/SS/SS/SS//[ SS /S
S 9[ [ S 5 S
S 9S9n[ [ SS /S
S 9[ R " UR
UR /S
S
/S9SS9nU " U5 n[ R " X25 g )Nr r
r! r
r, r r9 r: foonamer barr rH namesrK )r rH rD ) r r ranger from_productrH r r4 rf )r r df_expected df_results r 2test_rolling_functions_window_non_shrinkage_binaryr s
Q!Q!Q"a)sCju-E!H5)
B
sCju-%%rxx& U R SSS9R U SS9$ r r r s r rb rb r r c > U R SSS9R U SS9$ r r r s r rb rb r r c [ 5 n[ [ S/SS9[ / SS9S9nUS R S5 US' [ [ R " UR
UR /5 S9n[ [ R " UR
UR /SS/S9[ S/SS9SS 9nU " U5 n[ R " XS5 U " U5 n[ R " Xd5 g )
NrN r r r r rK rG r )rH r rD ) r r astyper r rH r r4 rf )r df1df2 df1_expected df2_expected
df1_result
df2_results r *test_moment_functions_zero_length_pairwiser s +C
E3%e 4E"5uenUse%(L 3J*33J*3r c \ rS rSr\R
R
SS S /5 S 5 r\R
R
SS S S S S
S /5 S 5 r\R
R
SS
S S S S S /5 S 5 r \R
R
SS S S \R " S \R
R \ ( + SSS9S9S S /5 S 5 r
\R
R S 5 \R
R
SS S! S" S# S$ S% /5 S& 5 5 r\R
R
SS' S( S) S* S+ S, /5 S- 5 rS. rS/ rS0 rS1rg2)3 TestPairwise r c " U R 5 $ N)r3 r s r rb TestPairwise. s
QUUWr c " U R 5 $ r )r> r s r rb r s
r c b U" U5 n[ R " UR UR 5 [ R " UR UR 5 U" U5 nUR 5 R
nUR 5 R
n[ R " XESS9 g NF check_dtyper4 assert_index_equalrH r dropnavaluesassert_numpy_array_equalselfr r r r; r[ s r test_no_flexTestPairwise.test_no_flex s ?#
fllO,C,CD
fnno.E.EF*+ ''??$++
##F%Hr c < U R 5 R SS9$ NTr expandingr3 r s r rb r s akkm'''6r c < U R 5 R SS9$ r r r> r s r rb r s akkm(($(7r c : U R SS9R SS9$ Nr! r_ Tr r r s r rb r s aiiqi)--t- r s r rb r s aeeel'''6r c * U" U5 n[ R " UR R S UR SS9 [ R " [ UR R S 5 [ UR
R
5 5 5 [ R " UR
UR
5 U" U5 nUR 5 R nUR 5 R n[ R " XESS9 g Nr FrS r r
r4 r rH levelsr r uniquer r r r s r test_pairwise_with_self$TestPairwise.test_pairwise_with_self s ?#
LL "O$9$9u
fll))!, - o--446 7
fnno.E.EF*+ ''??$++
##F%Hr c < U R 5 R SS9$ NFr r r s r rb r s akkm'''7r c < U R 5 R SS9$ r r r s r rb r s akkm((%(8r c : U R SS9R SS9$ Nr! r_ Fr r r s r rb r s aiiqi)--u-=r c : U R SS9R SS9$ r r r s r rb r s aiiqi)...>r c : U R SS9R SS9$ Nr! r Fr r r s r rb r s aeeel&&&6r c : U R SS9R SS9$ r r r s r rb r s aeeel'''7r c b U" U5 n[ R " UR UR 5 [ R " UR UR 5 U" U5 nUR 5 R
nUR 5 R
n[ R " XESS9 g r r r s r test_no_pairwise_with_self'TestPairwise.test_no_pairwise_with_self
s ?#
fllO,A,AB
fnno.E.EF*+ ''??$++
##F%Hr c > U R 5 R USS9$ r r ra ys r rb r ( s **1t* U R 5 R USS9$ r r r s r rb r ) s ++A+=r c < U R SS9R USS9$ r r r s r rb r * s !,00T0Br c < U R SS9R USS9$ r r r s r rb r - s QYYaY 0 5 5a$ 5 Gr zPrecision issues on 32 bitF)reasonstrict)marksc < U R SS9R USS9$ r r r s r rb r 2 s 1 ))!d);r c < U R SS9R USS9$ r r r s r rb r 3 s 1 **1t* U R 5 R USS9$ r r r s r rb r N s **1u*=r c > U R 5 R USS9$ r r r s r rb r O s ++A+>r c < U R SS9R USS9$ r r r s r rb r P s !,00U0Cr c < U R SS9R USS9$ r r r s r rb r Q s !,11!e1Dr c < U R SS9R USS9$ r r r s r rb r R s 1 ))!e) U R SS9R U5 $ Nr! r_ r r s r rb r p s !,003r c > U R SS9R U5 $ r r r s r rb r q s !,11!4r c > U R SS9R U5 $ Nr! r r r s r rb r r s 1 ))!,r c > U R SS9R U5 $ r r r s r rb r s s 1 **1-r c U" U[ / SQ5 5 n[ R " UR UR 5 [ R " UR UR 5 U" U[ / SQ5 5 nUR 5 R nUR 5 R n[ R " XESS9 U" [ / SQ5 U5 n[ R " UR UR 5 [ R " UR UR 5 U" [ / SQ5 U5 nUR 5 R nUR 5 R n[ R " XESS9 g )N)r r r! r Fr )r r4 r rH r r r r r s r test_pairwise_with_series&TestPairwise.test_pairwise_with_seriesk s ?F<$89
fllO,A,AB
fnno.E.EF*F<,@A ''??$++
##F%H6,' 9
fllO,A,AB
fnno.E.EFVL)+@A ''??$++
##F%Hr c [ [ S5 [ SSS9S9nUR S5 R U5 n[ [
R /S- [ SSS9S9n[ R " X#5 g )Nr
2020rE rG 12h) r r r r2 r> r- rM r4 rY )r ru r; r[ s r test_corr_freq_memory_error(TestPairwise.test_corr_freq_memory_error s] 58:fa#@ A5!&&q)266(Q,j.KL
v0r c
[ R " [ S5 [ S5 [ S5 /5 n[ S5 n[ [
R " S5 R SS5 X!S9nUR SS 9R 5 n[ R " [ S5 [ S5 [ S5 [ S5 /5 n[ R " [ S5 [ S5 [ S5 /5 n[ [
R " [
R " S
[
R 5 [
R " S
S 5 [
R " S
S 5 45 UUS9n[ R " XE5 g )
NabxyABr! r rd g?)alpha)r r g @@gK\ǸO@)r r listr r r- rJ reshaper r3 vstackfullrM r4 rf )r r rH r r; r[ s r test_cov_mulittindex!TestPairwise.test_cov_mulittindex s ))4:tDz4:*NOa
ryy},,Q2%
Qc"&&(''q4:tDz4PT:(VW ))4:tDz4:*NO IIGGFBFF+GGFI.GGFI.
f/r c [ R " SS/SS//SS/S9n[ [ R " S5 US 9nUR S
5 R
5 n[ [ R [ R " [ R " [ R " S [ R S 9S
5 SS/S - SS/S - // SQS9US9n[ R " X45 g )NMNPQrN rO r )r
r
r r! r
rC r
)NrN rO rd )
r from_arraysr r- onesr2 r> rM repeatrJ int64r4 rf )r r r r; r[ s r $test_multindex_columns_pairwise_func1TestPairwise.test_multindex_columns_pairwise_func s ((3*sCj)A#sT
rwwv
8A ##% FF ((IIbii :A>#JN#JN
'
f/r r N)__name__
__module__ __qualname____firstlineno__r mark parametrizer r r r xfailr r filterwarningsr
r r r) r4 __static_attributes__r r r r r s [[ S#46H"IJ I K I [[ 6 7 < = 5 6
I
I* [[ 7 8 = > 6 7
I
I [[ < = B LLG kk'' H%A% (
< <
"I#"I( [[ 78 [[ = > C D < =
9
99* [[ - . 3 4 , -
I
I,1020r r )#numpyr- r
pandas.compatr pandasr r r r r pandas._testing_testingr4 pandas.core.algorithmsr fixturer r r$ r< r@ rP r: r; r\ rn r rv r r r r r r r r r rF s
, Aq6Aq6Aq6Aq62QFCAq6Aq6Aq6Aq62QFCAq6Aq6Aq6Aq62S#JGAq6Aq6Aq6Aq62S!HEAq6Aq6Aq6Aq62S!HEAq6Aq6Aq6Aq62S!HEC:SzC:SzBQPSHUAs8aX3x!S:QHMAq6Aq6Aq6As84sCjI
G G
LQ
7 % 1@ 2@ FE?3% 4%0 58,C -C;.
-
- MN
2
2 MN44&a0 a0r