+
    yȇic                        ^ RI t^ RIt^ RIHt ^ RIHtHtHtH	t	H
t
HtHtHtHt ^ RIHt ^ RIHt ^ RIHt ]P,                  R 4       t]P,                  R 4       t ! R R	4      t ! R
 R4      t ! R R4      tR tR tR# )    N)Pandas4Warning)		DataFrameDatetimeIndexIndex
MultiIndexNamedAggSeries	Timestamp
date_rangeto_datetime)BaseIndexer)get_groupbyc                 B    \        R. ROR. ROR\        . RO4      /4      # )z0Frame for testing times argument in EWM groupby.ABC)
abcr   r   r   r   r   r   r   )
r   r   r      r   r      r   r      )

2020-01-01r   r   z
2020-01-02
2020-01-10z
2020-01-22z
2020-01-03
2020-01-23r   z
2020-01-04)r   r        p/Users/max/.openclaw/workspace/postharvest/venv/lib/python3.14/site-packages/pandas/tests/window/test_groupby.pytimes_framer      s1     C/	
 r   c                      \        R ^.^,          ^.^,          ,           ^.^,          ,           R\        P                  ! ^(4      /4      # r   r   r   nparanger   r   r   
roll_framer%   /   s:    cA38qcBh.!q8#ryy}MNNr   c                     a  ] tR t^4t o R tR tR t]P                  P                  R. RSO4      R 4       t
]P                  P                  RRR	.4      R
 4       t]P                  P                  R. RTO4      R 4       t]P                  P                  RR^.RR..4      R 4       t]P                  P                  RRR.4      R 4       t]P                  P                  RRR.4      R 4       t]P                  P                  RRRR.RR..3RRR.RR..3.4      R 4       tR tR t]P                  P                  RRR.RR..4      R 4       tR tR  t]P                  P                  R!. RUO4      R" 4       tR# tR$ tR% tR& tR' t]P                  P                  R(RR.4      R) 4       tR* tR+ tR, tR- t ]P                  P                  RVR.R/^R!^/]!PD                  R]!PD                  RR.3R0/ ]!PD                  R]!PD                  RRW.3.4      R1 4       t#]P                  P                  RXR2R3/R33R2R4/R53.4      R6 4       t$]P                  P                  R7RR.4      R8 4       t%R9 t&R: t'R; t(R< t)R= t*]P                  P                  RY^^ R>. RZO3^^R>. R[O3^^R>. R\O3^^R?. R]O3.4      R@ 4       t+]P                  P                  RA],PZ                  ! R^R_.4      RBRC..4      RD 4       t.RE t/]P                  P                  RFRG.RH. R`O/.RGR5.RI]0! RJ4      ]0! RK4      ]0! RJ4      ]0! RK4      .RH. RaO/..4      RL 4       t1RM t2RN t3RO t4RP t5RQ t6RRt7V t8R4# )bTestRollingc                    R p\         P                  ! \        VR7      ;_uu_ 4        VP                  R^R7       RRR4       R#   + '       g   i     R# ; i)z4groupby\(\) got an unexpected keyword argument 'foo'matchr   )fooN)pytestraises	TypeErrorgroupby)selfr%   msgs   && r   !test_groupby_unsupported_argument-TestRolling.test_groupby_unsupported_argument5   s:    E]]9C00s* 1000s   AA	c                   VP                  R 4      p\        VR R7      pVP                  P                  R 4      pVP	                  ^4      P                  4       P                  p\        P                  ! WT4       VP	                  ^4      P                  P                  4       p\        P                  ! WT4       VP                  P	                  ^4      P                  4       p\        P                  ! WT4       VP                  P                  VP                  4      P	                  ^4      P                  4       p\        P                  ! WT4       R# )r   byc                 @    V P                  ^4      P                  4       # )r   rollingmeanxs   &r   <lambda>*TestRolling.test_getitem.<locals>.<lambda>>   s    qyy|/@/@/Br   N)	r/   r   r   applyr9   r:   tmassert_series_equalr   )r0   r%   g	g_mutatedexpectedresults   &&    r   test_getitemTestRolling.test_getitem:   s    s#
s3	;;$$%BC1""$&&
v01$$&
v0Q$$&
v0%%jll3;;A>CCE
v0r   c                `   VP                  R 4      pVP                  ^^ R7      p\        VR R7      pVP                  P	                  R 4      pVP                  P                  4       p\        P                  ! We4       VP                  P                  4       p\        P                  ! We4       R# )r   min_periodsr5   c                 D    V P                  ^^ R7      P                  4       # )r   rI   )r9   countr;   s   &r   r=   3TestRolling.test_getitem_multiple.<locals>.<lambda>Q   s    qyyy/J/P/P/Rr   N)r/   r9   r   r   r?   rL   r@   rA   )r0   r%   rB   rrC   rD   rE   s   &&     r   test_getitem_multiple!TestRolling.test_getitem_multipleL   s    s#IIaQI'
s3	;;$$%RS
v0
v0r   fminmaxc                &  a VP                  R RR7      pVP                  ^R7      p\        VS4      ! 4       pVP                  V3R l4      p\        P
                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   F
group_keyswindowc                 D   < \        V P                  ^4      S4      ! 4       #    getattrr9   r<   rQ   s   &r   r=   *TestRolling.test_rolling.<locals>.<lambda>m   s    WQYYq\1%=%?r   N
r/   r9   r]   r?   r   from_arraysrangeindexr@   assert_frame_equalr0   rQ   r%   rB   rN   rE   rD   expected_indexs   &f&     r   test_rollingTestRolling.test_rollingY   sv      su5IIQIA77?@#//C%)0LM'
f/r   stdvarc                *  a VP                  R RR7      pVP                  ^R7      p\        VS4      ! ^R7      pVP                  V3R l4      p\        P
                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   FrU   rW   ddofc                 H   < \        V P                  ^4      S4      ! ^R7      # )r[   rl   r\   r^   s   &r   r=   /TestRolling.test_rolling_ddof.<locals>.<lambda>y   s    WQYYq\1%=1%Er   Nr`   re   s   &f&     r   test_rolling_ddofTestRolling.test_rolling_ddofs   sw    su5IIQIAA&77EF#//C%)0LM'
f/r   interpolationc                *  a VP                  R RR7      pVP                  ^R7      pVP                  RSR7      pVP                  V3R l4      p\        P
                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   FrU   rW   皙?rr   c                 H   < V P                  ^4      P                  RSR7      # )r[   rt   ru   )r9   quantiler<   rr   s   &r   r=   3TestRolling.test_rolling_quantile.<locals>.<lambda>   s    aiil++C}+Mr   N)
r/   r9   rw   r?   r   ra   rb   rc   r@   rd   )r0   rr   r%   rB   rN   rE   rD   rf   s   &f&     r   test_rolling_quantile!TestRolling.test_rolling_quantile   s     su5IIQIC}=77M
 $//C%)0LM'
