Ë
    ýÿæiA  ã                   ód  — d dl Zd dlZd dlmZ d dlmZmZmZm	Z	m
Z
 d dlmZ d dlmZ  ej                    eddgddgddgd	dggdd g¬
«       eddgddgddgd	dggddg¬
«       eddgddgddgd	dggddg¬
«       eddgddgddgd	dggdd g¬
«       eddgddgddgd	dggddg¬
«       eddgddgddgd	dggddg¬
«       eddgddgddgddggddg¬
«       eddgddgddgd	dggd dg¬
«       eddgddgddgd	dggddg¬
«      g	¬«      d„ «       Zej                   d„ «       Zej                   d„ «       Zd„ Zd„ Zd„ Zej.                  j1                  dddg«      d„ «       Zej.                  j1                  dddg«      d„ «       Zej.                  j1                  d  ed!«      «      d"„ «       Zd#„ Zd$„ Zd%„ Zej.                  j1                  dddg«      d&„ «       Z ej.                  j1                  d'd(„ d)„ g«      d*„ «       Z!ej.                  j1                  d'd+„ d,„ g«      d-„ «       Z" G d.„ d/«      Z#y)0é    N)ÚIS64)Ú	DataFrameÚIndexÚ
MultiIndexÚSeriesÚ
date_range)Ú	safe_sorté   é   é   é   é   ©ÚcolumnsÚCç      ð?g        ç       @g      @g      @g       @ÚX)Úparamsc                 ó   — | j                   S )zPairwise frames test_pairwise)Úparam)Úrequests    úv/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pandas/tests/window/test_pairwise.pyÚpairwise_framesr      s   € ð �=‰=Ðó    c                  ó8   — t        ddgddgddgddggddg¬«      S )z'Pairwise target frame for test_pairwiser
   r   r   r   r   r   r   ©r   © r   r   Úpairwise_target_framer   #   s.   € ô �q˜!�f˜q !˜f q¨! f¨q°!¨fÐ5ÀÀ1¸vÔFÐFr   c                  ó8   — t        g d¢g d¢g d¢g d¢gg d¢¬«      S )z&Pairwise other frame for test_pairwise)Nr   r   )Nr   r
   )Né   r
   )Nr   r   )ÚYÚZr   r   r   r   r   r   Úpairwise_other_framer$   )   s    € ô Ú	’|¢\²<Ð@Úôð r   c           	      ó@  — | }|t         j                  j                  d«      j                  t	        |«      «      z   }|j                  dd¬«      j                  |«      }t        j                  |j                  d   t        j                  |dd  |dd  «      d   «       y ©Nr
   é2   é   ©ÚwindowÚmin_periodséÿÿÿÿiÎÿÿÿ)r   r   )
ÚnpÚrandomÚdefault_rngÚstandard_normalÚlenÚrollingÚcovÚtmÚassert_almost_equalÚiloc©ÚseriesÚAÚBÚresults       r   Útest_rolling_covr<   2   s€   € Ø€AØ	ŒB�I‰I×!Ñ! !Ó$×4Ñ4´S¸³VÓ<Ñ<€Aà�Y‰Y˜b¨bˆYÓ1×5Ñ5°aÓ8€FÜ×Ñ˜6Ÿ;™; r™?¬B¯F©F°1°S°T°7¸A¸c¸d¸GÓ,DÀTÑ,JÕKr   c           	      ó@  — | }|t         j                  j                  d«      j                  t	        |«      «      z   }|j                  dd¬«      j                  |«      }t        j                  |j                  d   t        j                  |dd  |dd  «      d   «       y r&   )r-   r.   r/   r0   r1   r2   Úcorrr4   r5   r6   Úcorrcoefr7   s       r   Útest_rolling_corrr@   :   s€   € Ø€AØ	ŒB�I‰I×!Ñ! !Ó$×4Ñ4´S¸³VÓ<Ñ<€Aà�Y‰Y˜b¨bˆYÓ1×6Ñ6°qÓ9€FÜ×Ñ˜6Ÿ;™; r™?¬B¯K©K¸¸#¸$¸ÀÀ3À4ÀÓ,IÈ$Ñ,OÕPr   c                  ó   — t        t        j                  dt        j                  ¬«      t	        dd¬«      ¬«      } | j                  «       }t        j                  | d d t        j                  |d d | j                  t        | «      d¬	«      j                  |«      }t        j                  |j                  d
   | j                  |«      «       y )Né   ©Údtypeú
2020-01-01©Úperiods©Úindexr   é
   r   r)   r,   )r   r-   ÚarangeÚfloat64r   ÚcopyÚnanr2   r1   r>   r4   r5   r6   )ÚaÚbr;   s      r   Ú!test_rolling_corr_bias_correctionrQ   B   s”   € äÜ
