
    bMh^                        d dl Zd dlZd dlmZmZmZmZmZm	Z	m
Z
 d dlmZ d Zd Zej                            ddd ej        dg          g          d	             Zej                            d
d ej        dej                            d                    g          d             Zd Zd Zd Zd Zd Zej                            d eg dg dd          dgdgdd gfddgddgdd dgfg dg ddg dfgdf eg dg dd          dgdgdd gfddgddgdd dgfg dg ddg dfgdf eg dg dd          dgdgdd gfddgddgdd dgfg dg ddg dfgdf edgdgd          g df e            i g fgdf edej        dgej        ddgd          dgej        gdd gfdej        gej        dgdd dgfdej        dgej        ddgdg dfgdf edej        dgej        ddgd          dgej        gdd gfdej        gej        dgdd dgfdej        dgej        ddgdg dfgdf edej        dgej        ddgd          dgej        gdd gfdej        gej        dgdd dgfdej        dgej        ddgdg dfgdfg          d              Zej                            d! eg d          dgd gfddgd dgfg dg dfgdf eg d          dgd gfddgd dgfg dg dfgdf eg d          dgd gfddgd dgfg dg dfgdf eddg          dgd gfddgd dgfgdf eej        dg          ej        gd gfej        dgd dgfgdf eg d"#          g dfg          d$             Zd% Zd& Z ej                            d'd(d)g          d*             Z!ej                            d+g d,          ej                            d'g d-          ej                            d.d/d0g          ej                            d1d/d0g          ej                            d2g d3          d4                                                             Z"d5 Z#d6 Z$d7 Z%d8 Z&d9 Z'd: Z(ej                            d;d<ej)        fd=d> fd?d@ fdAdB fgg dCD          dE             Z*ej                            d;d<ej)        fd=ej+        fd?ej,        fdAej-        fgg dCD          dF             Z.dG Z/dH Z0ej                            dIdJ dK g          dL             Z1ej                            dIdM dN dO dP dQ dR dS dT dU dV dW dX dY dZ d[ g          d\             Z2d] Z3d^ Z4d_ Z5d` Z6da Z7db Z8dc Z9dd Z:ej                            dedfdgg          ej                            dhd/d0g          di                         Z;ej                            dje<e=g          dk             Z>ej                            dedfdgg          ej                            dhd/d0g          ej                            dje<e=g          dl                                     Z?dm Z@dS )n    N)	DataFrameDatetimeIndexIndex
MultiIndexSeriesisnanotnac                      t          ddddt          j        dgi          } |  |                     d                                           d S )NBr            )r   npnan	expandingsumdfs    b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/window/test_expanding.pytest_doc_stringr      sH    	C!Q261-.	/	/BBLLOO    c                 X     | t          d                    j        } |d           d S )N   r   min_periods)ranger   )frame_or_seriescs     r   test_constructorr      s7     	a!!+A A!r   w       @foor   c                      | t          d                    j        }d}t          j        t          |          5   ||           d d d            d S # 1 swxY w Y   d S )Nr   zmin_periods must be an integermatchr   )r   r   pytestraises
ValueError)r   r    r   msgs       r   test_constructor_invalidr*      s     	a!!+A
*C	z	-	-	-  	a                 s   AAAexpanderr   lsz,GH#16425 expanding with offset not supported)reason)marksc                    t                      }t                                          |                                           }t          j        ||           t          t          g                     }t          t          g                                         |                                           }t          j        ||           d S )Nindex)r   r   r   tmassert_frame_equalr   )r+   expectedresults      r   test_empty_df_expandingr6   )   s      {{H[[""8,,0022F&(+++ }R00111H]2..///99(CCGGIIF&(+++++r   c                     t          t          j        g          } |                     d                                          }t          dg          }t          j        ||           |                     d                                          }t          t          j        g          }t          j        ||           d S )Nr   r   g        r   )r   r   r   r   r   r2   assert_series_equal)xr5   r4   s      r   test_missing_minp_zeror:   D   s     	xA[[Q[''++--Fse}}H68,,, [[Q[''++--FrvhH68,,,,,r   c                 .   t          t          j        d                    }|                    |           }|dk    r)d}t          d t	          d          D                       }n>d}t          t          j        gdz  d t	          d	d
          D             z   gdz            }t          j        t          |          5  |	                    d	|           
                                }d d d            n# 1 swxY w Y   t          j        ||           d S )N)
      r   z7The 'axis' keyword in DataFrame.expanding is deprecatedc           	      f    i | ].}|t           j        gd z  d t          dd          D             z   /S )r   c                 ,    g | ]}t          |          S  float).0js     r   