f/r   zf, expected_valcorrcov      ?c                   \        R \        ^
4      R^.^,          ^.^,          ,           R. RO^,          /4      P                  RR.4      p\        R \        ^4      R. RO/4      P                  R4      p\        VP	                  ^ R7      P                  ^4      V4      ! V4      p\        P                  .V.^,          ,           ^,          p\        VR .\        P                  ! ^.^,          ^.^,          ,           ^.^,          ^.^,          ,           \        \        ^^4      4      ^,          .. ROR7      R7      p\        P                  ! WW4       R# )	valueidx1idx2levelnamescolumnsrc   N)r   r   r   r[      )r   r   r   )r   rb   	set_indexr]   r/   r9   r#   nanr   ra   listr@   rd   )r0   rQ   expected_valdfotherrE   expected_datarD   s   &&&     r   /test_rolling_corr_cov_other_same_size_as_groups;TestRolling.test_rolling_corr_cov_other_same_size_as_groups   s,    eBi!qA37):FOVWDWX

)VV$
% 	 7E!HfoFGQQ
 !,44Q7;EB&&\NQ$66!;I((C!GqcAg%C!GqcAg%q!%)
 /
 	f/r   c                   aa SP                  R 4      pVP                  ^R7      p\        VS4      ! S4      pVV3R lpVP                  V4      p\        P
                  VR &   \        P                  ! WW4       R# )r   rW   c                 F   < \        V P                  ^4      S4      ! S4      # rZ   r\   )r<   rQ   r%   s   &r   funcITestRolling.test_rolling_corr_cov_other_diff_size_as_groups.<locals>.func   s    199Q<+J77r   N)r/   r9   r]   r?   r#   r   r@   rd   r0   rQ   r%   rB   rN   rE   r   rD   s   &ff     r   /test_rolling_corr_cov_other_diff_size_as_groups;TestRolling.test_rolling_corr_cov_other_diff_size_as_groups   sd    s#IIQIAz*	8 774= 
f/r   c                   a VP                  R 4      pVP                  ^R7      p\        VP                  S4      ! RR7      pV3R lpVP	                  V4      p\
        P                  ! WW4       R# )r   rW   Tpairwisec                 \   < \        V P                  P                  ^4      S4      ! RR7      # )r[   Tr   )r]   r   r9   r^   s   &r   r   8TestRolling.test_rolling_corr_cov_pairwise.<locals>.func   s!    133;;q>1-t<<r   N)r/   r9   r]   r   r?   r@   rA   r   s   &f&     r   test_rolling_corr_cov_pairwise*TestRolling.test_rolling_corr_cov_pairwise   sY    s#IIQIa$/	= 774=
v0r   zfunc, expected_values      ?      @c           	        \        R . ROR. ROR. RO/4      pVP                  R 4      P                  ^4      p\        WA4      ! 4       p\        R^\        P
                  .,          V^ ,          ,           ^\        P
                  .,          ,           R^\        P
                  .,          V^,          ,           ^\        P
                  .,          ,           /\        P                  ! . R	O. R
OR7      R7      p\        P                  ! WV4       R# )r   r   r   Nr   rc   )g1g2r   r   )r   r   r   r   )r   r      r[   ))r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   NN)
r   r/   r9   r]   r#   r   r   from_tuplesr@   rd   )r0   r   expected_valuesr   rolrE   rD   s   &&&    r   test_rolling_corr_cov_unordered+TestRolling.test_rolling_corr_cov_unordered   s     -\\
 jjo%%a(#%Q"&&\OA$66bffXEQ"&&\OA$66bffXE ((	 (
& 	f/r   c                ,  a VP                  R RR7      pVP                  ^R7      pVP                  R SR7      pVP                  V3R l4      p\        P                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   FrU   rW   c                 "    V P                  4       # Nsumr;   s   &r   r=   0TestRolling.test_rolling_apply.<locals>.<lambda>   
    1557r   rawc                 J   < V P                  ^4      P                  R SR7      # )r[   c                 "    V P                  4       # r   r   ys   &r   r=   BTestRolling.test_rolling_apply.<locals>.<lambda>.<locals>.<lambda>   s
    !%%'r   r   )r9   r?   r<   r   s   &r   r=   r      s    QYYq\%7%78Is%7%Sr   N)	r/   r9   r?   r   ra   rb   rc   r@   rd   )r0   r   r%   rB   rN   rE   rD   rf   s   &f&     r   test_rolling_applyTestRolling.test_rolling_apply   s|    su5IIQI *477ST#//C%)0LM'
f/r   c                   \        R R.^,          R.^,          ,           R^.^,          /4      pVP                  R 4      p\        P                  ! . RO4      pR R.Vn        \        \
        P                  RR.^,          R.VR7      pVP                  ^R7      P                  4       p\        P                  ! WT4       VP                  4        VP                  ^R7      P                  4       p\        P                  ! WT4       R# )	r   r+   barr   N       @r   rW   ))r   r   )r   r[   )r   r   )r+   r   )r+   r   )r+   r   )r   r/   r   r   r   r#   r   r9   r   r@   rd   )r0   r   rB   mirD   rE   s   &     r   test_rolling_apply_mutability)TestRolling.test_rolling_apply_mutability   s    eWq[E7Q;6aS1WEFJJsO##T
 ;bffc3/!3cU"M!$((*
f/ 	
 !$((*
f/r   zexpected_value,raw_valueT        Fc                .   R  p\        R. ROR. RO/4      pVP                  R4      P                  P                  ^4      P	                  W2R7      p\        V.^,          \        P                  ! R	RR.R7      RR7      p\        P                  ! WV4       R# )
c                 H    \        \        V \        P                  4      4      # r   )int
isinstancer#   ndarrayr;   s   &r   isnumpyarray6TestRolling.test_groupby_rolling.<locals>.isnumpyarray  s    z!RZZ011r   idr   r   Nr   rc   namer   r   r   r   r   r   )r   r   r   r   r   r   )
r   r/   r   r9   r?   r	   r   r   r@   rA   )r0   expected_value	raw_valuer   r   rE   rD   s   &&&    r   test_groupby_rolling TestRolling.test_groupby_rolling  s    	2 i)<=D!''//2888Uq (()A$PTV

 	v0r   c                   \        \        ^^4      4      pVP                  V4      P                  R^R7      P	                  4       p\        \
        P                  .^,          \        P                  ! R	4      R7      p\        P                  ! W#4       \        \        ^^4      4      pVP                  V4      P                  R^R7      P	                  4       p\        \
        P                  .^,          \        P                  ! R
4      R7      p\        P                  ! W#4       \        RR.^,          R.^,          ,           R\        ^4      /4      pVP                  R4      P                  R^R7      P	                  4       p\        \
        P                  ^^^\
        P                  \
        P                  ^^^^	\
        P                  .\        P                  ! RRR.R7      R.R7      p\        P                  ! W#4       \        RR.^,          R.^,          ,           R\        ^
4      /4      pVP                  R4      P                  R^R7      P	                  4       p\        \
        P                  ^^^\
        P                  \
        P                  ^^^\
        P                  .
\        P                  ! RRR.R7      R.R7      p\        P                  ! W#4       R# )r   T)centerrX   r   r   r   Nr   )rc   r   )r   r   r   r   r   r[   r   )r   r[   )r   r   r   r   )r   r   r   r   r   r   r   r   r   r[   r   r   r   r   r      r      r   	   )r   
   )