�	‰	�"œBŸJ™JÔ'¬z¸,ÐPRÔ/Sô	€Að 	
�‰‹€AÜ�F‰F€A€b€q€EÜ�V‰V€A€c€r€Fà�Y‰Yœc !›f°!ˆYÓ4×9Ñ9¸!Ó<€FÜ×Ñ˜6Ÿ;™; r™?¨A¯F©F°1«IÕ6r   Úfuncr3   r>   c                 óB  —  t        |j                  dd¬«      | «      «       }|j                  t        d «      dfdf   }|j                  j                  d«      |_         t        |d   j                  dd¬«      | «      |d   «      }t        j                  ||d¬«       y )NrJ   r   r)   r   F©Úcheck_names)Úgetattrr2   ÚlocÚslicerI   Ú	droplevelr4   Úassert_series_equal)rR   Úframer;   Úexpecteds       r   Útest_rolling_pairwise_cov_corrr]   O   s�   € àCŒW�U—]‘]¨"¸!�]Ó<¸dÓCÓE€FØ�Z‰Zœ˜t› aÐ(¨!Ð+Ñ,€FØ—<‘<×)Ñ)¨!Ó,€F„LØHŒw�u˜Q‘x×'Ñ'¨r¸qÐ'ÓAÀ4ÓHÈÈqÉÓR€HÜ×Ñ˜6 8¸Ö?r   Úmethodc                 ó  ‡ ‡— |d   Š t        ‰j                  d¬«      ‰ «      |«      } t        |j                  d¬«      ‰ «      ‰«      }|j                  ˆ ˆfd„«      }t        j                  ||«       t        j                  ||«       t        t        j                  j                  d«      j                  |j                  «      |j                  |j                  ¬«      } t        |j                  d¬«      ‰ «      |«      }t        t        |j                  «      «      |_        t        |D �ci c]+  }| t        ||   j                  d¬«      ‰ «      ||   «      “Œ- c}«      }t        j                  ||«       y c c}w )Nr   rJ   ©r*   c                 óH   •—  t        ‰j                  d¬«      ‰«      | «      S )NrJ   r`   )rV   r2   )Úxr^   r8   s    €€r   Ú<lambda>z(test_flex_binary_frame.<locals>.<lambda>^   s   ø€ Ð J¤¨¯©¸b¨Ó(AÀ6Ó JÈ1Ô Mr   r
   ©rI   r   )rV   r2   Úapplyr4   Úassert_frame_equalr   r-   r.   r/   r0   ÚshaperI   r   r   Úlist)	r^   r[   ÚresÚres2ÚexpÚframe2Úres3Úkr8   s	   `       @r   Útest_flex_binary_framero   X   sC  ù€ à�1‰X€Fà
4Œ'�&—.‘.¨�.Ó+¨VÓ
4°UÓ
;€CØ4Œ7�5—=‘=¨�=Ó+¨VÓ4°VÓ<€DØ
�+‰+ÔMÓ
N€Cä×Ñ˜#˜sÔ#Ü×Ñ˜$ Ô$äÜ
�	‰	×Ñ˜aÓ ×0Ñ0°·±Ó=Ø�k‰kØ—‘ô€Fð 5Œ7�5—=‘=¨�=Ó+¨VÓ4°VÓ<€DÜœ˜dŸl™lÓ+Ó,€D„LÜ
ÙMRÓSÉUÈˆÐ8ŒG�E˜!‘H×$Ñ$¨BÐ$Ó/°Ó8¸À¹ÓCÑ	CÈUÑSó€Cô ×Ñ˜$ Õ$ùò 	Ts   Ä30Fr*   é   c                 óü   — t        t        j                  d«      «      }t        t        j                  d«      «      }|j	                  | ¬«      j                  |¬«      j                  «       j                  «       sJ ‚y )NrB   r`   )Úother)r   r-   ÚzerosrK   r2   r>   ÚisnaÚall)r*   Úsrr   s      r   Ú$test_rolling_corr_with_zero_variancerw   q   s\   € ô 	Œr�x‰x˜‹|Ó€AÜ”2—9‘9˜R“=Ó!€Eà�9‰9˜Fˆ9Ó#×(Ñ(¨uÐ(Ó5×:Ñ:Ó<×@Ñ@ÔBÐBÑBr   c                  óÂ  — t        t        j                  ddgddgddgddgd	d
gddgg«      «      } | d   j                  dd¬«      j	                  | d   «      }t        d„ |D «       «      sJ ‚t        t        j                  j                  d«      j                  d«      «      } | d   j                  dd¬«      j	                  | d   «      }t        d„ |D «       «      sJ ‚y )Ng†F$ÈÙë?g)—Òé¯Ç?g‰ËT5ÿ—ä?g´?«'NÉÓ?gFr�1�í?g+ÿ”	C-à?gR	²±`?gÕ1:b%§?g9T!X4!ë?gXÀÊÂS_Õ?gS®= é?g\–+“Ltä?r   r   T)Úcenterr   c              3   ór   K  — | ]/  }t        j                  t        j                  |«      «      d k  –— Œ1 y­w©r   N©r-   ÚabsÚ
nan_to_num©Ú.0rb   s     r   Ú	<genexpr>z#test_corr_sanity.<locals>.<genexpr>Š   ó(   è ø€ Ð:±c°Œr�v‰v”b—m‘m AÓ&Ó'¨1Õ,±cùó   ‚57r
   )é   r
   c              3   ór   K  — | ]/  }t        j                  t        j                  |«      «      d k  –— Œ1 y­wr{   r|   r   s     r   r�   z#test_corr_sanity.<locals>.<genexpr>Ž   r‚   rƒ   )r   r-   Úarrayr2   r>   ru   r.   r/   )Údfri   s     r   Útest_corr_sanityrˆ   z   så   € ä	Ü
�‰à˜ZÐ(Ø˜YÐ'Ø˜ZÐ(Ø˜ZÐ(Ø˜ZÐ(Ø˜ZÐ(ðó		
ó
€Bð ˆQ‰%�-‰-˜ $ˆ-Ó
'×
,Ñ
,¨R°©UÓ
3€CÜÑ:±cÓ:Ô:Ð:Ð:ä	”2—9‘9×(Ñ(¨Ó+×2Ñ2°7Ó;Ó	<€BØ
ˆQ‰%�-‰-˜ $ˆ-Ó
'×
,Ñ
,¨R°©UÓ
3€CÜÑ:±cÓ:Ô:Ð:Ñ:r   c                  ó’  — t        g d¢t        d«      ¬«      } t        ddgt        ddd«      ¬«      }| j                  dd¬«      j                  |«      }t        g d	¢«      }t	        j
                  ||«       t        g d