<listcomp>z2test_expanding_axis.<locals>.<dictcomp>.<listcomp>Z   s    ???Qa???r         )r   r   r   rC   is     r   
<dictcomp>z'test_expanding_axis.<locals>.<dictcomp>Z   s>    SSSAQ1??%2,,????SSSr   r=   z7Support for axis=1 in DataFrame.expanding is deprecatedr   c                 ,    g | ]}t          |          S r@   rA   rH   s     r   rE   z'test_expanding_axis.<locals>.<listcomp>_   s    -M-M-M1eAhh-M-M-Mr   rF      r<   r$   axis)r   r   ones_get_axis_numberr   r   r2   assert_produces_warningFutureWarningr   r   r3   )
axis_framer   rN   r)   r4   r5   s         r   test_expanding_axisrT   R   sI   	278$$	%	%Bz**DqyyGSSrSSS
 

 Hrvhl-M-Ma-M-M-MMNQSSTT		#M	=	=	= 8 8aj1155778 8 8 8 8 8 8 8 8 8 8 8 8 8 8&(+++++s   ?*C55C9<C9c                      | t          d                                        d                                          } | t          j        t          j        dddg          }t          j        ||           d S )Nr   rF   r         @      @      @r   r   countr   r   r2   assert_equalr   r5   r4   s      r   %test_expanding_count_with_min_periodsr]   f   sj    _U1XX&&00Q0??EEGGFS#>??HOFH%%%%%r   c                     dddt           j        dddg}g d} | |                                                                          } | |          }t	          j        ||           d S )Nr   r   rF   r   r      )      ?r!   rV   rV   rW   rX   g      @)r   r   r   rZ   r2   r[   )r   valuesexpected_countsr5   r4   s        r   9test_expanding_count_default_min_periods_with_null_valuesrc   m   sq    Arvq!Q'F999O_V$$..006688F//HOFH%%%%%r   c                 0    | t          d                                        d                                          } | t          j        t          j        t          j        t          j        t          j        g          }t          j        ||           d S )Nr   r_   r   rY   r\   s      r   =test_expanding_count_with_min_periods_exceeding_series_lengthre   w   sp    _U1XX&&00Q0??EEGGFGHHHOFH%%%%%r   zdf,expected,min_periodsr   r   rF   )r   r   r_   )Ar   r   r   r   r   r   rF   r_   r`   c                     d |D             }t          ||                     |                    D ]\  }}t          j        ||           d S )Nc                 6    g | ]\  }}t          ||           S r0   )r   rC   ra   r1   s      r   rE   z1test_iter_expanding_dataframe.<locals>.<listcomp>   s)    OOO?FE	&...OOOr   )zipr   r2   r3   )r   r4   r   actuals       r   test_iter_expanding_dataframero   ~   sc    ~ POhOOOH",,{*C*CDD 0 0&