r   r   r   r   r   r   r   r   r   r   )r	   rb   r/   r9   r:   r#   r   r   r   r@   rA   r   rd   )r0   seriesrE   rD   r   s   &    r   "test_groupby_rolling_center_center.TestRolling.test_groupby_rolling_center_center  sD   a$'//tA/FKKMVVHqL(()QR
 	v0a$'//tA/FKKMVVHqL(()IJ
 	v0cUQY#2CrCDC((Q(?DDFVVQ1bffbffaAq"&&A(( Dk  E%
( 	f/cUQY#2CrCDC((Q(?DDFVVQ1bffbffaArvv>(( Dk E#
& 	f/r   c           
        \        R \        RR4      RR.^,          R.^,          ,           R\        ^
4      /R7      pVP                  R4      P	                  ^R R^R	7      P
                  P                  4       p\        P                  ! VR,          VR ,          .RR .R
7      p\        . RORVR7      p\        P                  ! W$4       R# )Dater   r   gbgroup_1group_2r   )dataT)onr   rJ   r   r   rc   N)
r         ?r   g      @      @g      @      @      @r   r   )r   r   rb   r/   r9   r   r:   r   ra   r	   r@   rA   )r0   r   rE   r   rD   s   &    r   test_groupby_rolling_center_on*TestRolling.test_groupby_rolling_center_on]  s    
<>ykAoa7r
 JJtWQ6$AW>U446 	
 ##RXr&z$:4.Q>

 	v0r   rJ   c                z   \        R R.^
,          R.^
,          ,           R\        ^4      /4      p^pVP                  R 4      P                  VRVR7      P	                  4       pVP                  4       R R.,          p. ROpV Uu. uF  qfR,           NK  	  pp\        ^ V^,
          4      p\        P                  .V,          p	WV^
V,
           ,           V	,           p
WV^
V,
           ,           V	,           p\        R R.^
,          R.^
,          ,           RW,           /4      p\        P                  ! WL4       R# u upi )	groupr   r   r   T)r   rJ         $@N)
r   r   r   r   r         @      @r   r          @)r   rb   r/   r9   r:   reset_indexrS   r#   r   r@   rd   )r0   rJ   r   window_sizerE   
grp_A_meanr<   
grp_B_meannum_nansnansgrp_A_expectedgrp_B_expectedrD   s   &&           r   'test_groupby_rolling_center_min_periods3TestRolling.test_groupby_rolling_center_min_periodss  s    #sebj!8&%)LMJJwW[;WGTV 	
 ##%w&78G