¢t        d«      ¬«      }| j                  dd¬«      j                  |«      }t	        j
                  ||«       y )N©r   r
   r!   r!   rH   r   r   r   r
   r)   )NNr   ©r   Nr!   )r   Úranger2   r3   r4   rZ   ©Ús1Ús2r;   r\   Ús2as        r   Útest_rolling_cov_diff_lengthr‘   ‘   sŸ   € ä	’	¤ q£Ô	*€BÜ	��A�œe A q¨!›nÔ	-€BØ�Z‰Z˜q¨aˆZÓ0×4Ñ4°RÓ8€FÜÒ'Ó(€HÜ×Ñ˜6 8Ô,ä
’¤U¨1£XÔ
.€CØ�Z‰Z˜q¨aˆZÓ0×4Ñ4°SÓ9€FÜ×Ñ˜6 8Õ,r   c                  ó’  — t        g d¢t        d«      ¬«      } t        ddgt        ddd«      ¬«      }| j                  dd¬«      j                  |«      }t        g d	¢«      }t	        j
                  ||«       t        g d
¢t        d«      ¬«      }| j                  dd¬«      j                  |«      }t	        j
                  ||«       y )NrŠ   r!   rH   r   r   r   r
   r)   )NNr   r‹   )r   rŒ   r2   r>   r4   rZ   r�   s        r   Útest_rolling_corr_diff_lengthr“   ž   sŸ   € ä	’	¤ q£Ô	*€BÜ	��A�œe A q¨!›nÔ	-€BØ�Z‰Z˜q¨aˆZÓ0×5Ñ5°bÓ9€FÜÒ'Ó(€HÜ×Ñ˜6 8Ô,ä
’¤U¨1£XÔ
.€CØ�Z‰Z˜q¨aˆZÓ0×5Ñ5°cÓ:€FÜ×Ñ˜6 8Õ,r   c                 ó  — t        ddd¬«      }t        ddd¬«      }t        g d¢|¬«      }t        g d¢|¬«      }d	}t        j                  t        |¬
«      5   t        |j                  d«      | «      |«       d d d «       y # 1 sw Y   y xY w)NrE   r!   ÚD)rG   Úfreqr   rŠ   rH   )r   r
   r!   r   r   z-Variable rolling window requires .* Got 3 < 5©ÚmatchÚ2D)r   r   ÚpytestÚraisesÚ
ValueErrorrV   r2   )rR   Ú	idx_shortÚidx_longrv   rr   Úmsgs         r   Ú+test_time_based_rolling_other_longer_raisesr    «   sp   € ô ˜<°¸Ô=€IÜ˜,°¸Ô<€HÜŠy 	Ô*€AÜ’?¨(Ô3€EØ
9€CÜ	�‰”z¨Ö	-Ø&Œ�—	‘	˜$“ Ó& uÔ-÷ 
.×	-Ñ	-ús   Á"BÂBÚfc                 óJ   — | j                  dd¬«      j                  | d¬«      S ©NrJ   r   r)   T©Úpairwise©r2   r3   ©rb   s    r   rc   rc   º   ó!   € �!—)‘) 2°1�)Ó5×9Ñ9¸!ÀdÐ9ÔKr   c                 óJ   — | j                  dd¬«      j                  | d¬«      S r£   ©r2   r>   r§   s    r   rc   rc   »   ó!   € �!—)‘) 2°1�)Ó5×:Ñ:¸1ÀtÐ:ÔLr   c           	      óF  — t        ddgddgddgddggt        dd	gd
¬«      t        t        d«      d¬«      ¬«      }t        t        dd	gd
¬«      t        j                  |j
                  |j                  gdd