fh////0 0r   zser,expected,min_periodsint64dtypec                     d |D             }t          ||                     |                    D ]\  }}t          j        ||           d S )Nc                 6    g | ]\  }}t          ||           S rk   )r   rl   s      r   rE   z.test_iter_expanding_series.<locals>.<listcomp>   s)    LLLvU+++LLLr   )rm   r   r2   r8   )serr4   r   rn   s       r   test_iter_expanding_seriesrv      sb     ML8LLLH#--*D*DEE 1 1&
vx00001 1r   c                      t                      } t          j        t          d          5  |                     d           d d d            d S # 1 swxY w Y   d S )Nz.* got an unexpected keywordr$   T)center)r   r&   r'   	TypeErrorr   r   s    r   test_center_invalidrz      s    	B	y(F	G	G	G " "
D!!!" " " " " " " " " " " " " " " " " "s   AAAc                 6    | g d          }|                                                                 }t          |t                    rt	          |d         j                  }t	          t          j        gdgdz  z             }t          j	        ||           d S )Nrh   r   g۞?r   )
r   sem
isinstancer   r   ra   r   r   r2   r8   )r   objr5   r4   s       r   test_expanding_semr      s    
/)))
$
$C]]__  ""F&)$$ *q	())rvh(a/00H68,,,,,r   methodskewkurtc                 d   t          t          j                            d                              d                    } t	          |                    d          |                       }|dz   } t	          |                    d          |                       }t          j        ||           d S )Nr   r<   rF   i  )r   r   randomdefault_rnggetattrr   r2   r8   )r   sr4   r5   s       r   ,test_expanding_skew_kurt_numerical_stabilityr      s     	ry$$Q''..r2233A.wq{{1~~v..00H	DA,WQ[[^^V,,..F68,,,,,r   window)r   rF   r<   r=   )minmaxaveragepctTF	ascending	test_data)default
duplicatesnansc           
         d}|dk    rAt          t          j                            d                              |                    }n|dk    rBt          t          j                            d                              d|                    }nl|dk    rft          t          j                            d                              dd	d
t          j        t          j        t          j         g|                    }|                    |                               fd          }|                    |           	                              }t          j        ||           d S )Nr=   r   r   )datar   rF   r   r`   g      ?g      ?c                 J    |                                j        d         S )Nr   r   r   )rankiloc)r9   r   r   r   s    r   <lambda>ztest_rank.<locals>.<lambda>  s!    !&&C9&EEJ2N r   r   )r   r   r   r   choicer   infr   applyr   r2   r8   )	r   r   r   r   r   lengthru   r4   r5   s	    ```     r   	test_rankr      sV    FI")//2299&AABBB	l	"	"")//2299!VDDEEE	f		&&q))00dD"&"&26':F 
 
 
 }}V$$**NNNNNN H ]]6""''v3)'TTF68,,,,,r   c                    |                                  }|t          j                            d                              t          |                    z   d d         }|                                                    |          }|                    t          |          d                              |          }t          j
        ||           d S Nr   r   r   r   )dropnar   r   r   standard_normallenr   corrrollingr2   assert_almost_equalseriesrg   r   r5   rolling_results        r   test_expanding_corrr     s    A	
RY""1%%55c!ff==	=ssCA[[]]""FYYc!ff!Y<<AA!DDN>622222r   c                     |                      d                                          }t          j        ||                     t          |           d                                                     d S )Nr   r   r   )r   rZ   r2   r   r   r   )r   r5   s     r   test_expanding_countr     sj    !,,2244Fc&kkqAAGGII    r   c                     |                                                      d          }|                     t          |           d                              d          }t	          j        ||           d S )N      ?r   r   )r   quantiler   r   r2   r   )r   r5   r   s      r   test_expanding_quantiler     sd    ((--F^^3v;;A^FFOOPSTTN6>22222r   c                    | }|t           j                            d                              t	          |                    z   d d         }|                                                    |          }|                    t	          |          d                              |          }t          j	        ||           d S r   )
r   r   r   r   r   r   covr   r2   r   r   s        r   test_expanding_covr   %  s    A	
RY""1%%55c!ff==	=ssCA[[]]q!!FYYc!ff!Y<<@@CCN>622222r   c                     |                                                                  }|                     t          |           d                                          }t	          j        ||           d S Nr   r   )r   r   r   r   r2   r3   framer5   r   s      r   test_expanding_cov_pairwiser   0  s]    __""$$F]]#e**!]DDHHJJN&.11111r   c                     |                                                                  }|                     t          |           d                                          }t	          j        ||           d S r   )r   r   r   r   r2   r3   r   s      r   test_expanding_corr_pairwiser   8  s]    __##%%F]]#e**!]DDIIKKN&.11111r   zfunc,static_compr   meanc                 .    t          j        | d          S Nr   rM   r   r   r9   s    r   r   r   C  s    2711--- r   r   c                 .    t          j        | d          S r   )r   r   r   s    r   r   r   D      "&+++ r   r   c                 .    t          j        | d          S r   )r   r   r   s    r   r   r   E  r   r   )r   r   r   r   )idsc           	          |t          j        t          t          d                    t           j        gdz  z                       }d}t          j        t          |          5  |                    dd          }d d d            n# 1 swxY w Y    t          ||                       }t          ||          sJ d}d }|t          u r|t           j        u rt          }t          j        ||d	          5   ||d d
                   }d d d            n# 1 swxY w Y   |t          u rt          j        |d         |           d S t          j        |j        d         |d           d S )Nr<   z@The 'axis' keyword in (Series|DataFrame).expanding is deprecatedr$   r   r   r   rN   z:The behavior of DataFrame.sum with axis=None is deprecatedF)r%   check_stacklevelrG   check_names)r   arraylistr   r   r2   rQ   rR   r   r   r}   r   r   r   r   r8   r   )	funcstatic_compr   r   r)   r~   r5   warnr4   s	            r   test_expanding_funcr   ?  s    ?28DrOOrvhm$CDDEED