(23
1$hh

3q+/*x("8b8m DDtK8b8m DDtKsebjC52:-v~7VW
 	f/ 4s   D8c           	        \        R . ROR. R	OR. R
O/4      pVP                  R 4      R.,          P                  ^4      P                  4       p\        \        P
                  \        P
                  R\        P
                  .R.\        P                  ! RR R.R7      R7      p\        P                  ! W#4       VP                  R 4      R,          P                  ^4      P                  4       p\        \        P
                  \        P
                  R\        P
                  .\        P                  ! RR R.R7      RR7      p\        P                  ! W#4       R# )r   r   r   r   Nr   r   r   )r   r   r   r   )r   r   r   r   )r         r  )r   r   r   r   r   )r   r/   r9   rS   r#   r   r   r   r@   rd   r	   rA   r0   r   rE   rD   s   &   r   test_groupby_subselect_rolling*TestRolling.test_groupby_subselect_rolling  s
   ,%93@PQ
 C#'//2668VVRVVS"&&)E((0d
 	f/C%--a0446VVRVVS"&&)((0d 
 	v0r   c                    ! R  R\         4      p\        R. RO^,          /^ .^,          ^.^,          ,           ^.^,          ,           R7      pVP                  VP                  4      P	                  V! ^R7      ^R7      P                  4       pVP                  VP                  4      P	                  ^^R7      P                  4       p\        P                  ! W44       R# )	c                   *   a  ] tR tRt o RR ltRtV tR# )FTestRolling.test_groupby_rolling_custom_indexer.<locals>.SimpleIndexeri  Nc                    Vf   V P                   M^ p\        P                  ! V\        P                  R7      ^,           pW`P                   ,
          pW'V^ 8  &   Wv3# )Ndtype)r  r#   r$   int64)r0   
num_valuesrJ   r   closedstependstarts   &&&&&&  r   get_window_boundsXTestRolling.test_groupby_rolling_custom_indexer.<locals>.SimpleIndexer.get_window_bounds  sQ     3>2Ed..1ii
"((;a?...#.eai z!r   r   )r   NNNN)__name__
__module____qualname____firstlineno__r  __static_attributes____classdictcell____classdict__s   @r   SimpleIndexerr    s     " "r   r)  r   r   )r  rI   rX   rJ   N)r   r   r   r   r   )r   r   r/   rc   r9   r   r@   rd   )r0   r)  r   rE   rD   s   &    r   #test_groupby_rolling_custom_indexer/TestRolling.test_groupby_rolling_custom_indexer  s    	"K 	" +a/0a1#'8IQCRSG8S
 JJrxx W]q1qWASU 	
 ::bhh'//qa/HLLN
f/r   c                B   \        R \        ^4      R\        ^4      RR.^,          R.^,          ,           RR Uu. uF  p\        V4      NK  	  up^,          /4      pVP                  R4      P	                  RRRR7      R ,          P                  4       p\        \        P                  \        P                  R	R	\        P                  \        P                  R
R
.\        P                  ! VRR.,          RR.R7      R R7      p\        P                  ! W44       R# u upi column1column2r   r   r   date1Dleftr   r  r   g      "@r   r   N)
2019-01-01r5  
2019-01-02r6  r   rb   r
   r/   r9   r   r	   r#   r   r   
from_framer@   rA   r0   r1  r   rE   rD   s   &    r   'test_groupby_rolling_subset_with_closed3TestRolling.test_groupby_rolling_subset_with_closed  s   5858#cUQY. X X dO X 			
 JJw'''G	RVVX 	 VVRVVS#rvvrvvsC@''GV$%' 
 	v0%s   Dc                   \        R . ROR. ROR. RO/4      pVP                  R 4      P                  ^4      P                  \	        R\
        R	7      \	        R\        R	7      R
7      p\        R\        P                  RR\        P                  RR.R\        P                  RR\        P                  RR./\        RR.. RO.. RO. RO.R R.R7      R7      p\        P                  ! W#4       R# )kindcatdogheight333333"@r         #@r  weightcolumnaggfunctotal_weight
min_heightrH  1@gfffff3@     i@g      n@rI  Nr   r   r>  r?  r>  r?  r>  r?  rA  r   rB  g      A@      (@r  皙@r   g#@g     h@r   g      E@r   r   r   r   r[   r   r   r   r   r   r   r   r   r   r[   r   r   r   )r   r/   r9   aggr   r   rR   r#   r   r   r@   rd   r  s   &   r   !test_groupby_rolling_agg_namedagg-TestRolling.test_groupby_rolling_agg_namedagg  s    B:<
 JJvWQZS%XsC#8SA   	 tRVVUE JrvvsCcB !34#%78tn

 	f/r   c                T   \        R \        ^4      R\        ^4      RR.^,          R.^,          ,           RR Uu. uF  p\        V4      NK  	  up^,          /4      pVP                  R4      R R.,          P	                  RRRR7      R ,          P                  4       p\        \        P                  \        P                  R	R	\        P                  \        P                  R
R
.\        P                  ! VRR.,          RR.R7      R R7      p\        P                  ! W44       R# u upi r.  r7  r9  s   &    r   .test_groupby_subset_rolling_subset_with_closed:TestRolling.test_groupby_subset_rolling_subset_with_closed  s   5858#cUQY. X X dO X 			
 JJwF 34WTfVW4Y@SU 	
 VVRVVS#rvvrvvsC@''GV$%' 
 	v0+s   D%r   c                r   \        . RO\        P                  ! . R	ORR.R7      RR7      p\        VP	                  V4      P                  ^4      V4      ! 4       p\        \        P                  \        P                  R.\        P                  ! . R
O. ROR7      RR7      p\        P                  ! W44       R# )r   r   12r   r   r   N)r   r   r   ))r   r<   )r   r   )r   z))r   r   r<   )r   r   r   )r   r   r]  )r   r[  r\  )
r	   r   r   r]   r/   r9   r#   r   r@   rA   )r0   r   dsrE   rD   s   &&   r   "test_groupby_rolling_index_changed.TestRolling.test_groupby_rolling_index_changed  s     ((4S#J 
 B//2D9;VVRVVS!((=_ 
 	v0r   c           	        \        R . /4      pVP                  R 4      P                  ^R7      P                  4       pVP	                  R R7      p\
        P                  ! \        . RR7      \        . RR7      .R R.R7      Vn        \        P                  ! W!4       \        R . R. /4      pVP                  R R.4      P                  ^R7      P                  4       pVP	                  R R.R7      p\
        P                  ! \        . RR7      \        . RR7      \        . RR7      .. R	OR7      Vn        \        P                  ! W!4       R# )
s1rW   r   float64r  r  Nr   s2)rb  re  N)r   r/   r9   r   dropr   from_productr   rc   r@   rd   )r0   rD   rE   s   &  r    test_groupby_rolling_empty_frame,TestRolling.test_groupby_rolling_empty_frame/  s$   dBZ(!!$'//q/9==?===. $002Y'r)AB4QU,
 	f/dBb12!!4,/77q7AEEG==$=6#00b	*b	*b(
 %
 	f/r   c                   \        R R\        R^^^	4      .RR\        R^^^	4      .RR\        R^^^	4      .RR\        R^^^	4      .RR\        R^^^	4      ... ROR7      P                  R4      pVP                  R	4      pVP	                  4       VR&   VP                  RR
R7      pVP                  R 4      p\        R R\        R^^^	4      R.RR\        R^^^	4      R.RR\        R^^^	4      R.RR\        R^^^	4      R.RR\        R^^^	4      R... ROR7      P                  R	R.4      p\        P                  ! WE4       R# )r   r     r   Zr   HErc   r   	eventTimerc  count_to_date10Dr   c                 (    V P                   ^ ,          # )r   )shape)r   s   &r   r=   ?TestRolling.test_groupby_rolling_string_index.<locals>.<lambda>Z  s    !r   r   r   r   N)rc   r   ro  )rc   r   ro  rp  )	r   r
   r   r/   cumcountr9   r?   r@   rd   )r0   r   groupsrolling_groupsrE   rD   s   &     r   !test_groupby_rolling_string_index-TestRolling.test_groupby_rolling_string_indexJ  s~   i4Aq!9:i4Aq!9:i4Aq!9:i4Aq!9:i4B!:; 4	
 )G
 	 G$$oo/?+>%%&<=i4Aq!93?i4Aq!93?i4Aq!93?i4Aq!93?i4B!:C@ E	
 )Wg&
' 	 	f/r   c           	     `   \        R ^^.R^^./4      P                  R RR7      P                  ^4      P                  4       p\        \        P
                  ! RR.RR..4      R R.\        P                  ! R
R.R R.R7      R7      pVP                  R R	7      p\        P                  ! W4       R# )r+   r   F)sortr   r   Nr   r   rc  )r   r   r   )r   r/   r9   rR   r#   arrayr   r   rf  r@   rd   )r0   rE   rD   s   &  r   test_groupby_rolling_no_sort(TestRolling.test_groupby_rolling_no_sortg  s     uq!feaV45WUW'WQZSU	 	 HHsCj3*-.EN((&&)9%O
 ===/
f/r   c                ^   \        R \        ^4      R\        ^4      R^RR.,          R\        R^VR7      /4      pRp\        P                  ! \
        VR	7      ;_uu_ 4        VP                  R4      P                  R
RRR7      R ,          P                  4       pRRR4       \        . RORV R2R7      p\        P                  ! . ROV.RR.R7      p\        \        P                  RR\        P                  RR.R VR7      p\        P                  ! XV4       R#   + '       g   i     L; i)r/  r0  r   r   r   r1  20190101)r  periodsunitz:'d' is deprecated and will be removed in a future version.r)   3dr3  r4  NM8[]r  r   r   r   )z
2018-12-27z
2018-12-29z
2018-12-31z
2018-12-28z
2018-12-30r5  )r   r   r   r   r   r   )r   rb   r   r@   assert_produces_warningr   r/   r9   rL   r   r   ra   r	   r#   r   rA   )r0   r  r   r1   rE   dtir   rD   s   &&      r   $test_groupby_rolling_count_closed_on0TestRolling.test_groupby_rolling_count_closed_onx  s   5858c3Z
z14H	
 K''cBB

7#&8D  C  vQ-

 ##. F#
 VVS#rvvsC0

 	vx0; CBs   !9DD,	r9   rX   	expandingc                r   \        R ^.R ^.R^.R^.R^..R R.R7      p\        VP                  R 4      V4      ! R/ VB P                  4       p\        R \        P
                  .^,          RV/\        P                  ! . ROR R.R7      R7      pVP                  R R7      p\        P                  ! WV4       R# )r   r   rc  Nr   r   r   )r   r   )r   r   )r   r   r   r[   )r   r]   r/   semr#   r   r   r   rf  r@   rd   )r0   r   kwargsr   r   rE   rD   s   &&&&   r   test_groupby_rolling_sem$TestRolling.test_groupby_rolling_sem  s     1XQx#qC8c1X>c

 C$/9&9==?266(Q,_5((B3PT+
 ===-
f/r   r   r   Nrc   c           	     ^   \        R \        RR\        P                  .4      R. R	OR. R
O/4      pVR8X  d   VP	                  R 4      p\
        P                  ! \        V R2R7      ;_uu_ 4        VP                  R4      P                  ! R/ VB  RRR4       R#   + '       g   i     R# ; i)r   z2020-06-01 12:00z2020-06-01 14:00r   r   rc   z values must not have NaTr)   Nr   r   )60min)
r   r   r#   r   r   r,   r-   
ValueErrorr/   r9   )r0   rollingskeyr   s   &&& r   "test_groupby_rolling_nans_in_index.TestRolling.test_groupby_rolling_nans_in_index  s    
 ["46H"&&!QRYY
 '>c"B]]:u4M-NOOJJsO##8x8 POOOs   /"BB,	rV   c                F   . RO. RO.p\         P                  ! VRR7      p\        . R	OVR7      pVP                  RR.VR7      P	                  ^4      P                  4       p\        . R
O\         P                  ! . RO. ROR7      R7      p\        P                  ! WV4       R# )val1r   r   r   r   rU   Nr  r  val2r   r   r   )r   r   r   )r  r  r  r  r  )r  r  r  r  )r   r   r   r   )	r   ra   r	   r/   r9   r:   r   r@   rA   )r0   rV   arraysrc   srE   rD   s   &&     r   test_groupby_rolling_group_keys+TestRolling.test_groupby_rolling_group_keys  s    
 +,DE&&v5EF9E*FF+
CKKANSSU((
 7

 	v0r   c                `   . RO. RO.p\         P                  ! VRR7      p\        R. R	OR\        ^4      /VR7      pVP	                  RR.4      P                  ^4      P                  4       p\        R. R
O/\         P                  ! . RO. ROR7      R7      p\        P                  ! WE4       R# )r  r   r   r   r   r   Nr  r  )r   r   r   )r   r   r   )r  r   r  r  r  )r  r   r  r  )r   r   r   r   )
r   ra   r   rb   r/   r9   r:   r   r@   rd   r0   r  rc   r   rE   rD   s   &     r   1test_groupby_rolling_index_level_and_column_label=TestRolling.test_groupby_rolling_index_level_and_column_label  s    *,DE&&v5EFYU1X6eDVSM*2215::</"((
 4

 	f/r   c                4   \        R \        P                  ! R4      R^^.^,          /4      pVP                  R4      P	                  ^4      P                  4       p\        P                  ! . RORR.R7      p\        P                  ! VP                  V4       R# )r   r  r   Nr   )r   r   )r   r[   )r   r   r   r  )r   r   )r   r   )r   r#   r$   r/   r9   r:   r   r   r@   assert_index_equalrc   r0   r   rE   rf   s   &   r   )test_groupby_rolling_resulting_multiindex5TestRolling.test_groupby_rolling_resulting_multiindex  sx    
 RYYs^S1a&1*=>C((+002#//L+
 	fllN;r   c           	     N   \        R \        P                  ! R4      R^^.^,          R. RO^,          /4      pVP                  RR.4      P	                  ^4      P                  4       p\        P                  ! . RO. ROR7      p\        P                  ! VP                  V4       R# )	r   rN  r   r   Nr   r   r   r   r[   ))r   r   r   )r   r   r[   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r[   r   )r   r[   r   )r   r[      )r   r   N)r   r#   r$   r/   r9   r   r   r   r@   r  rc   r  s   &   r   *test_groupby_rolling_resulting_multiindex26TestRolling.test_groupby_rolling_resulting_multiindex2  s    RYYt_cAq6A:sLSTDTUVS#J'//2668#// #
" 	fllN;r   c           	     t   \        R \        P                  ! R4      R^^.^,          R. R
O^,          /4      pVP                  RRR7      pVP	                  R4      P                  ^4      P                  4       p\        P                  ! . RO. ROR7      p\        P                  ! VP                  VRR	7       R# )r   r  r   r   T)appendNr   equiv)exactr  ))r   r   r   r   )r   r[   r   )r   r   r   )r   r   r   )r   r   r[   )r   r   r   )r   r   r[   )r   Nr   )r   r#   r$   r   r/   r9   r:   r   r   r@   r  rc   r  s   &   r   *test_groupby_rolling_resulting_multiindex36TestRolling.test_groupby_rolling_resulting_multiindex3  s    RYYs^S1a&1*c<RSCSTU\\#d\+C((+002#//	 #
 	fllN'Jr   c                    VP                  R RR7      pVP                  R 4      P                  pVP                  ^R7      pVP                  R 4      P                  p\        P
                  ! WS4       R# )r   FrU   c                 @    V P                  ^4      P                  4       # rZ   r9   r   r;   s   &r   r=   UTestRolling.test_groupby_rolling_object_doesnt_affect_groupby_apply.<locals>.<lambda>5  s    QYYq\%5%5%7r   rW   c                 @    V P                  ^4      P                  4       # rZ   r  r;   s   &r   r=   r  7  s    199Q<#3#3#5r   N)r/   r?   rc   r9   r@   r  )r0   r%   rB   rD   _rE   s   &&    r   7test_groupby_rolling_object_doesnt_affect_groupby_applyCTestRolling.test_groupby_rolling_object_doesnt_affect_groupby_apply2  s_    su57778>>IIQI56<<
f/r   r3  rightc           	     N   \        . RO4      pVP                  . RO4      P                  WVR7      P                  ^ 4      p\        \        P
                  ! VRR7      \        \        P
                  ! ^^.4      . R	O.. R
O. RO.R7      R7      p\        P                  ! Wg4       R# )r   )rX   rJ   r  rd  r  )levelscodesr   N)r   r   r   r[   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r[   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r[   r   r   r   r   r   )	r   r/   r9   rj   r#   r}  r   r@   rd   )r0   rX   rJ   r  rD   r   rE   expected_results   &&&&&   r   test_groupby_rolling_var$TestRolling.test_groupby_rolling_var:  s     /0JJ/0WFFWKSV 	
 $HHXY/!Q(*BC/1IJ
 	f6r   r   r   r   c                |   \        ^^ ..^,          ^^ ..^,          ,           ^^ ..^,          ,           VR7      pVP                  R4      pVP                  P                  RR7      pVP	                  ^4      pVP                  4       pRVP                  9  g   Q h\        P                  ! VP                  V4       R# )r   rc  r   T)deepN)	r   r/   objcopyr9   r   r   r@   rd   )r0   r   r   rB   original_objrN   rE   s   &&     r   test_by_column_not_in_values(TestRolling.test_by_column_not_in_valuesS  s    
 Ax"}Ax"}4Ax!|CWUJJsOuuzztz,IIaL&..(((
aee\2r   c                   . R	O. R
O.p\         P                  ! VRR7      p\        R. RO/VR7      pVP                  ^ R7      R,          P	                  ^4      P                  4       p\        \        P                  R\        P                  R.\         P                  ! . RO. ROR7      RR7      p\        P                  ! WE4       R# )Falconr   z	Max Speedr   r   g      @g      I@r   N)r  r  Parrotr  )CaptiveWildr  r  )AnimalType)g     `x@g     u@g      >@g      4@))r  r  r  )r  r  r  )r  r  r  )r  r  r  )r  r  r  )r   ra   r   r/   r9   r   r	   r#   r   r   r@   rA   r  s   &     r   test_groupby_levelTestRolling.test_groupby_level`  s     52
 &&v5GH%?@N!$[199!<@@BVVUBFFD)(( 3 
 	v0r   zby, expected_datar   numr1  
2018-01-01
2018-01-02c                   . RO. RO. RO. RO.p\        V. ROR7      pVR,          P                  RV R24      VR&   VP                  R.4      pV Uu. uF  p\        WV4      NK  	  ppVP	                  VRR7      P                  ^^R7      P                  4       pR. RO/p	V	P                  V4       \        V	VP                  R	7      p	RV9   d    V	R,          P                  RV R24      V	R&   \        P                  ! W4       R
# u upi )r   r   r1  rc  r  r  F)as_indexr*  r   N)r   r        Y@)r   r        i@)r   r       b@)r   r       @o@)r   r1  r  )r   r   r   r   )r   astyper   r]   r/   r9   r:   updaterc   r@   rd   )
r0   r6   r   r  r   r   attrgp_byrE   rD   s
   &&&&      r   test_as_index_falseTestRolling.test_as_index_falsex  s	   * '&&&	
 t%:;Z&&TF!}56
\\6(#/12rt"r2JJuuJ-55QA5NSSU 	 ./&((
 ]"'/66TF!}EHV
f/ 3s   Dc                |   \         P                  ! V4      P                  pR p\         P                  ! V! ^ 4      V4      p^VR	&   Rp\         P                  ! WS4      p\         P                  V^ &   ^ VR	&   \        RVRV/4      P                  R4      pVP                  R4      R,          P                  ^
^R7      P                  4       p\        VR\        P                  ! \        ^ .R,          ^.,           VRR7      \        ^ .R,          ^.,           VRR7      .4      R7      p	\        P                  ! W4       R# )
  g52DErc   adl2r*  i  )r  r   r   N)r#   r  typerepeatr   r   r   r/   r9   r:   r	   r   ra   r   r@   rA   )
r0   any_int_numpy_dtypetypsizeidxvalarrr   rE   rD   s
   &&        r   test_nan_and_zero_endpoints'TestRolling.test_nan_and_zero_endpoints  s   hh*+00iiA%Bii"AB

 )G
 	 G$V,44BA4NSSU((1#)qc/7C1#)qc/7C	
 	v0r   c                `   . R
OpRp\        V Uu. uF)  pR\        ^V,          V,          4      RV^,           R^*/NK+  	  up4      p\        P                  ! \        RR7      ;_uu_ 4        VP                  R4      P                  RRR7       R	R	R	4       R	# u upi   + '       g   i     R	# ; i)r   r  tr<   r   z.* must be monotonicr)   3s)r   rX   N)r   r   r   r   )r   r
   r,   r-   r  r/   r9   )r0   shuffledsecr<   r   s   &    r   "test_groupby_rolling_non_monotonic.TestRolling.test_groupby_rolling_non_monotonic  s      IQRAc9QUS[)3AsB?R
 ]]:-DEEJJsO##s4#8 FE SEEEs   /B*#BB-	c                   . R	O. R