g¬«      d¬«      } | |«      }t        j                  ||«       y )Nr   r   r!   r
   é	   r,   r   r9   r:   Úfoo©Únamer   Úbar©r   rI   ©ÚnamesrL   )r   rI   rD   )	r   r   rŒ   r   Úfrom_productrI   r   r4   rf   )r¡   r‡   Údf_expectedÚ	df_results       r   Ú2test_rolling_functions_window_non_shrinkage_binaryr¸   ·   sŸ   € ô 
Ø
ˆQˆ�!�Q�˜!˜Q˜ " a Ð)Ü�s˜C�j uÔ-Ü”E˜!“H 5Ô)ô
€Bô
 Ü�s˜C�j uÔ-Ü×%Ñ% r§x¡x°·±Ð&<ÀUÈEÀNÔSØô€Kñ
 �"“€IÜ×Ñ˜) [Õ1r   c                 óJ   — | j                  dd¬«      j                  | d¬«      S r£   r¦   r§   s    r   rc   rc   Ñ   r¨   r   c                 óJ   — | j                  dd¬«      j                  | d¬«      S r£   rª   r§   s    r   rc   rc   Ò   r«   r   c                 óþ  — t        «       }t        t        dgd¬«      t        g d¬«      ¬«      }|d   j                  d«      |d<   t        t        j                  |j
                  |j                  g«      ¬«      }t        t        j                  |j
                  |j                  gddg¬«      t        dgd¬«      d¬	«      } | |«      }t        j                  ||«        | |«      }t        j                  ||«       y )
NrO   r®   r¯   r±   r²   rL   rH   r³   )rI   r   rD   )	r   r   Úastyper   rµ   rI   r   r4   rf   )r¡   Údf1Údf2Údf1_expectedÚdf2_expectedÚ
df1_resultÚ
df2_results          r   Ú*test_moment_functions_zero_length_pairwiserÃ   Î   sÍ   € ô ‹+€CÜ
œE 3 %¨eÔ4¼EÀ"È5Ô<QÔ
R€CØ�3‰x�‰˜yÓ)€Cˆ�Hä¤:×#:Ñ#:¸C¿I¹IÀsÇ{Á{Ð;SÓ#TÔU€LÜÜ×%Ñ% s§y¡y°#·+±+Ð&>ÀuÈeÀnÔUÜ�s�e %Ô(Øô€Lñ �3“€JÜ×Ñ˜* lÔ3á�3“€JÜ×Ñ˜* lÕ3r   c                   óÐ  — e Zd Zej                  j                  dd„ d„ g«      d„ «       Zej                  j                  dd„ d„ d„ d„ d	„ d
„ g«      d„ «       Zej                  j                  dd„ d„ d„ d„ d„ d„ g«      d„ «       Zej                  j                  dd„ d„ d„  ej                  d„ ej                  j                  e dd¬«      ¬«      d„ d„ g«      d„ «       Zej                  j                  d«      ej                  j                  dd„ d „ d!„ d"„ d#„ d$„ g«      d%„ «       «       Zej                  j                  dd&„ d'„ d(„ d)„ d*„ d+„ g«      d,„ «       Zd-„ Zd.„ Zd/„ Zy0)1ÚTestPairwiser¡   c                 ó"   — | j                  «       S ©N)r3   r§   s    r   rc   zTestPairwise.<lambda>ê   s
   € ¨Q¯U©U¬Wr   c                 ó"   — | j                  «       S rÇ   )r>   r§   s    r   rc   zTestPairwise.<lambda>ê   s
   € ÀÇÁÄr   c                 ód  —  ||«      }t        j                  |j                  |j                  «       t        j                  |j                  |j                  «        ||«      }|j	                  «       j
                  }|j	                  «       j
                  }t        j                  ||d¬«       y ©NF©Úcheck_dtype©r4   Úassert_index_equalrI   r   ÚdropnaÚvaluesÚassert_numpy_array_equal©Úselfr   r   r¡   r;   r\   s         r   Útest_no_flexzTestPairwise.test_no_flexê   s„   € ñ �?Ó#ˆÜ