LC		#M	=	=	= 4 4nnn334 4 4 4 4 4 4 4 4 4 4 4 4 4 4WS$!!Ffo.....
FCD)##rv(=(=		#De	L	L	L * *;tCRCy))* * * * * * * * * * * * * * *&  
vbz844444
v{2eLLLLLLs$   &B

BB8DDDc                    t          t          j                            d                              d                    }d}t          j        t          |          5   t          |	                    dd          |                       }d d d            n# 1 swxY w Y   |d d         
                                                                sJ t          j        |j        d	          ||d d                              t          j        t          |          5   t          |	                    d
d          |                       }d d d            n# 1 swxY w Y   t          |j        d                   sJ t          |j        d                   sJ t          t          j                            d                              d                    }t          j        t          |          5   t          |	                    dd          |                       }d d d            n# 1 swxY w Y   t          |d                   sJ t          |d                   sJ t          j        t          |          5   t          |	                    dd          |                       }d d d            n# 1 swxY w Y   t          j        t          |          5   t          |	                    dd          |                       }d d d            n# 1 swxY w Y   t          j        ||           t          j        t          |          5   t          |	                    dd          |                       }d d d            n# 1 swxY w Y   t          j        |j        d	          ||d d                              d S )Nr   2   z4The 'axis' keyword in Series.expanding is deprecatedr$      r   r      r            r=   r   rF   r   r   )r   r   r   r   r   r2   rQ   rR   r   r   r   allr   r   r	   )r   r   ru   r)   r5   ser2result0result1s           r   test_expanding_min_periodsr   ^  s2    &&q))99"==
>
>C
@C		#M	=	=	= H HE2A>>EEGGH H H H H H H H H H H H H H H#2#;!!#####6;r?KKCRC,A,ABBB 
	#M	=	=	= H HE2A>>EEGGH H H H H H H H H H H H H H HB     R!!!!!")''**::2>>??D		#M	=	=	= H HEAA>>EEGGH H H H H H H H H H H H H H Hq	?? 
	#M	=	=	= H HE'#--AA->>EEGGH H H H H H H H H H H H H H H		#M	=	=	= H HE'#--AA->>EEGGH H H H H H H H H H H H H H H7G,,,		#M	=	=	= G GD11==tDDFFG G G G G G G G G G G G G G G6;r?KKCRC,A,ABBBBBsl   .BBB.EEE5.H//H36H3.J==KK#.LL!$L!.NNNc           	         | \  }} |t          j        t          t          d                    t           j        gdz  z                       }|                    d                              d ||          }t          ||          sJ |t          u r9t          j
        |d         t          j        |d d         d	                     d S t          j        |j        d         t          j        |d d         d	          d
           d S )Nr<   r   r   c                 *    |                                  S Nr   r   s    r   r   z&test_expanding_apply.<locals>.<lambda>      !&&(( r   rawengine	   rG   r   rM   Fr   )r   r   r   r   r   r   r   r}   r   r2   r   r   r8   r   )engine_and_rawr   r   r   r   r5   s         r   test_expanding_applyr     s    KFC?28DrOOrvhm$CDDEED^^^**00F 1  F fo.....&  