O. RO. RO. RO. RO. RO. RO. RO. RO.
p\        V. ROR7      p\        VR,          4      VR&   VP                  R4      pVP                  R4      P	                  R4      P                  R 4      pVP	                  R4      P                  RRR7      R,          P                  4       p\        P                  ! WC4       R# )Davidr   r1  amount)r   r   c                 N    V P                  R 4      R,          P                  4       # )180Dr  r  r;   s   &r   r=   4TestRolling.test_groupby_monotonic.<locals>.<lambda>  s    QYYv.x8<<>r   r  rr  N)r  z1/1/2015d   )r  z1/5/2015  )r  z	5/30/20152   )r  z	7/25/2015r  )Ryanz1/4/2014r  )r  z	1/19/2015r  )r  z	3/31/2016r  )Joez7/1/2015r  )r  z9/9/2015r  )r  z
10/15/2015r  )r   r1  r  )
r   r   sort_valuesr   r/   r?   r9   r   r@   rA   )r0   r   r   rD   rE   s   &    r   test_groupby_monotonic"TestRolling.test_groupby_monotonic  s     '&&&%&%$$%
 D*DE F,6
^^F# LL WV_U>? 	
 F#++Fv+>xHLLN
v0r   c           
        \        R ^RR7      p\        R^.^,          ^.^,          ,           ^.^,          ,           R\        P                  ! W34      R\        P                  ! ^(4      /4      pVP                  R4      P                  R4      P                  R 4      pVP                  R4      P                  RRR7      P                  P                  4       p\        P                  ! WC4       R	# )
z2016-01-01 09:30:00r  )r  r  freqr   r   r   c                 N    V P                  R 4      R,          P                  4       # )4sr   r8   r;   s   &r   r=   JTestRolling.test_datelike_on_monotonic_within_each_group.<locals>.<lambda>  s    199T?3;O;T;T;Vr   r  rr  N)r   r   r#   concatenater$   r   r/   r?   r9   r   r:   r@   rA   )r0   datesr   rD   rE   s   &    r   ,test_datelike_on_monotonic_within_each_group8TestRolling.test_datelike_on_monotonic_within_each_group  s     !6MaS2Xb(A372R^^UN3RYYr]
 LL%%c*001VW 	 C((#(688==?
v0r   c                |   \        R ^.^,          ^.^,          ,           RR Uu. uF  p\        V^^4      NK  	  up^,          R\        ^4      /4      p\        P                  ! \
        RR7      ;_uu_ 4        VP                  R 4      P                  RRR7       RRR4       R# u upi   + '       g   i     R# ; i)	r   r   r   z&Each group within B must be monotonic.r)   365Drr  N)i  i  rk  )r   r
   rb   r,   r-   r  r/   r9   )r0   yearr   s   &  r   0test_datelike_on_not_monotonic_within_each_group<TestRolling.test_datelike_on_not_monotonic_within_each_group  s    aS1WsQw&8JK8Jia+8JKaOU1X
 ]]:-UVVJJsO##Fs#3 WV	 L WVVs   B%8#B**B;	r   