×Ñ˜fŸl™l¨O×,CÑ,CÔDÜ
×Ñ˜fŸn™n¨o×.EÑ.EÔFÙÐ*Ó+ˆð —‘“×'Ñ'ˆØ—?‘?Ó$×+Ñ+ˆä
×#Ñ# F¨HÀ%ÖHr   c                 óB   — | j                  «       j                  d¬«      S ©NTr¤   ©Ú	expandingr3   r§   s    r   rc   zTestPairwise.<lambda>ü   s   € �a—k‘k“m×'Ñ'°Ð'Ô6r   c                 óB   — | j                  «       j                  d¬«      S rÖ   ©rØ   r>   r§   s    r   rc   zTestPairwise.<lambda>ý   s   € �a—k‘k“m×(Ñ(°$Ð(Ô7r   c                 óF   — | j                  d¬«      j                  d¬«      S ©Nr!   r`   Tr¤   r¦   r§   s    r   rc   zTestPairwise.<lambda>þ   s   € �a—i‘i q�iÓ)×-Ñ-°tÐ-Ô<r   c                 óF   — | j                  d¬«      j                  d¬«      S rÜ   rª   r§   s    r   rc   zTestPairwise.<lambda>ÿ   s   € �a—i‘i q�iÓ)×.Ñ.¸Ð.Ô=r   c                 óF   — | j                  d¬«      j                  d¬«      S ©Nr!   ©ÚcomTr¤   ©Úewmr3   r§   s    r   rc   zTestPairwise.<lambda>   s   € �a—e‘e �e“l×&Ñ&°Ð&Ô5r   c                 óF   — | j                  d¬«      j                  d¬«      S rß   ©rã   r>   r§   s    r   rc   zTestPairwise.<lambda>  s   € �a—e‘e �e“l×'Ñ'°Ð'Ô6r   c                 ó0  —  ||«      }t        j                  |j                  j                  d   |j                  d¬«       t        j                  t	        |j                  j                  d   «      t	        |j
                  j                  «       «      «       t        j                  |j
                  |j
                  «        ||«      }|j                  «       j                  }|j                  «       j                  }t        j                  ||d¬«       y ©Nr   FrT   r   rË   ©
r4   rÎ   rI   Úlevelsr	   r   ÚuniquerÏ   rÐ   rÑ   rÒ   s         r   Útest_pairwise_with_selfz$TestPairwise.test_pairwise_with_selfù   sÔ   € ñ �?Ó#ˆÜ
×ÑØ�L‰L×Ñ Ñ" O×$9Ñ$9Àuõ	
ô 	×ÑÜ�f—l‘l×)Ñ)¨!Ñ,Ó-Ü�o×-Ñ-×4Ñ4Ó6Ó7ô	
ô 	×Ñ˜fŸn™n¨o×.EÑ.EÔFÙÐ*Ó+ˆð —‘“×'Ñ'ˆØ—?‘?Ó$×+Ñ+ˆä
×#Ñ# F¨HÀ%ÖHr   c                 óB   — | j                  «       j                  d¬«      S ©NFr¤   r×   r§   s    r   rc   zTestPairwise.<lambda>  s   € �a—k‘k“m×'Ñ'°Ð'Ô7r   c                 óB   — | j                  «       j                  d¬«      S rí   rÚ   r§   s    r   rc   zTestPairwise.<lambda>  s   € �a—k‘k“m×(Ñ(°%Ð(Ô8r   c                 óF   — | j                  d¬«      j                  d¬«      S ©Nr!   r`   Fr¤   r¦   r§   s    r   rc   zTestPairwise.<lambda>  s   € �a—i‘i q�iÓ)×-Ñ-°uÐ-Ô=r   c                 óF   — | j                  d¬«      j                  d¬«      S rð   rª   r§   s    r   rc   zTestPairwise.<lambda>  s   € �a—i‘i q�iÓ)×.Ñ.¸Ð.Ô>r   c                 óF   — | j                  d¬«      j                  d¬«      S ©Nr!   rà   Fr¤   râ   r§   s    r   rc   zTestPairwise.<lambda>   s   € �a—e‘e �e“l×&Ñ&°Ð&Ô6r   c                 óF   — | j                  d¬«      j                  d¬«      S ró   rå   r§   s    r   rc   zTestPairwise.<lambda>!  s   € �a—e‘e �e“l×'Ñ'°Ð'Ô7r   c                 ód  —  ||«      }t        j                  |j                  |j                  «       t        j                  |j                  |j                  «        ||«      }|j	                  «       j
                  }|j	                  «       j
                  }t        j                  ||d¬«       y rÊ   rÍ   rÒ   s         r   Útest_no_pairwise_with_selfz'TestPairwise.test_no_pairwise_with_self  s„   € ñ �?Ó#ˆÜ