vay"'$ss)!*D*D*DEEEEE
KNBGD"IA666E	
 	
 	
 	
 	
 	
r   c                 f   | \  }}t          t          j                            d                              d                    }|                    d                              d ||          }|d d                                                                         sJ t          j
        |j        d         t          j        |d d                              |                    d	                              d
 ||          }t          |j        d                   sJ t          |j        d                   sJ t          t          j                            d                              d                    }|                    d                              d ||          }t          |d                   sJ t          |d                   sJ |                    d                              d ||          }|                    d                              d ||          }t          j
        ||           |                    d                              d ||          }t          j
        |j        d         t          j        |d d                              d S )Nr   r   r   r   c                 *    |                                  S r   r   r   s    r   r   z2test_expanding_min_periods_apply.<locals>.<lambda>  r   r   r   r   r   r   c                 *    |                                  S r   r   r   s    r   r   z2test_expanding_min_periods_apply.<locals>.<lambda>  r   r   r   r   r=   r   c                 *    |                                  S r   r   r   s    r   r   z2test_expanding_min_periods_apply.<locals>.<lambda>  r   r   rF   r   r   c                 *    |                                  S r   r   r   s    r   r   z2test_expanding_min_periods_apply.<locals>.<lambda>  r   r   r   c                 *    |                                  S r   r   r   s    r   r   z2test_expanding_min_periods_apply.<locals>.<lambda>  r   r   c                 *    |                                  S r   r   r   s    r   r   z2test_expanding_min_periods_apply.<locals>.<lambda>  r   r   )r   r   r   r   r   r   r   r   r   r2   r   r   r   r	   )r   r   r   ru   r5   r   r   r   s           r    test_expanding_min_periods_applyr     s    KFC
&&q))99"==
>
>C]]r]**00F 1  F #2#;!!#####6;r?BGCH,=,=>>> ]]r]**00F 1  F B     R!!!!!")''**::2>>??D^^^**00F 1  F q	?? mmm**00F 1  G mmm**00F 1  G 7G,,,]]q]))//F 0  F 6;r?BGCH,=,=>>>>>r   fc                 X    |                      d                              | d          S Nr   r   Tpairwiser   r   r   s    r   r   r     s'    1;;1;--11!d1CC r   c                 X    |                      d                              | d          S r   r   r   r   s    r   r   r     s'    1;;1;--221t2DD r   c                 "   t                      }t          t          dgd          t          g d                    }|d                             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        ||           d S )
Nar"   namebarcolumnsr1   float64r0   )names)r1   r   rr   )	r   r   astyper   from_productr1   r   r2   r3   )r   df1df2df1_expecteddf2_expected
df1_result
df2_results          r   *test_moment_functions_zero_length_pairwiser
    s	    ++C
E3%e444E"5<Q<Q<Q
R
R
RC3xy))CH:#:CIs{;S#T#TUUUL%sy#+&>uenUUUse%(((  L 3J*l3333J*l33333r   c                 N    |                                                                  S r   )r   rZ   r   s    r   r   r     s    !++--%%'' r   c                 X    |                      d                              | d          S Nr   r   Fr   r   r   s    r   r   r     s'    !++!+,,00U0CC r   c                 X    |                      d                              | d          S r  r   r   s    r   r   r     s'    !++!+,,11!e1DD r   c                 R    |                      d                                          S Nr   r   )r   r   r   s    r   r   r          !++!+,,0022 r   c                 R    |                      d                                          S r  )r   r   r   s    r   r   r     r  r   c                 R    |                      d                                          S r  )r   r   r   s    r   r   r     r  r   c                 R    |                      d                                          S r  )r   r   r   s    r   r   r          !++!+,,1133 r   c                 R    |                      d                                          S r  )r   stdr   s    r   r   r     r  r   c                 R    |                      d                                          S r  )r   varr   s    r   r   r     r  r   c                 R    |                      d                                          S r  )r   r   r   s    r   r   r     r  r   c                 R    |                      d                                          S r  )r   r   r   s    r   r   r     r  r   c                 T    |                      d                              d          S )Nr   r   r   )r   r   r   s    r   r   r     s"    !++!+,,55c:: r   c                 R    |                      d                                          S r  )r   medianr   s    r   r   r     s     !++!+,,3355 r   c                 b    |                      d                              t          d          S )Nr   r   Fr   r   r   r   r   s    r   r   r     s'    !++!+,,223E2BB r   c                 b    |                      d                              t          d          S )Nr   r   Tr   r!  r   s    r   r   r     s'    !++!+,,223D2AA r   c                    t          t          j                  }|}t                      }|}t          dg          }|d                             d          |d<   |} | |          }t          j        ||            | |          }t          j        ||            | |          }	t          j        |	|           d S )Nrq   r   r   r   )r   r   r   r   r  r2   r8   r3   )
r   r   
s_expectedr  r  r  r  s_resultr  r	  s
             r   !test_moment_functions_zero_lengthr'    s    , 	RZ   AJ