r   r:   rR   rS   firstlastrL   kurtskewnuniquelinearlowerhighermidpointnearest)r   r[   r   )r   r  r   g3Ey?)r  r  )rX   rJ   r  rD   )Nr   r   r   Nr   r   r   )NNr   r   NNr   r   )NNNNNNNN)NNNr   NNNr   )r    )r   r   )r  r  r  r  )r  r  r  r  )9r!  r"  r#  r$  r2   rF   rO   r,   markparametrizerg   rp   rz   r   r   r   r   r   r   r   r   r   r
  r  r+  r:  rU  rX  r_  rh  ry  r~  r  r#   r   r  r  r  r  r  r  r  r  r  r   r   r  r  r
   r  r  r  r  r  r  r%  r&  r'  s   @r   r'   r'   4   sg    +
1$1 [[	
	0	0 [[S5%.1	0 2	0 [[M00 [[.&!ucl0KL0 M00 [[S65/20 30 [[S65/2
1 3
1 [[
3*sCj)	*VsCj3*5M,NO0	0@
00. [[73+U|9TU1 V1?0B1, [[]I60 7041208160<1< [[VeU^41 51(060:0")1V [[- 1mQ/bffc3/
 "rvvsBFFCHI	

0
0  [[c{C0D$<2IJ99 [[\D%=91 :1,0(<<.K(0 [[76EF6GH6KL7JK		
77  [[J**Iz+BCc3ZP3310 [[Ve9:;w!,/!,/!,/!,/	 7	
$0%$061>	91@1*
4 
4r   r'   c                   p  a  ] tR tRt o ]P
                  R 4       t]P                  P                  R. RO4      R 4       t	]P                  P                  RRR.4      R 4       t
]P                  P                  R. RO4      R	 4       t]P                  P                  RR
R.4      R 4       tR tR tRtV tR# )TestExpandingi  c                    \        R ^.^,          ^.^,          ,           ^.^,          ,           R\        P                  ! ^(4      /4      # r!   r"   )r0   s   &r   frameTestExpanding.frame  s:    #sRx1#(2aS1W<c299R=QRRr   rQ   c                "  a VP                  R RR7      pVP                  4       p\        VS4      ! 4       pVP                  V3R l4      p\        P
                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   FrU   c                 B   < \        V P                  4       S4      ! 4       # r   r]   r  r^   s   &r   r=   .TestExpanding.test_expanding.<locals>.<lambda>*  s    WQ[[]A%>%@r   N
r/   r  r]   r?   r   ra   rb   rc   r@   rd   r0   rQ   r)  rB   rN   rE   rD   rf   s   &f&     r   test_expandingTestExpanding.test_expanding  sn      MM#%M0KKMA77@A#//sU2Y0GH'
f/r   ri   rj   c                &  a VP                  R RR7      pVP                  4       p\        VS4      ! ^ R7      pVP                  V3R l4      p\        P
                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   FrU   rl   c                 F   < \        V P                  4       S4      ! ^ R7      # )r   rl   r-  r^   s   &r   r=   3TestExpanding.test_expanding_ddof.<locals>.<lambda>6  s    WQ[[]A%>A%Fr   Nr/  r0  s   &f&     r   test_expanding_ddof!TestExpanding.test_expanding_ddof0  so    MM#%M0KKMAA&77FG#//sU2Y0GH'
f/r   rr   c                &  a VP                  R RR7      pVP                  4       pVP                  RSR7      pVP                  V3R l4      p\        P
                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   FrU   rt   ru   c                 F   < V P                  4       P                  R SR7      # )rt   ru   )r  rw   rx   s   &r   r=   7TestExpanding.test_expanding_quantile.<locals>.<lambda>E  s    akkm,,S,Nr   N)
r/   r  rw   r?   r   ra   rb   rc   r@   rd   )r0   rr   r)  rB   rN   rE   rD   rf   s   &f&     r   test_expanding_quantile%TestExpanding.test_expanding_quantile<  sw     MM#%M0KKMC}=77N
 $//sU2Y0GH'
f/r   r|   r}   c                "  aa SP                  R 4      pVP                  4       p\        VS4      ! S4      pVV3R lpVP                  V4      p\	        \        ^^=4      4      \	        \        ^H^q4      4      ,           p\        P                  VP                  V^3&   \        P                  VR &   \        P                  ! WW4       \        VP                  S4      ! RR7      pV3R lp	VP                  V	4      p\        P                  ! WW4       R# )r   c                 D   < \        V P                  4       S4      ! S4      # r   r-  )r<   rQ   r)  s   &r   func_05TestExpanding.test_expanding_corr_cov.<locals>.func_0S  s    1;;=!,U33r   Tr   c                 Z   < \        V P                  P                  4       S4      ! R R7      # )Tr   )r]   r   r  r^   s   &r   func_15TestExpanding.test_expanding_corr_cov.<locals>.func_1b  s    133==?A.==r   N)r/   r  r]   r?   r   rb   r#   r   ilocr@   rd   r   rA   )
r0   rQ   r)  rB   rN   rE   r?  rD   null_idxrB  s
   &ff       r   test_expanding_corr_cov%TestExpanding.test_expanding_corr_covL  s    MM#KKMAu%	4 776? b"&eBn)==%'VVhk" 
f/a$/	> 776?
v0r   c                (  a VP                  R RR7      pVP                  4       pVP                  R SR7      pVP                  V3R l4      p\        P                  ! VR ,          \        ^(4      .4      pWvn        \        P                  ! WV4       R# )r   FrU   c                 "    V P                  4       # r   r   r;   s   &r   r=   4TestExpanding.test_expanding_apply.<locals>.<lambda>m  r   r   r   c                 H   < V P                  4       P                  R  SR7      # )c                 "    V P                  4       # r   r   r   s   &r   r=   FTestExpanding.test_expanding_apply.<locals>.<lambda>.<locals>.<lambda>n  s
    1557r   r   )r  r?   r   s   &r   r=   rJ  n  s    Q[[]%8%89JPS%8%Tr   N)	r/   r  r?   r   ra   rb   rc   r@   rd   )r0   r   r)  rB   rN   rE   rD   rf   s   &f&     r   test_expanding_apply"TestExpanding.test_expanding_applyh  st    MM#%M0KKM *477TU#//sU2Y0GH'
f/r   c           	     \   \        R . ROR. ROR. RO/4      pVP                  R 4      P                  ^4      P                  \	        R\
        R7      \	        R\        R7      R7      p\        R. ROR. RO/\        RR.. RO.. RO. RO.R R	.R
7      R7      p\        P                  ! W#4       R	# )r=  r>  r?  r@  rC  rD  rG  rH  rI  Nr   r   rL  rM  rO  )rP  rJ  g;@r   rK  g     n@)rA  rA  rA  r   r   r   rQ  rR  rS  )
r   r/   r  rT  r   r   rR   r   r@   rd   r  s   &   r   #test_groupby_expanding_agg_namedagg1TestExpanding.test_groupby_expanding_agg_namedaggt  s    B:<
 JJvYq\S%XsC#8SA   	  D< !34#%78tn

 	f/r   r   Nr  r  )r!  r"  r#  r$  r,   fixturer)  r$  r%  r1  r6  r;  rF  rN  rQ  r%  r&  r'  s   @r   r'  r'    s     ^^S S [[	
	0	0 [[S5%.1	0 2	0 [[M00 [[S65/21 316
00 0r   r'  c            
       R  a  ] tR tRt o ]P
                  P                  RR. RO.R]P                  RRR..R]P                  R	R
R...4      R 4       t	R t
]P
                  P                  RR]P                  RR^..R]P                  R	RR...4      R 4       tR tR tR tRtV tR# )TestEWMi  zmethod, expected_datar:   ri   绹۞?禀?t ?rj   r~   n۶m۶?++?c                $   \        R R.^,          R\        ^4      /4      p\        VP                  R 4      P	                  RR7      V4      ! 4       p\        RV/\
        P                  ! . ROR R.R7      R7      p\        P                  ! WE4       R# )	r   r   r   r   comNr   r   r   r   r   r   )	r   rb   r]   r/   ewmr   r   r@   rd   r0   methodr   r   rE   rD   s   &&&   r   test_methodsTestEWM.test_methods  s     cUQYU1X67C,,,5v>@- (( Dk
 	f/r   c                   \        R R.^,          R\        ^4      /4      pVP                  R 4      P                  RR7      P	                  \        RRR7      \        RRR7      \        RRR7      R	7      p\        R
. ROR\        P                  RRR.R\        P                  RRR./\        P                  ! . ROR R.R7      R7      p\        P                  ! W#4       R# )r   r   r   r   r\  r:   rD  ri   rj   )B_meanB_stdB_varre  rf  rV  rW  rX  rg  r~   rY  rZ  Nr   r   r   gUUUUUU?gm۶m?g""""""@r^  )r   rb   r/   r_  rT  r   r#   r   r   r   r@   rd   r  s   &   r   test_groupby_ewm_agg_namedagg%TestEWM.test_groupby_ewm_agg_namedagg  s    cUQYU1X67JJsOSSS\SsF;c59c59   	   "&&(Hh?"&&#'9;MN	 (( Dk
* 	f/r   r|   r   r}   grڶ?g߿yq+?c                  a \        R R.^,          R\        ^4      /4      p\        VP                  R 4      P	                  RR7      S4      ! 4       p\        RV/\
        P                  ! . R	O. R
OR7      R7      p\        P                  ! WE4       VP                  R 4      R.,          P                  V3R l4      p\        P                  ! WE4       R# )r   r   r   r   r\  Nr   r   c                 F   < \        V P                  R R7      S4      ! 4       # )r   r\  )r]   r_  )r<   ra  s   &r   r=   /TestEWM.test_pairwise_methods.<locals>.<lambda>  s    gaeeenf57r   ))r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   NN)
r   rb   r]   r/   r_  r   r   r@   rd   r?   r`  s   &f&   r   test_pairwise_methodsTestEWM.test_pairwise_methods  s     cUQYU1X67C,,,5v>@- (( (
 	f/::c?C5)//7
 	f/r   c                   R pVP                  R4      pVP                  R4      P                  W#R7      P                  4       p\	        R. RO/\
        P                  ! . R	ORR.R7      R7      p\        P                  ! WE4       R# )
23 daysr   r   halflifetimesr   Nr   r   )
r   gCus=?g;pGR?gZkB?r   g>(?g8Kr?r   gҧU?gA9w?)
r   r   )r   r   )r   r   )r   r   r  r   )r   r   )r   r   )r   r   )	popr/   r_  r:   r   r   r   r@   rd   )r0   r   rs  rt  rE   rD   s   &&    r   
test_timesTestEWM.test_times  s    $$$S)--x-MRRT  (( Dk
> 	f/r   c                   R pVP                  R4      pVP                  R4      pVP                  W#R7      P                  4       pVP                  W#P                  R7      P                  4       p\
        P                  ! WV4       R# )rq  r   r   rr  N)ru  r/   r_  r:   valuesr@   rd   )r0   r   rs  rt  r   rE   rD   s   &&     r   test_times_arrayTestEWM.test_times_array  si    $  %7<<>668<<6@EEG
f/r   c                   \        R . ROR\        R^RR7      R\        ^4      /4      pVP                  R 4      P	                  RRR7      pVP                  4       p\        R\        R^RR7      RR.^,          /\        P                  ! . RO\        \        ^4      4      .R R	.R
7      R7      p\        P                  ! W44       VR,          P                  4       p\        R.^,          \        P                  ! . RO\        R^RR7      .R R.R
7      RR7      p\        P                  ! W54       VP                  4       p\        P                  ! W44       R	# )r   	timestampz2021-9-1h)r  r
  r   1hrr  r   Nr   r   r   )r   r   r   r   r   )r   r   rb   r/   r9   rL   r   ra   r   r@   rd   r	   rA   )r0   r   grprE   expected_dfexpected_seriess   &     r   "test_dont_mutate_obj_after_slicing*TestEWM.test_dont_mutate_obj_after_slicing  s4   /Z
ACHU1X
 jj&&t&<Z
ACHcUQY ((*DqN;D$<
 	f2S! EAI((-z13? [) 

 	v7
f2r   r   Nrh  )r!  r"  r#  r$  r,   r$  r%  r#   r   rb  ri  rn  rv  rz  r  r%  r&  r'  s   @r   rU  rU    s     [[VWRVVXx:;RVVS"46HIJ	
00$!0F [[
2663Q'	(52663(2S*TU0	0.%0N0%3 %3r   rU  c                     \        R . ROR. RO/4      p V P                  R .4      pVP                  ^4      P                  V R7      p\        P
                  ! . R	OR R.R7      p\        R \        P                  \        P                  \        P                  .R\        P                  RR./VR7      p\        P                  ! W$4       R# )
r   r   r   Nr   r   r   )r   r  r  r[   r   r   ))r  r   )r  r   )r  r   
r   r/   r9   r|   r   r   r#   r   r@   rd   r   r   rE   rc   rD   s        r   .test_rolling_corr_with_single_integer_in_indexr  D  s    	C+S)<	=B	SE	BZZ]b)F""#DSRVKXE	rvvrvvrvv&bffc3-?@H &+r   c                     \        R . ROR. RO/4      p V P                  R .4      pVP                  ^4      P                  V R7      p\        P
                  ! . R	OR R.R7      p\        R \        P                  \        P                  \        P                  .R\        P                  RR./VR7      p\        P                  ! W$4       R# )
r   r   r  Nr   r   r   )r   r   r   r  ))r   r   )r   r   )r   r   r  r  s        r   &test_rolling_corr_with_tuples_in_indexr  P  s    	  	

B& 
SE	BZZ]b)F""/T{E 	rvvrvvrvv&bffc3-?@H &+r   )numpyr#   r,   pandas.errorsr   pandasr   r   r   r   r   r	   r
   r   r   pandas._testing_testingr@   pandas.api.indexersr   pandas.core.groupby.groupbyr   rS  r   r%   r'   r'  rU  r  r  r   r   r   <module>r     s      (
 
 
  + 3  0 O OZ4 Z4z0 0Dn3 n3b	,,r   