×Ñ˜fŸl™l¨O×,AÑ,AÔBÜ
×Ñ˜fŸn™n¨o×.EÑ.EÔFÙÐ*Ó+ˆð —‘“×'Ñ'ˆØ—?‘?Ó$×+Ñ+ˆä
×#Ñ# F¨HÀ%ÖHr   c                 óD   — | j                  «       j                  |d¬«      S rÖ   r×   ©rb   Úys     r   rc   zTestPairwise.<lambda>4  s   € ˜Ÿ™›×*Ñ*¨1°tÐ*Ô<r   c                 óD   — | j                  «       j                  |d¬«      S rÖ   rÚ   rø   s     r   rc   zTestPairwise.<lambda>5  s   € ˜Ÿ™›×+Ñ+¨A¸Ð+Ô=r   c                 óH   — | j                  d¬«      j                  |d¬«      S rÜ   r¦   rø   s     r   rc   zTestPairwise.<lambda>6  s   € ˜Ÿ™¨!˜Ó,×0Ñ0°¸TÐ0ÔBr   c                 óH   — | j                  d¬«      j                  |d¬«      S rÜ   rª   rø   s     r   rc   zTestPairwise.<lambda>9  s   € ˜QŸY™Y¨a˜YÓ0×5Ñ5°aÀ$Ð5ÔGr   zPrecision issues on 32 bitF)ÚreasonÚstrict)Úmarksc                 óH   — | j                  d¬«      j                  |d¬«      S rß   râ   rø   s     r   rc   zTestPairwise.<lambda>>  s   € ˜Ÿ™ 1˜›×)Ñ)¨!°dÐ)Ô;r   c                 óH   — | j                  d¬«      j                  |d¬«      S rß   rå   rø   s     r   rc   zTestPairwise.<lambda>?  s   € ˜Ÿ™ 1˜›×*Ñ*¨1°tÐ*Ô<r   c                 óà  —  |||«      }t        j                  |j                  j                  d   |j                  d¬«       t        j                  t	        |j                  j                  d   «      t	        |j
                  j                  «       «      «        |||«      }|j                  «       j                  }|j                  «       j                  }t        j                  ||d¬«       y rç   rè   )rÓ   r   r   r$   r¡   r;   r\   s          r   Útest_pairwise_with_otherz%TestPairwise.test_pairwise_with_other1  sÁ   € ñ* �?Ð$8Ó9ˆÜ
×ÑØ�L‰L×Ñ Ñ" O×$9Ñ$9Àuõ	
ô 	×ÑÜ�f—l‘l×)Ñ)¨!Ñ,Ó-ÜÐ*×2Ñ2×9Ñ9Ó;Ó<ô	
ñ Ð*Ð,@ÓAˆð —‘“×'Ñ'ˆØ—?‘?Ó$×+Ñ+ˆä
×#Ñ# F¨HÀ%ÖHr   zignore:RuntimeWarningc                 óD   — | j                  «       j                  |d¬«      S rí   r×   rø   s     r   rc   zTestPairwise.<lambda>Z  s   € ˜Ÿ™›×*Ñ*¨1°uÐ*Ô=r   c                 óD   — | j                  «       j                  |d¬«      S rí   rÚ   rø   s     r   rc   zTestPairwise.<lambda>[  s   € ˜Ÿ™›×+Ñ+¨A¸Ð+Ô>r   c                 óH   — | j                  d¬«      j                  |d¬«      S rð   r¦   rø   s     r   rc   zTestPairwise.<lambda>\  s   € ˜Ÿ™¨!˜Ó,×0Ñ0°¸UÐ0ÔCr   c                 óH   — | j                  d¬«      j                  |d¬«      S rð   rª   rø   s     r   rc   zTestPairwise.<lambda>]  s   € ˜Ÿ™¨!˜Ó,×1Ñ1°!¸eÐ1ÔDr   c                 óH   — | j                  d¬«      j                  |d¬«      S ró   râ   rø   s     r   rc   zTestPairwise.<lambda>^  s   € ˜Ÿ™ 1˜›×)Ñ)¨!°eÐ)Ô<r   c                 óH   — | j                  d¬«      j                  |d¬«      S ró   rå   rø   s     r   rc   zTestPairwise.<lambda>_  s   € ˜Ÿ™ 1˜›×*Ñ*¨1°uÐ*Ô=r   c                 óD  — |j                   j                  r	 |||«      nd }|�‹|j                  j                  |j                  «      }|j                   j                  |j                   «      }t	        j
                  |j                  |«       t	        j
                  |j                   |«       y t        j                  t        d¬«      5   |||«       d d d «       t        j                  t        d¬«      5   |||«       d d d «       y # 1 sw Y   Œ7xY w# 1 sw Y   y xY w)Nz'arg1' columns are not uniquer—   z'arg2' columns are not unique)	r   Ú	is_uniquerI   Úunionr4   rÎ   rš   r›   rœ   )rÓ   r   r$   r¡   r;   Úexpected_indexÚexpected_columnss          r   Útest_no_pairwise_with_otherz(TestPairwise.test_no_pairwise_with_otherV  sì   € ð  ×&Ñ&×0Ò0ñ ˆoÐ3Ô4àð 	ð
 Ðà,×2Ñ2×8Ñ8Ð9M×9SÑ9SÓTˆNØ.×6Ñ6×<Ñ<Ø$×,Ñ,ó Ðô ×!Ñ! &§,¡,°Ô?Ü×!Ñ! &§.¡.Ð2BÕCä—‘œzÐ1PÖQÙ�/Ð#7Ô8÷ Rä—‘œzÐ1PÖQÙÐ&¨Ô8÷ RÐQ÷ RÐQúçQÐQús   Ã