++CL
SE
"
"
"C3xy))CHLqttH8Z0003J*l3333J*l33333r   c                     | \  }}t          g t          j                  }t          j        ||                                                    d ||                     d S )Nrq   c                 *    |                                  S r   r   r   s    r   r   z3test_expanding_apply_empty_series.<locals>.<lambda>  s    QVVXX r   r   )r   r   r   r2   r8   r   r   )r   r   r   ru   s       r   !test_expanding_apply_empty_seriesr*    sd     KFC
2:
&
&
&CS]]__""#5#53v"NN    r   c                     | \  }}t          g d          }|                    d                              d ||          }t          g d          }t          j        ||           d S )N)NNNr   r   c                      t          |           S r   )r   r   s    r   r   z4test_expanding_apply_min_periods_0.<locals>.<lambda>	  s    A r   r   )r`   r!   rV   )r   r   r   r2   r8   )r   r   r   r   r5   r4   s         r   "test_expanding_apply_min_periods_0r-    sv     KFC!!!""A[[Q[''--.>.>CPV-WWFooo&&H68,,,,,r   c                     t          g dg d          } t          ddgddg          }|                                                     |          }t          g d          }t          j        ||           t          g d	g d          }|                                                     |          }t          j        ||           t          g d
g d          } t          g dg d          }|                                                     |          }t          g d          }t          j        ||           d S )Nrf   rh   r0   r   rF   r   r   )NNr!   r   NrF         r<   r   r   rF   r1  r   r<   r   r   rF   )NNNg      @)r   r   r   r2   r8   s1s2r5   r4   s2as        r   test_expanding_cov_diff_indexr:    s<   					+	+	+B	Aq!f	%	%	%B\\^^##F'''((H68,,,
YYY
/
/
/C\\^^$$F68,,,	


)))	,	,	,B	


)))	,	,	,B\\^^##F---..H68,,,,,r   c                     t          g dg d          } t          ddgddg          }|                                                     |          }t          g d          }t          j        ||           t          g d	g d          }|                                                     |          }t          j        ||           t          g d
g d          } t          g dg d          }|                                                     |          }t          g d          }t          j        ||           d S )Nrf   rh   r0   r   rF   r   r   )NNr`   r/  r0  r3  r4  r5  )NNNr`   )r   r   r   r2   r8   r6  s        r   test_expanding_corr_diff_indexr<  !  s<   					+	+	+B	Aq!f	%	%	%B\\^^  $$F'''((H68,,,
YYY
/
/
/C\\^^  %%F68,,,	


)))	,	,	,B	


)))	,	,	,B\\^^  $$F---..H68,,,,,r   c                     t          ddgddgddggt          ddgd	          
          } t          ddgddggddgt          ddgd	                    }t          ddgd d gddggt          ddgd	          
          }t          ddgddggddgt          ddgd	                    }|                                                     |d          j        d         }|                                                     |d          j        d         }|                                                    |d          j        d         }|                                                    |d          j        d         }t          ddgddggt          ddgd	          t          ddgd	                    }t          j        ||           t          j        ||           t          j        ||           t          j        ||           d S )Nr   r   rF   r   r   rg   r   r"   r   r$  r   r1   r   r_   XYTr   g      g      g      g      $r   )r   r   r   r   locr2   r3   	r  df1ar  df2ar   result2result3result4r4   s	            r   'test_expanding_cov_pairwise_diff_lengthrH  4  s9   
aVaVaV,eS#JU6S6S6S
T
T
TC
Q!QAsCju0M0M0M  D 
Q$1v&sCju0M0M0M  C 
Q!QAsCju0M0M0M  D
 mmoo!!#!559!<Gmmoo!!$!66:1=Gnn""3"66:1=Gnn""4$"77;A>G
e}%sCju---S#JU+++  H
 '8,,,'8,,,'8,,,'8,,,,,r   c            	      0   t          ddgddgddggddgt          t          d          d          	          } t          ddgddggt          d
dgd          ddg          }t          ddgd d gddggddgt          t          d          d          	          }t          ddgddggt          d
dgd          ddg          }|                                                     |d          j        d         }|                                                     |d          j        d         }|                                                    |d          j        d         }|                                                    |d          j        d         }t          ddgddggddgt          ddg          	          }t          j        ||           t          j        ||           t          j        ||           t          j        ||           d S )Nr   r   rF   r   rg   r   r   r   r   r   r>  r   r_   r?  r@  Tr   g      )r   r   r   r   r   rA  r2   r3   rB  s	            r   (test_expanding_corr_pairwise_diff_lengthrJ  Q  sA   

Q!Q!Q 3*E%((QV<W<W<W  C 
Q!Qq!f5 9 9 9C:  D 
Q$1v&c
E!HH5)))  C
 
Q!Qq!f5 9 9 9C:  D mmoo""3"66:1=Gmmoo""4$"77;A>Gnn##C$#77;A>Gnn##D4#88<Q?G
d|$sCjsCj@Q@Q  H '8,,,'8,,,'8,,,'8,,,,,r   c                    d }| \  }}t          t          j                            d                              d                    }|                                                    t          j        ||          dz   }|                                                    |||d          }t          j        ||           |                                                    ||dd	i
          }t          j        ||           d S )Nc                 0    t          j        |           |z   S r   r   )r9   consts     r   
mean_w_argz4test_expanding_apply_args_kwargs.<locals>.mean_w_argo  s    wqzzE!!r   r   )r=   rF   )r   r   g      4@)r=   )r   r   argsrM  r=   )r   kwargs)	r   r   r   r   r   r   r   r2   r3   )r   rN  r   r   r   r4   r5   s          r    test_expanding_apply_args_kwargsrQ  n  s    " " " !KFC	29((++227;;	<	<B||~~##BGF#DDtKH\\^^!!*V5!QQF&(+++\\^^!!*#wm!LLF&(+++++r   c                    | }t          dgddd          }|d                             t                    |d<   |                                }t	          ||d           }| ||          }|rddgng d}||                             |g                              d	
                              t                    }t          |j	                  |k    sJ t          j        ||           d S d S )Nr   r   rF   r   br   r   numeric_onlyr   rT  Tdrop)r   r  objectr   r   aggreset_indexrB   r   r   r2   r3   )	arithmetic_win_operatorsrV  kernelr   r   opr5   r   r4   s	            r   test_numeric_only_framer_    s    %F	!11--	.	.BgnnV$$BsGI	FD	)	)B	~... ,A3**///g;??F8,,88d8CCJJ5QQH$%%0000
fh///// ~r   r]  r   r   use_argc                    t          g dddd          }|d                             t                    |d<   |r|fnd}|                                }t	          ||           } ||d|i}|rdd	gng d}||                             t
                    }	|r|	fnd}
|	                                }t	          ||           } ||
d|i}t          j        ||           d S )
Nrf   r   rF   rS  r   r@   rV  r   rT  )r   r  rY  r   r   rB   r2   r3   )r]  rV  r`  r   argr   r^  r5   r   r  arg2
expanding2op2r4   s                 r    test_numeric_only_corr_cov_framerf    s     
33	4	4BgnnV$$BsG
"2%%CI	F	#	#BR0<00F )=sCjjoooG
W+

U
#
#C$C66"DJ
*f
%
%CsD4|44H&(+++++r   rr   c                    | }t          dg|          }|                                }t          ||          }|rQ|t          u rHd| d}t	          j        t          |          5   ||           d d d            d S # 1 swxY w Y   d S  ||          }|                    |g                              d          	                    t                    }	t          j        ||	           d S )	Nr   rq   
Expanding.  does not implement numeric_onlyr$   rU  TrW  )r   r   r   rY  r&   r'   NotImplementedErrorrZ  r[  r  rB   r2   r8   )
r\  rV  rr   r]  ru   r   r^  r)   r5   r4   s
             r   test_numeric_only_seriesrk    sE    &F
!E
"
"
"CI	F	#	#B 1C6CCC].c::: 	* 	*BL))))	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* ...77F8$$00d0;;BB5II
vx00000s   %A??BBc                    t          g d|          }|r|fnd}|                                }t          ||           }|rM|t          u rDd|  d}t	          j        t          |          5   ||d|i d d d            d S # 1 swxY w Y   d S  ||d|i}	|                    t                    }
|r|
fnd}|
                                }t          ||           } ||d|i}t          j
        |	|           d S )Nrf   rq   r@   rh  ri  r$   rV  )r   r   r   rY  r&   r'   rj  r  rB   r2   r8   )r]  r`  rV  rr   ru   rb  r   r^  r)   r5   r   rc  rd  re  r4   s                  r   !test_numeric_only_corr_cov_seriesrm    sw   
 %
(
(
(C
#3&&CI	F	#	#B 1C6CCC].c::: 	0 	0B/,///	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 S4|44zz%  !)wwr^^%%
j&))38<88
vx00000s   +	BBBc                      t          g d          } t          j        t                    5  |                                                     d           d d d            d S # 1 swxY w Y   d S )N)r   r   rF   r   r   )r   )r   r2   rQ   rR   r   r   )ru   s    r    test_keyword_quantile_deprecatedro    s    


C		#M	2	2 / /  # .../ / / / / / / / / / / / / / / / / /s   )A!!A%(A%)Anumpyr   r&   pandasr   r   r   r   r   r   r	   pandas._testing_testingr2   r   r   markparametrizer   r*   paramxfailr6   r:   rT   r]   rc   re   r   ro   rv   rz   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r
  r'  r*  r-  r:  r<  rH  rJ  rQ  r_  rf  intrY  rk  rm  ro  r@   r   r   <module>ry     s                                   sE828QC==9::  ;: 	+##E $  	
 	
 	
 , , ,- - -, , ,(& & && & && & &  IIIIIII6677s!%%s+1vQF++aV4 yyyyy11999=
 	
 IIIIIII6677s!%%s+1vQF++aV4 yyyyy11999=
 	
 IIIIIII6677s!%%s+1vQF++aV4 yyyyy11999=
 	
 
!A3''	(	("a0	Bxj!$IQN"&!Q@@AAuBF8,,qc226{"&!551v>261~RVQN;;YYYG
 	
 IQN"&!Q@@AAuBF8,,qc226{"&!551v>261~RVQN;;YYYG
 	
 IQN"&!Q@@AAuBF8,,qc226{"&!551v>261~RVQN;;YYYG
 	
a9< <z0 0{< <z0 						qcA3Z1a&1a&)9IIIyyy;QRTUV						qcA3Z1a&1a&)9IIIyyy;QRTUV						qcA3Z1a&1a&)9IIIyyy;QRTUV	AA3*1v1v&67;			1#"&!q!f0EFJ	'	"	"	"B*
 
1 1
 
1" " "- - - FF#344- - 54- >>>22#<#<#<==u..tUm44&G&G&GHH- - IH 54 /. >= 32
-*3 3 3  3 3 33 3 32 2 22 2 2 		--.	++,	++,	 	&%%  	 	M M	 	M* RV_vrw'%5"&/J%%%   
C C 
CB
 
 
 $? $? $?N DDEE 4 4 4& ''CCDD2222223322223333::55BBAA (4 4) (4(  - - -- - -&- - -&- - -:- - -:, , ,"0 0 0" FE?33T5M22, , 32 43,( 3-001 1 101  FE?33T5M223-001 1 10 32 431*/ / / / /r   