D
Ã7
DÄ
DÄDc                 ó@   — | j                  «       j                  |«      S rÇ   r×   rø   s     r   rc   zTestPairwise.<lambda>z  s   € ˜Ÿ™›×*Ñ*¨1Ô-r   c                 ó@   — | j                  «       j                  |«      S rÇ   rÚ   rø   s     r   rc   zTestPairwise.<lambda>{  s   € ˜Ÿ™›×+Ñ+¨AÔ.r   c                 óD   — | j                  d¬«      j                  |«      S ©Nr!   r`   r¦   rø   s     r   rc   zTestPairwise.<lambda>|  s   € ˜Ÿ™¨!˜Ó,×0Ñ0°Ô3r   c                 óD   — | j                  d¬«      j                  |«      S r  rª   rø   s     r   rc   zTestPairwise.<lambda>}  s   € ˜Ÿ™¨!˜Ó,×1Ñ1°!Ô4r   c                 óD   — | j                  d¬«      j                  |«      S ©Nr!   rà   râ   rø   s     r   rc   zTestPairwise.<lambda>~  s   € ˜Ÿ™ 1˜›×)Ñ)¨!Ô,r   c                 óD   — | j                  d¬«      j                  |«      S r  rå   rø   s     r   rc   zTestPairwise.<lambda>  s   € ˜Ÿ™ 1˜›×*Ñ*¨1Ô-r   c                 ó$  —  ||t        g d¢«      «      }t        j                  |j                  |j                  «       t        j                  |j                  |j                  «        ||t        g d¢«      «      }|j                  «       j                  }|j                  «       j                  }t        j                  ||d¬«        |t        g d¢«      |«      }t        j                  |j                  |j                  «       t        j                  |j                  |j                  «        |t        g d¢«      |«      }|j                  «       j                  }|j                  «       j                  }t        j                  ||d¬«       y )N)r   r   r!   r   FrË   )r   r4   rÎ   rI   r   rÏ   rÐ   rÑ   rÒ   s         r   Útest_pairwise_with_seriesz&TestPairwise.test_pairwise_with_seriesw  s   € ñ �?¤Fª<Ó$8Ó9ˆÜ
×Ñ˜fŸl™l¨O×,AÑ,AÔBÜ
×Ñ˜fŸn™n¨o×.EÑ.EÔFÙÐ*¬F²<Ó,@ÓAˆð —‘“×'Ñ'ˆØ—?‘?Ó$×+Ñ+ˆÜ
×#Ñ# F¨HÀ%ÕHá”6š,Ó'¨Ó9ˆÜ
×Ñ˜fŸl™l¨O×,AÑ,AÔBÜ
×Ñ˜fŸn™n¨o×.EÑ.EÔFÙ”VšLÓ)Ð+@ÓAˆð —‘“×'Ñ'ˆØ—?‘?Ó$×+Ñ+ˆÜ
×#Ñ# F¨HÀ%ÖHr   c                 ó  — t        t        d«      t        dd¬«      ¬«      }|j                  d«      j	                  |«      }t        t
        j                  gdz  t        dd¬«      ¬«      }t        j                  ||«       y )Nr   Ú2020rF   rH   Ú12h)	r   rŒ   r   r2   r>   r-   rN   r4   rZ   )rÓ   rv   r;   r\   s       r   Útest_corr_freq_memory_errorz(TestPairwise.test_corr_freq_memory_error˜  s_   € ä”5˜“8¤:¨f¸aÔ#@ÔAˆØ—‘˜5Ó!×&Ñ& qÓ)ˆÜœ2Ÿ6™6˜( Q™,¬j¸ÈÔ.KÔLˆÜ
×Ñ˜v xÕ0r   c           
      ó
  — t        j                  t        d«      t        d«      t        d«      g«      }t        d«      }t	        t        j                  d«      j                  dd«      ||¬«      }|j                  d¬	«      j                  «       }t        j                  t        d«      t        d«      t        d«      t        d«      g«      }t        j                  t        d«      t        d«      t        d«      g«      }t	        t        j                  t        j                  d
t
        j                  «      t        j                  d
d«      t        j                  d
d«      f«      ||¬«      }t        j                  ||«       y )NÚabÚxyÚABr!   é   r   rd   gš™™™™™¹?)Úalpha)r   r   g      @@gK\Ç¸âðO@)r   rµ   rh   rŒ   r   r-   rK   Úreshaperã   r3   ÚvstackÚfullrN   r4   rf   )rÓ   r   rI   r‡   r;   r\   s         r   Útest_cov_mulittindexz!TestPairwise.test_cov_mulittindexŸ  s  € ô ×)Ñ)¬4°«:´t¸D³zÄ4ÈÃ:Ð*NÓOˆÜ�a“ˆÜ”r—y‘y “}×,Ñ,¨Q°Ó2¸%ÈÔQˆà—‘˜c�Ó"×&Ñ&Ó(ˆä×'Ñ'¬¨q«´4¸³:¼tÀD»zÌ4ÐPTË:Ð(VÓWˆÜ×)Ñ)¬4°«:´t¸D³zÄ4ÈÃ:Ð*NÓOˆÜÜ�I‰Iä—G‘G˜F¤B§F¡FÓ+Ü—G‘G˜F IÓ.Ü—G‘G˜F IÓ.ðóð Øô

ˆô 	×Ñ˜f hÕ/r   c                 óÚ  — t        j                  ddgddggddg¬«      }t        t        j                  d«      |¬	«      }|j                  d
«      j                  «       }t        t        j                  t        j                  t        j                  t        j                  dt        j                  ¬«      d«      ddgdz  ddgdz  gg d¢¬«      |¬«      }t        j                  ||«       y )NÚMÚNÚPÚQrO   rP   r³   )r   r
   r   r!   r   rC   r
   )NrO   rP   rd   )r   Úfrom_arraysr   r-   Úonesr2   r>   rN   ÚrepeatrK   Úint64r4   rf   )rÓ   r   r‡   r;   r\   s        r   Ú$test_multindex_columns_pairwise_funcz1TestPairwise.test_multindex_columns_pairwise_func¸  sÂ   € ä×(Ñ(¨3°¨*°s¸C°jÐ)AÈ#ÈsÈÔTˆÜ”r—w‘w˜v“°Ô8ˆØ—‘˜A“×#Ñ#Ó%ˆÜÜ�F‰FÜ×(Ñ(ä—I‘IœbŸi™i¨´·±Ô:¸AÓ>Ø˜#�J ‘NØ˜#�J ‘Nðò
 'ôð ô
ˆô 	×Ñ˜f hÕ/r   N)Ú__name__Ú
__module__Ú__qualname__rš   ÚmarkÚparametrizerÔ   rë   rö   r   Úxfailr   r  Úfilterwarningsr  r  r  r'  r1  r   r   r   rÅ   rÅ   è   sÀ  „ à‡[�[×Ñ˜SÑ#4Ñ6HÐ"IÓJñIó KðIð ‡[�[×ÑØá6Ù7Ù<Ù=Ù5Ù6ð	
ó
ñIó
ðIð* ‡[�[×ÑØá7Ù8Ù=Ù>Ù6Ù7ð	
ó
ñIó
ðIð ‡[�[×ÑØá<Ù=ÙBàˆF�L‰LÙGØ—k‘k×'Ñ'Ø�HÐ%AÈ%ð (ó ôñ <Ù<ð	
óñ"Ió#ð"Ið( ‡[�[×ÑÐ 7Ó8Ø‡[�[×ÑØá=Ù>ÙCÙDÙ<Ù=ð	
ó
ñ9ó
ó 9ð9ð* ‡[�[×ÑØá-Ù.Ù3Ù4Ù,Ù-ð	
ó
ñIó
ðIò,1ò0ó20r   rÅ   )$Únumpyr-   rš   Úpandas.compatr   Úpandasr   r   r   r   r   Úpandas._testingÚ_testingr4   Úpandas.core.algorithmsr	   Úfixturer   r   r$   r<   r@   rQ   r5  r6  r]   ro   rŒ   rw   rˆ   r‘   r“   r    r¸   rÃ   rÅ   r   r   r   Ú<module>r@     s  ðÛ Û å ÷õ õ Ý ,ð €‡�á�A�q�6˜A˜q˜6 A q 6¨A¨q¨6Ð2¸QÀ¸FÔCÙ�A�q�6˜A˜q˜6 A q 6¨A¨q¨6Ð2¸QÀ¸FÔCÙ�A�q�6˜A˜q˜6 A q 6¨A¨q¨6Ð2¸SÀ#¸JÔGÙ�A�q�6˜A˜q˜6 A q 6¨A¨q¨6Ð2¸SÀ!¸HÔEÙ�A�q�6˜A˜q˜6 A q 6¨A¨q¨6Ð2¸SÀ!¸HÔEÙ�A�q�6˜A˜q˜6 A q 6¨A¨q¨6Ð2¸SÀ!¸HÔEÙ�C˜�:  S˜z¨C°¨:¸¸S°zÐBÈQÐPSÈHÔUÙ�A�s�8˜a ˜X¨¨3 x°!°S°Ð:ÀQÈÀHÔMÙ�A�q�6˜A˜q˜6 A q 6¨A¨s¨8Ð4¸sÀC¸jÔIð
ôñóðð
 ‡�ñGó ðGð
 ‡�ñó ðòLòQò
7ð ‡�×Ñ˜ %¨ Ó1ñ@ó 2ð@ð ‡�×Ñ˜ F¨E ?Ó3ñ%ó 4ð%ð0 ‡�×Ñ˜¡5¨£8Ó,ñCó -ðCò;ò.
-ò
-ð ‡�×Ñ˜ %¨ Ó1ñ.ó 2ð.ð ‡�×ÑØáKÙLðóñ2óð2ð  ‡�×ÑØáKÙLðóñ4óð4÷&a0ò a0r   