+
    xȇib                     H   ^ RI HtH t  ^ RIt^ RIt^ RIHt ^ RIt^ RIH	t	H
t
HtHtHt ^ RIHt ^ RIHt R tR tR t]P,                  P/                  R	]	! R
. ROR. RO/4      . RO3]	! R
. ROR. RO/4      ^ ^.3]	! R
^./4      ^.3]	! R
. ROR]! ^4      /4      ^^.3]	! R
. ROR. RO/4      . RO3]	! R
]! R4      R. ROR. RO/4      . RO3]	! . RO. RO.. ROR7      ^^.3.. ROR7      R 4       tR tR tR t]P,                  P/                  RR R R R .4      R 4       tR tR t R t!R t"R  t#R! t$R" 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/R0 t0]P,                  P/                  R1R)R*.4      R2 4       t1R3 t2R4 t3R5 t4R6 t5R7 t6R8 t7R9 t8R: t9R; t:R< t;]P,                  P/                  R=R>. ROR?. RO/^^.^ ^..R?^ ^./.R>. ROR?. RO/^^.^^..R?^^./..4      R@ 4       t<RA t=]P,                  P/                  RBRCRD.4      RE 4       t>RF t?]P,                  P/                  RG] P                  ! 4       ] P                  ! 4       P                  4       ] P                  ! 4       P                  4       .4      RH 4       tBRI tC]P,                  P/                  RJR*R).4      RK 4       tDRL tE]P,                  P/                  RM]P                  ! ]! RN4      4      ]P                  ! ^ ^4      ]P                  ! RO^RPRQ7      ]P                  ! . RO4      .4      RR 4       tJ]P,                  P/                  RSRT ^ ^.^^..3RU ^ ^0^^0.3RV RR.3RW ^ ^ ^^/^ ^^^/.3RX ^ ^ /^^/.^ ^/^^/..3.4      RY 4       tKRZ tL]P,                  P/                  R[R\ R] .4      R^ 4       tMR_ tN]P,                  P/                  R`Ra Rb .4      ]P,                  P/                  R(R)R*.4      Rc 4       4       tORd tPRe tQRf tRRg tSRh tT]P,                  P/                  RiR*]	! . RO. RO.]
! . RO]URj7      R7      .R)]! ^^.]P                  ! RR.R
R.Rk7      Rl7      ..4      Rm 4       tVRn tWRo tXRp tYRq tZRr t[Rs t\]P,                  P/                  Rt. RO. RO.4      Ru 4       t]]P,                  P/                  Rv. RO. RO.. RO. RO.. RO. RO..4      Rw 4       t^]P,                  P/                  RxR)./ 3. RyR)/3.4      Rz 4       t_]P,                  P/                  R{R|R.4      R} 4       t`]P,                  P/                  R~RCR 3RCR 3RRRR.4      R 4       taR tb]P,                  P/                  R]c^3]d^3]e^3.4      R 4       tfR tgR thR tiR# )    )datedatetimeN)Pandas4Warning)	DataFrameIndex
MultiIndexSeriesbdate_range)get_groupby_method_argsc                    a \        ^\        \        ^
4      4      ^
,          ^ .R7      P                  4       p . oV3R lpV P	                  R4      P                  V4       \        ^ ^.^
,          /\        P                  ! ^ ^d^
4      R7      pVP                  P                  \        4      Vn        \        P                  ! S^ ,          V4       R# )   )indexcolumnsc                 *   < SP                  V 4       R # N)append)groupgroupss   &o/Users/max/.openclaw/workspace/postharvest/venv/lib/python3.14/site-packages/pandas/tests/groupby/test_apply.pystoreFtest_apply_func_that_appends_group_to_list_without_copy.<locals>.store   s    e    r   r   N)r   listrangereset_indexgroupbyapplypd
RangeIndexr   astypeobjecttmassert_frame_equal)dfr   expected_valuer   s      @r   7test_apply_func_that_appends_group_to_list_without_copyr'      s     
1DrOb01#	>	J	J	LBF JJwe$A38}BMM!S"4MNN+33::6BN&)^4r   c                 j   . ROp\        R. RO/\        \        P                  ! V4      RR7      R7      pVP	                  VP
                  P                  4      P                  4       pVP	                  VP
                  P                  4      P                  R 4      p\        P                  ! WC4       R# )	2011-05-16 00:00value	date_timenamer   c                 "    V P                  4       # r   idxmaxxs   &r   <lambda>'test_apply_index_date.<locals>.<lambda>K   s
    qxxzr   Nr)   z2011-05-16 01:00z2011-05-16 02:00z2011-05-16 03:00z2011-05-17 02:00z2011-05-17 03:00z2011-05-17 04:00z2011-05-17 05:00z2011-05-18 02:00z2011-05-18 03:00z2011-05-18 04:00z2011-05-18 05:00	/?ݓ?Q?j?r7   r8   r9   r:   r7   r8   r9   r:   )r   r   r   to_datetimer   r   r   r0   r   r#   r$   )using_infer_stringtsr%   expectedresults   &    r   test_apply_index_dater@   '   s    
B 
 	
  BNN2&[9#
B& zz"((--(//1HZZ&,,-ABF&+r   c                  (   . R	Op \        V  Uu. uF  qP                  4       NK  	  upRR.R7      p. R
OVR&   \        . RORR7      p\        . ROVR7      pVP	                  R4      P                  R 4      p\        P                  ! WT4       R# u upi )r)   r   timer   r*   r,   r   c                 L    V R ,          V R,          P                  4       ,          # )rB   r*   r/   r1   s   &r   r3   .test_apply_index_date_object.<locals>.<lambda>q   s    &	!G*:K:K:M0Nr   Nr5   r6   )z
2011-05-16z
2011-05-17z
2011-05-18)z00:0002:00rF   )r   splitr   r	   r   r   r#   assert_series_equal)r=   rowr%   exp_idxr>   r?   s         r   test_apply_index_date_objectrK   O   s    
B 
2.2CIIK2.8H	IBBwK >VLG1AHZZ%%&NOF6,% /s   Bzdf, group_namesabtwo
aaabbbccccBCrC   )idsc                    a . oV3R  lpV3R lpV3R lpV3R lpV3R lpW#WEV3 F/  pSR V P                  R4      P                  V4       SV8X  d   K/  Q h	  R# )c                 Z   < SP                  V P                  4       V P                  4       # r   )r   r-   copyr   namess   &r   f_copy/test_group_apply_once_per_group.<locals>.f_copy   s    UZZ zz|r   c                 >   < SP                  V P                  4       V # r   r   r-   rV   s   &r   f_nocopy1test_group_apply_once_per_group.<locals>.f_nocopy   s    UZZ r   c                 >   < SP                  V P                  4       ^ # r   r[   rV   s   &r   f_scalar1test_group_apply_once_per_group.<locals>.f_scalar   s    UZZ r   c                 >   < SP                  V P                  4       R # r   r[   rV   s   &r   f_none/test_group_apply_once_per_group.<locals>.f_none   s    UZZ r   c                 \   < SP                  V P                  4       \        R ^.R^./4      # )rL   rM   )r   r-   r   rV   s   &r   f_constant_df6test_group_apply_once_per_group.<locals>.f_constant_df   s+    UZZ #sC!-..r   NNNrL   N)r   r   )	r%   group_namesrX   r\   r`   rc   rf   funcrW   s	   &&      @r   test_group_apply_once_per_grouprk   u   sc    N E


!/
 8]C!H


3d####	 Dr   c                     ^p\        R. R	OR. R
O/. R
OR7      pVP                  RRR7      P                  R 4       V P                  4       P                  P                  R4      pW18X  g   Q hR# )   group_by_columntest_columnr   F
group_keysc                     \        R 4      # )function_called)printr%   s   &r   r3   2test_group_apply_once_per_group2.<locals>.<lambda>   s
    5*+r   rs   N)r   r   r   r   r   r   r   r   )02468101214)r   r   r   
readouterroutcount)capsysr>   r%   r?   s   &   r    test_group_apply_once_per_group2r      sx     H	7F	
 :
B JJ UJ399+  $$**+<=Fr   c                      \        R . ROR\        ^4      /4      p R pR pV P                  R RR7      P                  V4      pV P                  R RR7      P                  V4      p\        P
                  ! W44       R# )ArM   c                     V # r    r   s   &r   slow,test_apply_fast_slow_identical.<locals>.slow   s    r   c                 "    V P                  4       # r   rU   r   s   &r   fast,test_apply_fast_slow_identical.<locals>.fast   s    zz|r   Frp   N)r   r   r   )r   r   r   r   r#   r$   )r%   r   r   fast_dfslow_dfs        r   test_apply_fast_slow_identicalr      so     
CCq2	3B jjj/55d;Gjjj/55d;G'+r   c                     \        R . ROR. RO/4      P                  R 4      p V P                  R.RR7      P                  R 4      pV P                  R.RR7      P                  R 4      p\        P
                  ! W4       R# )	r-   ageFrp   c                     V # r   r   r   s   &r   r3   6test_apply_fast_slow_identical_index.<locals>.<lambda>   s    RWr   c                 "    V P                  4       # r   r   r   s   &r   r3   r      s
    ejjlr   N)AliceBobCarl)      r   )r   	set_indexr   r   r#   r$   )r%   grp_by_same_valuegrp_by_copys      r   $test_apply_fast_slow_identical_indexr      s    	,<	


 i  

E7u
=CCDWX**eW*7=="K +9r   rj   c                     V # r   r   r1   s   &r   r3   r3      s    !r   c                     V R ,          # )rh   r   r1   s   &r   r3   r3      s    !A$r   c                 &    V P                  R R7      # )Fdeepr   r1   s   &r   r3   r3      s    !&&e&$r   c                 &    V P                  R R7      # )Tr   r   r1   s   &r   r3   r3      s    !&&d&#r   c           	          \        R . ROR. ROR. RO/4      pVP                  R RR7      P                  V 4      p\        P                  ! W!RR.,          4       R# )	grL   rM   Frp   Nr   rm   rm   rm   r   rm         )            r   r   r   r#   r$   )rj   r%   r?   s   &  r   5test_groupby_apply_identity_maybecopy_index_identicalr      sN     
CsL#|L	MBZZZ.44T:F&c3Z.1r   c                     \        R \        P                  P                  ^4      P	                  ^4      R. R
O/4      p V P                  R ^R7      pT p\        P                  ! W4       \        R. RO/4      p V P                  R,          V R&   V P                  R4      P                  4       P                  4       pV P                  RRR7      P                  4       p\        P                  ! W44       R	# )foo1foo2c                     V # r   r   r1   s   &r   r3   -test_apply_with_mixed_dtype.<locals>.<lambda>  s    r   axisc1       @c2Fas_indexNonerN   rN   threer   rN   )r   rm   r   r   r   )r   nprandomdefault_rngstandard_normalr   r#   r$   r   r   meanr   )r%   r?   r>   result1result2s        r   test_apply_with_mixed_dtyper     s    	BII))!,<<Q?@	

B XXkX*FH&+ 
D/*	+Buus{BtHjj##%113Gjjj.335G'+r   c           
         \        R . R	OR. R
OR\        ^4      /4      pVP                  RV R7      p\        R . ROR. ROR. RO/. ROR7      pVP                  ^4      p\        P
                  ! WC4       V '       d    . ROp\        P                  ! VRR.R7      pM\        . RO4      p\        R \        R4      R. RO/VR7      pVP                  R 4      p\        P
                  ! WC4       R# )item_iduser_idrB   r   r   NrW   babac                 $    V P                  ^4      # rm   headr1   s   &r   r3   -test_groupby_as_index_apply.<locals>.<lambda>D  s    q	r   )rM   rM   rL   crL   rM   )r   rm   r   r   r   r   )rM   rM   rL   rL   )r   rm   r   r   )r   r   rm   r   )r   r   r   rm   )rm   r   )r   r   )r   rm   r   r   )r   r   r   r   r#   r$   r   from_tuplesr   r   r   )r   r%   gbr>   r?   tpr   s   &      r   test_groupby_as_index_applyr      s    	5)E!H	

B 
I	1B+|L	

 H WWQZF&+ -&&r)T1BCl#tF|L	
 H XX)*F&+r   c                      \        \        R 4      4      p \        ^^.^^.^^.^^.^^..V R7      pVP                  ^ RRR7      P	                  R 4      P
                  p\        P                  ! W 4       R# )abcder   F)r   rq   c                     V # r   r   r1   s   &r   r3   1test_groupby_as_index_apply_str.<locals>.<lambda>K  s    !r   N)r   r   r   r   r   r   r#   assert_index_equal)indr%   ress      r   test_groupby_as_index_apply_strr   H  sn    
W
C	QFQFQFQFQF;3	GB
**Q5*
9
?
?
L
R
RC##r   c                 R   V P                  R R.4      pR pR pR pVP                  V4      pVP                  P                  R8X  g   Q hVP                  V4      pVP                  P                  R8X  g   Q hVP                  V4      pVP                  P                  R8X  g   Q hR# )r   rP   c                 H    V P                  4       pR VP                  n        V# )stat)describer   r-   r   r?   s   & r   desc.test_apply_concat_preserve_names.<locals>.descR  s    !"r   c                 d    V P                  4       pR VP                  n        VR\        V 4       pV# )r   N)r   r   r-   lenr   s   & r   desc2/test_apply_concat_preserve_names.<locals>.desc2W  s/    !"#e*%r   c                 ~    V P                  4       pR \        V 4      R 2VP                  n        VR\        V 4       pV# )stat_dN)r   r   r   r-   r   s   & r   desc3/test_apply_concat_preserve_names.<locals>.desc3^  s>    ! $CJq>2#e*%r   N)r   rP   r   )r   rP   N)r   r   r   rW   )three_groupgroupedr   r   r   r?   r   result3s   &       r    test_apply_concat_preserve_namesr   O  s    !!3*-G
 ]]4 F<<!3333mmE"G=="4444mmE"G=="2222r   c                     R  p \        R^
R7      p\        \        P                  P	                  ^4      P                  ^
4      VR7      pVP                  R RR7      pVP                  V 4      p\        V\        4      '       g   Q h\        VR4      '       d   Q h\        P                  ! VP                  VP                  4       R# )	c                     \         P                  ! R R7      ;_uu_ 4        \         P                  ! V 4      pRRR4       \        RV RW P	                  4       ,
          RX/4      #   + '       g   i     L5; i)ignore)invalidNr*   demeanedlogged)r   errstatelogr   r   )piecer   s   & r   f%test_apply_series_to_frame.<locals>.fs  sR    [[**VVE]F +eZ)=xP
 	
 +*s   A$$A4	z1/1/2000periodsr   c                     V P                   # r   monthr1   s   &r   r3   ,test_apply_series_to_frame.<locals>.<lambda>}      177r   Frp   r-   N)r
   r	   r   r   r   r   r   r   
isinstancer   hasattrr#   r   r   )r   drr=   r   r?   s        r   test_apply_series_to_framer  r  s    
 
Z	,B			%%a(88<B	GBjj*uj=G]]1Ffi((((vv&&&&&,,1r   c                     V P                  R R.4      R,          P                  \        4      pVP                  P                  R,          R8X  g   Q hR# )r   rP   rQ   Nrm   NN)r   rP   )r   r   r   r   rW   r%   r?   s   & r    test_apply_series_yield_constantr    sA    ZZc
#C(..s3F<<b!Z///r   c                 P   V P                  R R.4      P                  \        4      p\        V\        4      '       g   Q hVP
                  e   Q hV P                  R R.4      RR.,          P                  \        4      p\        V\        4      '       g   Q hVP
                  e   Q hR# )r   rP   NrQ   D)r   r   r   r  r	   r-   r  s   & r   test_apply_frame_yield_constantr    s    ZZc
#))#.Fff%%%%;;ZZc
#S#J/55c:Fff%%%%;;r   c                 0   V P                  R R.4      pVP                  \        4      pVP                  4       R,          p\        P
                  ! VP                  VP                  4       \        P                  ! VP                  VP                  4       R# )r   rP   rQ   N)	r   r   r   r   r#   r   r   assert_numpy_array_equalvaluesr%   r   r?   r>   s   &   r   test_apply_frame_to_seriesr    sb    jj#s$G]]3F}}s#H&,,7x?r   c                    V P                  R R.RR7      pVP                  \        4      pVP                  4       P	                  R\
        P                  /R7      P                  RR7      p\        P                  ! VP                  VP                  4       \        P                  ! VP                  VP                  4       R# )r   rP   Fr   rQ   rC   r  N)r   r   r   r   renamer   nandropr#   r   r   r  r  r  s   &   r   )test_apply_frame_not_as_index_column_namer    s    jj#sej4G]]3F}}%%sBFFm%<AA#ANH&,,7x?r   c            
      0  a R  p V3R lp\        R\        P                  P                  ^4      P	                  ^ ^R4      R\        P                  P                  ^4      P	                  ^ ^R4      R\        P                  P                  ^4      P                  R4      /4      oSP                  R4      P                  V 4      pSP                  R4      R,          P                  V4      p\        P                  ! W#RR7       VP                  R8X  g   Q hR# )	c                     V P                  R 4      R,          P                  4       P                  4       P                  R,          # )rP   rQ   r  )r   sumsort_valuesilocr   s   &r   trans-test_apply_frame_concat_series.<locals>.trans  s3    }}S!#&**,88:??CCr   c                    < V P                  SP                  V P                  4      R ,          4      pVP                  4       P	                  4       P
                  R,          # )rP   r  )r   reindexr   r  r   r!  )r   r   r%   s   & r   trans2.test_apply_frame_concat_series.<locals>.trans2  sC    --

5;; 7 <={{}((*//33r   r   i  rP   rQ   F)check_namesN)r   r   r   r   integersr   r   r   r#   rH   r-   )r"  r&  r?   expr%   s       @r   test_apply_frame_concat_seriesr+    s    D4 
&&q)221a>&&q)221a>&&q)99$?	

B ZZ_""5)F
**S/#

$
$V
,C6E:;;#r   c                     V P                  R  RR7      pVP                  R 4      pVP                  R 4      p\        P                  ! W#4       R# )c                     V P                   # r   r  r1   s   &r   r3   &test_apply_transform.<locals>.<lambda>  r  r   Frp   c                     V ^,          # r   r   r1   s   &r   r3   r.    s    QUr   c                     V ^,          # r   r   r1   s   &r   r3   r.    s    1q5r   N)r   r   	transformr#   rH   )r=   r   r?   r>   s   &   r   test_apply_transformr2    sA    jj*uj=G]]?+F  1H6,r   c                     V P                  R  R .4      pR pVP                  V4      pV F3  w  rE\        P                  ! VP                  V,          V! V4      4       K5  	  R# )c                     V P                   # r   yearr1   s   &r   r3   ,test_apply_multikey_corner.<locals>.<lambda>  s    r   c                     V P                   # r   r  r1   s   &r   r3   r7    s    177r   c                 *    V P                  R 4      RR # )r   Nr   r   s   &r   r   %test_apply_multikey_corner.<locals>.f  s      %bc**r   N)r   r   r#   r$   loc)tsframer   r   r?   keyr   s   &     r   test_apply_multikey_cornerr@    sT    oo/1BCDG+ ]]1F

fjjoqx8 r   rq   TFc                 L   \        R . ROR\        ^	4      /4      pVP                  R V R7      P                  R 4      pVR.,          P	                  . RO4      pV '       d.   \
        P                  ! . ROVP                  .R R.R7      Vn        \        P                  ! W#4       R# )	r?  r*   rp   c                 (    V P                   R ,          # )r  r!  r1   s   &r   r3   'test_apply_chunk_view.<locals>.<lambda>  s    affRjr   Nr   )	r   r   r   rm   rm   rm   r   r   r   )r   r   r   r   r   r   )r   r   rm   rm   r   r   )
r   r   r   r   taker   from_arraysr   r#   r$   )rq   r%   r?   r>   s   &   r   test_apply_chunk_viewrG    s     
E6qJ	KBZZ*Z5;;<PQF7)}!!"45H#//0
 &+r   c                      \        R . ROR. ROR\        ^	RR4      /4      p V P                  R R.4      pVP                  R 4       R# )r-   name2r*   c                 (    V P                  R RR7      # )r*   Tinplacer;  r1   s   &r   r3   4test_apply_no_name_column_conflict.<locals>.<lambda>  s    AMM'4M@r   N)
r   r   r   r   r   r   rm   rm   rm   rm   )
r   r   r   r   r   r   r   r   r   r   )r   r   r   r   )r%   r   s     r   "test_apply_no_name_column_conflictrO    sM    	23U1b"%	

B jj&'*+GMM@Ar   c                  H   \        R . R
OR\        P                  ! . RO^4      R\        P                  ! RR4      /4      p R pV P	                  R RR7      P                  V4      pV RR.,          p\        P                  ! . RO^4      VR&   \        P                  ! W#4       R	# )r         ?r   v      @c                     V R ,          pWP                  4       ,
          VP                  4       VP                  4       ,
          ,          V R&   V # rR  v2minmaxr   rR  s   & r   r   #test_apply_typecast_fail.<locals>.f  7    #J557{quuw'89dr   Frp   rV  NrQ  rQ  rQ  r   r   r   rL   rM   r           g      ?r   )r   r   tilearanger   r   r#   r$   )r%   r   r?   r>   s       r   test_apply_typecast_failrc    s    	/!,3$	

B
 ZZZ.44Q7F3*~HWW]A.HTN&+r   c                     \         P                  ! . RO. RO.4      p \        R. ROR\        P                  ! . RO^4      R\        P
                  ! RR4      /V R7      pR pVP                  RRR	7      P                  V4      pVRR.,          p\        P                  ! . RO^4      VR
&   \        P                  ! W44       R# )r   r   rQ  r   rR  rS  r   c                     V R ,          pWP                  4       ,
          VP                  4       VP                  4       ,
          ,          V R&   V # rU  rW  rZ  s   & r   r   %test_apply_multiindex_fail.<locals>.f  r\  r   Frp   rV  N)r   r   r   r   r   r   r   rm   r   r   rm   r   r]  r^  r_  )
r   rF  r   r   ra  rb  r   r   r#   r$   )r   r%   r   r?   r>   s        r   test_apply_multiindex_failrh    s    ""$68J#KLE	/!,3$	

 
B
 ZZZ.44Q7F3*~HWW]A.HTN&+r   c                     V P                  R  RR7      P                  R 4      pV ^,          p\        P                  ! W4       R# )c                     V P                   # r   r5  r1   s   &r   r3   #test_apply_corner.<locals>.<lambda>  s    qvvr   Frp   c                     V ^,          # r   r   r1   s   &r   r3   rk    s    QRUVQVr   N)r   r   r#   r$   )r>  r?   r>   s   &  r   test_apply_cornerrm    s7    __-%_@FFWF{H&+r   c            	          \        R . ROR. ROR. RO/4      p R pR pV P                  R 4      P                  V4      pV P                  R 4      P                  V4      p\        P                  ! WC4       R# )	id_fieldcategoryr*   c                 |    V P                   ^ ,          ^8X  d   V P                  4       # W P                  R8H  ,          # r   r   )shaperU   rp  r1   s   &r   filt1&test_apply_without_copy.<locals>.filt1+  s.    771:?668OZZ3&''r   c                 `    V P                   ^ ,          ^8X  d   V # W P                  R8H  ,          # rr  )rs  rp  r1   s   &r   filt2&test_apply_without_copy.<locals>.filt21  s(    771:?HZZ3&''r   N)d   ry     i,  )rL   rM   r   r   r   r   )datart  rw  r>   r?   s        r   test_apply_without_copyr|    sp     ,,\	
D(( ||J'--e4H\\*%++E2F&+r   test_seriesc                 |   \        R R.R R.R R..RR.. ROR7      pV '       dV   VP                  R4      R,          pVP                  ^ RR7      P                  R 4      pTp\        P
                  ! W44       R# VP                  RRR	7      P                  R
 4      pVR.,          p\        P                  ! W44       R# )r2   poXYr   r   F)levelrq   c                     V # r   r   r1   s   &r   r3   <test_apply_with_duplicated_non_sorted_axis.<locals>.<lambda>D  s    r   rp   c                     V # r   r   r1   s   &r   r3   r  H      1r   N)r   rm   rm   )r   r   r   r   r#   rH   r$   )r}  r%   serr?   r>   s   &    r   *test_apply_with_duplicated_non_sorted_axisr  <  s     

sc3Z#s,sCj	
B ll3$17==kJ
v0CE288Ese9
f/r   c                      . ROp . ROp\        RRR.^,          RV /VR7      p\        WRR7      pR pVP                  RRR	7      P                  P	                  V4      p\
        P                  ! W54       R
# )r   r   Group1Group2r*   r   r   r-   c                     V P                  \        P                  ! V P                  P	                  4       V P                  P                  4       ^,           4      4      # r   )r%  r   rb  r   rX  rY  r1   s   &r   reindex_helper1test_apply_reindex_values.<locals>.reindex_helperW  s5    yy177;;=!''++-!2CDEEr   Frp   Nr   r   r   rm   rm   )r   r	   r   r*   r   r#   rH   )r  indicesr%   r>   r  r?   s         r   test_apply_reindex_valuesr  M  sr    
 FG	Gh1A5wGw	WBf':HF ZZEZ288>>~NF8,r   c            
      f   ^
p \         P                  P                  ^4      P                  ^ ^dV R7      p\	        RVR\         P                  P                  ^4      P                  V 4      R. R
OV ^,          ,          /4      pVP                  RRR7      pR pVP                  V4      pRV9   g   Q hR	# )
   sizer?  value1value2Frp   c                 ,    V R ,          ^,          V R&   V # )r  value3r   r   s   &r   r   "test_apply_corner_cases.<locals>.fn  s    kAo(r   r  N)foobarbazquxrL   )r   r   r   r)  r   r   r   r   )Nlabelsr%   r   r   r?   s         r   test_apply_corner_casesr  _  s     	AYY""1%..q#A.>F	6bii++A.>>qA716B	

B jj5j1G ]]1Fvr   c            	      \   \        R ^^.RR.^,          RRR./4      p V P                  R .4      P                  R 4      p\        P                  ! V P
                  4      V n        V P                  R .4      P                  R 4      p\        P                  ! VR,          VR,          4       R# )	NumberDatez
2017-03-02Strr  infc                 (    V P                   ^ ,          # r_   rC  r1   s   &r   r3   ;test_apply_numeric_coercion_when_datetime.<locals>.<lambda>  s    affQir   c                 (    V P                   ^ ,          # r_   rC  r1   s   &r   r3   r    s    AFF1Ir   N)r   r   r   r   r;   r  r#   rH   )r%   r>   r?   s      r   )test_apply_numeric_coercion_when_datetimer  v  s     
	Aq66L>A#5uuenM
B zz8*%++,?@HnnRWW%BGZZ
#))*=>F6%=(5/:r   c            
      "   \        R . ROR. ROR\        P                  ! R4      .^,          /4      p R pV P                  R 4      P	                  V4      R,          pV P
                  pV P                  Vn        \        P                  ! W#4       R# )r   rP   Tz12:31:22c                 8    V P                   ^ ,          R.,          # )r   rP   rC  r  s   &r   get_B@test_apply_numeric_coercion_when_datetime_getitem.<locals>.get_B  s    vvay#r   N)r  r      )r  3ry   )
r   r   	Timestampr   r   rP   r   r   r#   rH   )r%   r  r?   r>   s       r   1test_apply_numeric_coercion_when_datetime_getitemr    su    		lC!2C",,z:R9SVW9WX
B  ZZ_""5)#.FttHTTHN6,r   c                     R  p \        R. ROR. ROR. ROR. R	O/4      pVP                  4       p\        P                  ! VP                  4      Vn        VP                  R4      P                  V 4      P                  pVP                  R4      P                  V 4      P                  p\        P                  ! W44       R# )
c                    \        . RO\        R7      pR\        V P                  4      9   d=   \	        W P                  R8H  ,          P
                  P                  ^ ,          4      VR &   R\        V P                  4      9   dy   \	        W P                  R8H  ,          P
                  P                  ^ ,          4      VR&   \	        W P                  R8H  ,          P                  P                  ^ ,          4      VR&   V# )p1p2useTime)r   dtypestep1step2)r  r  r  )r	   r"   r   StatestrMachiner  oTime)toolr   s   & r   predictionsGtest_apply_numeric_coercion_when_datetime_with_nat.<locals>.predictions  s    2&Ad4::&&Dw!67??FFqIJCId4::&&Dw!67??FFqIJCI jjG&;!<!B!B!I!I!!LMC	N
r   Keyr  r  r  N)rP   rP   r   r   )r  r  r  r  ) z2016-09-19 05:24:33r  z2016-09-19 23:59:04)2336L36Rr  )
r   rU   r   r;   r  r   r   r  r#   rH   )r  df1df2r>   r?   s        r   2test_apply_numeric_coercion_when_datetime_with_natr    s     '9K2		
C ((*Csyy)CI{{5!''477H[[%%k255F8,r   c            
         \        R . R	OR\        P                  ! R4      .^,          /4      p V P                  V P                  ,
          V R&   V P	                  R 4      P                  R 4      p\        R . R	OR\        P                  ! ^ R4      .^,          R\        P                  ! R4      .^,          /4      P                  R 4      p\        P                  ! W4       R# )
clientidr   z2017-02-01 00:00:00time_delta_zeroc                     \        R V P                  P                  4       RV P                  P                  4       /4      # )clientid_ager   )r	   r  rX  r   )ddfs   &r   r3   ?test_apply_aggregating_timedelta_and_datetime.<locals>.<lambda>  s1    FS00446@P@P@RS
r   r  r  r   Nr   rP   rQ   )
r   r   
datetime64r   r   r   timedelta64r   r#   r$   r%   r?   r>   s      r   -test_apply_aggregating_timedelta_and_datetimer    s     
'<=>B	

B KK"++5BZZ
#))	
F
 R^^As34q8R]]#89:Q>	
 i
  &+r   c                  (   R ^
.R^.R^.R^(.R^2..p \        V RR.\        P                  ! RR4      R7      pVP                  R4      P	                  4       p\        R. ROR. RO/4      pVP                  RRR	7       \        P                  ! W#4       R
# )r   rP   rQ   NameValuez
2020-09-01z
2020-09-05r  TrK  Nr  )r  2   Z   )r   r   
date_ranger   r  r   r#   r$   )r{  r%   r?   r>   s       r    test_apply_groupby_datetimeindexr    s     "IRy3)c2Yb	BD	vw'r}}\</X
B ZZ##%F&/7LIJHvt,&+r   c            
      &   \        R ^R\        ^
4       U u. uF  p \        P                  ! 4       NK  	  up /4      pR pR pVP	                  R .R7      P                  V4      p\        R^/^.R7      pR VP                  n        VP	                  R .R7      P                  V4      p\        R\        P                  ! R^^4      R^/^.R7      pR VP                  n        \        P                  ! WE4       \        P                  ! Wg4       R# u up i )	rL   rM   c                     \        R ^/4      # )r   r	   batchs   &r   func_with_no_date.test_time_field_bug.<locals>.func_with_no_date  s    sAhr   c                 6    \        R \        R^^4      R^/4      # )rM     r   )r	   r   r  s   &r   func_with_date+test_time_field_bug.<locals>.func_with_date  s    sHT1a0#q9::r   byr   r   r  N)r   r   r   nowr   r   r   r-   r   r  r#   r$   )nnr%   r  r  dfg_no_conversiondfg_no_conversion_expecteddfg_conversiondfg_conversion_expecteds           r   test_time_field_bugr    s     
CC59!E9R(,,.9!EF	GB ; 

se
,223DE!*C8A3!?,/$$)ZZC5Z)//?N'	bll4A&Q/s *-!!&+H.B' "Fs   Dc                     a \        R . R	OR. R
OR. ROR. RO/4      p V P                  R R.4      p V P                  R R.RR7      pR oVP                  V3R l4       R# )group1group2weightr*   T)r  sortc                 \    \         P                  ! W,          4      P                  ^4      pV# )r   )r   arrayrepeat)r*   r  r   s   && r   noddy7test_gb_apply_list_of_unequal_len_arrays.<locals>.noddy  s"    hhu~&--a0
r   c                 >   < S! V P                   V P                  4      # r   )r*   r  )r2   r  s   &r   r3   :test_gb_apply_list_of_unequal_len_arrays.<locals>.<lambda>  s    uQWWahh7r   N)rL   rL   rL   rM   rM   rM   rL   rL   rL   rM   rM   rM   )r   r   r   r   r   er   r   r   r   r   r  )g?rm   r   r   r   r   rm   r   r   r   r   rm   )gffffff@r   	   r        r   r   r   r   r   r   )r   r   r   r   )r%   
df_groupedr  s     @r   (test_gb_apply_list_of_unequal_len_arraysr    sl    	RR<>		

B 
x*	+B8X"6TBJ 78r   c                      \        R . ROR. RO/4      p R pV P                  R 4      P                  V4      p\        R.RR7      p\        P                  ! W#4       R# )r   random_varsc                     R # r   r   r1   s   &r   	test_func.test_groupby_apply_all_none.<locals>.test_func  s    r   int64)r   r  Nr   r   r   r   )r   r   r   r   r   )test_dfr  r?   r>   s       r   test_groupby_apply_all_noner    sS     <MNG __X&,,Y7F-@H&+r   zin_data, out_idx, out_datar   varsc                     \        V 4      pR  pVP                  R4      P                  V4      p\        P                  ! VRR.R7      p\        W&R7      p\
        P                  ! WW4       R# )c                 `    V P                   ^ ,          ^8  d   R# V P                  ^ R.,          # )r   NrN  )rs  r!  r1   s   &r   r  0test_groupby_apply_none_first.<locals>.test_func1  s&    771:>vvq"gr   r   Nr   r   )r   r   r   r   rF  r#   r$   )in_dataout_idxout_datatest_df1r  r   index1	expected1s   &&&     r   test_groupby_apply_none_firstr    s_    " !H
 x(..y9G##GHd3CDF(1I'-r   c            	         \        R ^ ^.RRR./4      p V P                  R4      pVP                  R 4      p\        ^ .R \        P
                  ! RR.^ ..RR.R7      P                  R4      R7      p\        P                  ! W#4       R# )r*   r   filledemptyc                 :    W P                   ^8g  ,          R,          # )r   r*   )r*   r   s   &r   r3   7test_groupby_apply_return_empty_chunk.<locals>.<lambda>@  s    kkQ.>(?(Hr   Nr   )r-   r   )	r   r   r   r	   r   from_productr  r#   rH   )r%   r   r?   r>   s       r   %test_groupby_apply_return_empty_chunkr#  <  s    	GaVWx.AB	CBZZ F\\HIF	
%%x 1#&wo

$w-H 6,r   methr   r1  c           	          \        R RP                  4       R. ROR. R	O/4      pVP                  R RR7      p\        W 4      ! R 4      p\        R. R
OR. RO/4      p\        P
                  ! W44       R# )r   za a brP   rQ   Frp   c                 .    W P                  4       ,          # r   r  r1   s   &r   r3   -test_apply_with_mixed_types.<locals>.<lambda>Q  s    EEGr   Nr   rm   r   r   r   r   )gUUUUUU?gUUUUUU?r   )皙?g333333?rQ  )r   rG   r   getattrr#   r$   )r$  r%   r   r?   r>   s   &    r   test_apply_with_mixed_typesr-  K  se     
C#y#yI	JB


35
)AQ34F#4c?KLH&+r   c                     \        R ^^./\        ^^.4      R7      p V P                  R 4      P                  R 4      p\	        \        ^.4      \        ^.4      .\        ^^.R R7      R7      p\
        P                  ! W4       R# )rL   r   c                     V P                   # r   r   r  s   &r   r3   *test_func_returns_object.<locals>.<lambda>Y  s    QWWr   r,   N)r   r   r   r   r	   r#   rH   r  s      r   test_func_returns_objectr1  V  sm    	C!Q=q!f	6BZZ_""#45FuaSz5!:.eQF6MNH6,r   group_column_dtlikec           	          \        R R.RV ./4      pVP                  R 4      P                  R 4      p\        R.\        R.RR R7      ^*.R7      p\        P
                  ! W#4       R# )	rL   r  rM   c                      \        R .^*.R7      # )spamr   r  r1   s   &r   r3   +test_apply_datetime_issue.<locals>.<lambda>j  s    VVHRD-Ir   r5  r  r  r-   rC   Nr   r   r   r   r#   r$   )r2  r%   r?   r>   s   &   r   test_apply_datetime_issuer9  _  sc     
C%#(;'<=	>BZZ_""#IJF&5%C#HSURVWH&+r   c                     \        R ^ \        P                  ! R4      ^\        P                  ! R4      ^\        P                  ! R4      ^\        P                  ! R4      ^\        P                  ! R4      /R^ R^R^R^R^R/R^ R^R^R^R^R//4      p R pV P                  R 4      P	                  V4      R,          p\        R.\        P                  ! R.R R	7      RR
7      p\        P                  ! W#4       R# )dayz2015-02-24 00:00:00	userAgentzsome UA stringzanother UA stringuserId17661101c           	          \        V P                  4        UUu/ uF&  w  rWP                  4       P                  ^ ,          bK(  	  upp4      # u uppi r_   )r	   itemsvalue_countsr   )r%   r   ss   &  r   most_common_valuesEtest_apply_series_return_dataframe_groups.<locals>.most_common_values  s<    
K
q..*0033
KLLKs   ,A
z
2015-02-24r,   r  N)	r   r   r  r   r   r	   DatetimeIndexr#   rH   )tdfrC  r?   r>   s       r   )test_apply_series_return_dataframe_groupsrG  p  s   
2<< 562<< 562<< 562<< 562<< 56 ###&# :::::	
C4M [[%%&89(CF	B,,l^%HxH 6,r   rp  c                 T   ^^.^,          pV '       d=   \         P                  ! V. ROR7      p\         P                  ! . RO. RORR7      p. ROpM\        ^^.RR7      p^^.p\	        RVRV/VR7      p\	        R\
        P                  ! ^
4      RVR\        \        ^
4      4      R\        \        ^
4      4      /4      P                  RR.4      pVP                  RR	R
7      P                  R 4      p\        P                  ! Wd4       VP                  P                  RR.8X  g   Q hR# )r   )
categoriesrP   )rI  r-   r,   rQ   r  r   r   F)observedc                 "    V P                  4       # r   r'  r1   s   &r   r3   -test_apply_multi_level_name.<locals>.<lambda>  s
    QUUWr   Nr)  )r      r   )r   CategoricalCategoricalIndexr   r   r   rb  r   r   r   r   r   r#   r$   r   rW   )rp  rM   expected_indexexpected_valuesr>   r%   r?   s   &      r   test_apply_multi_level_namerR    s     
A
ANN13,,Y9SVW%1vC0r(	osO4NH 
	biimS!S$uRy/3U2YP
ic
  ZZeZ,223DEF&+88>>c3Z'''r   c           
      l   \         P                  ! \        P                  ! R ^^4      RR^R3\        P                  ! R^^4      RR^R3\        P                  ! R^^4      R	R
^R3\        P                  ! R^^4      R	R^R3.. ROR7      pVP	                  R4      P                  R 4      P                  pV '       d&   \        P                  ! \        P                  R7      M\        p\        \        P                  ! R4      V\        P                  V.. ROR7      p\        P                  ! W$4       R# )i  reddarkr{   r  greenstormy9i  bluebrightr|   i  calmpotatocolorrC   c                 (    V P                   ^ ,          # r_   rC  r  s   &r   r3   ;test_groupby_apply_datetime_result_dtypes.<locals>.<lambda>  s    166!9r   )na_valuezdatetime64[us]r   N)observationr]  mood	intensityscore)ra  rb  rc  rd  )r   from_recordsr   r  r   r   dtypesStringDtyper   r  r"   r	   r  r  r#   rH   )r<   r{  r?   r  r>   s   &    r   )test_groupby_apply_datetime_result_dtypesrh    s    !!\\$1%ufa=\\$1%w!SA\\$1%vxDA\\$1%vvq(C		
 GD \\'"(()<=DDF/ABNNBFF+vE	"	#UBHHe<;H 6,r   r   abc2020r  )r  freqc                     \        R . ROR. RO/V R7      pVP                  R RR7      P                  R 4      p\        P                  ! W!R.,          4       R# )	r   r*   r   Frp   c                     V # r   r   r1   s   &r   r3   8test_apply_index_has_complex_internals.<locals>.<lambda>  r  r   Nr   r   rm   )r   r   r   r   )r   r%   r?   s   &  r   &test_apply_index_has_complex_internalsrp    sJ     
GY;5	IBZZEZ288EF&gY-0r   zfunction, expected_valuesc                 6    V P                   P                  4       # r   )r   to_listr1   s   &r   r3   r3     s    177??$r   c                 H    \        V P                  P                  4       4      # r   )setr   rr  r1   s   &r   r3   r3     s    3qww()r   c                 H    \        V P                  P                  4       4      # r   )tupler   rr  r1   s   &r   r3   r3     s    5*+r   c                 Z    \        \        V P                  P                  4       4      4      # r   )dict	enumerater   rr  r1   s   &r   r3   r3     s    d9QWW__%678r   c                 ~    \        V P                  P                  4       4       UUu. uF  w  rW/NK
  	  upp# u uppi r   )ry  r   rr  )r2   nis   &  r   r3   r3     s,    Iaggoo6G,HI,H&1v,HIIs   9c                     \        . ROR.R7      pVP                  R4      P                  V 4      p\        V\	        R R.RR7      R7      p\
        P                  ! W44       R# )r   rP   r   rC   r,   r   N)r   r   rP   rP   r   r   r   r	   r   r#   rH   )functionrQ  r%   r?   r>   s   &&   r   1test_apply_function_returns_non_pandas_non_scalarr    sQ    $ 
'(	<BZZ!''1FoUC:H-MNH6,r   c            	         R  p \        R. R	OR^^^\        P                  ./4      pVP                  R4      P	                  V 4      p\        RR.R.\        P                  ..\        . R
ORR7      R7      p\        P                  ! W#4       R# )c                 D    V R ,          P                   P                  4       # )rP   )r  flattenr   s   &r   fct4test_apply_function_returns_numpy_array.<locals>.fct  s    Sz  ((**r   r   rP   rQ  r         @r,   r   N)rL   rL   rM   none)rL   rM   r  )	r   r   r  r   r   r	   r   r#   rH   )r  r%   r?   r>   s       r   'test_apply_function_returns_numpy_arrayr    s|    + 
C0#1a7HI	JBZZ_""3'F
scURVVH%U3EC-PH 6,r   r  c                     V P                   # r   r   grs   &r   r3   r3     s    r   c                 6    V P                   ^,           ^,
          # r  r   r  s   &r   r3   r3     s    bhhQRlUVFVr   c                    \        . ROR.R7      pVP                  R4      P                  V 4      p\        \	        . RO4      \	        . RO4      \	        ^^.4      .\	        . R	ORR7      R7      p\
        P                  ! W#4       R# )
r   idrC   r,   r   N)
r   rm   rm   rm   r   rm   r   r   r   r   )r   r   r   r  )r   rm   r   r   r)  r~  )r  r%   r?   r>   s   &   r    test_apply_function_index_returnr    sl     
1D6	BBZZ##H-F	|	eL15!Q=AID)H 6,r   c                      \        R . ROR. RO/4      p V P                  R RR7      P                  R 4      p\        R . ROR. R	O/4      p\        P                  ! W4       R# )
r   r   Fr   c                 "    V P                  4       # r   )r   r1   s   &r   r3   Atest_apply_function_with_indexing_return_column.<locals>.<lambda>  s
    r   Nr   )r   rm   r   r   r   r   )r   r   rN   )r        @r  r   r  s      r   /test_apply_function_with_indexing_return_columnr    sg    	@&	

B ZZZ/556HIF+O	
H &+r   udfc                 "    V P                  4       # r   r   r1   s   &r   r3   r3     s
    qvvxr   c                 B    V P                  4       P                  R  4      # )c                     V ^,           # r  r   ys   &r   r3   <lambda>.<locals>.<lambda>  s    QUr   )rU   r  r1   s   &r   r3   r3     s    1668???#Cr   c                    \        R RR.R^^./4      pVP                  R V R7      P                  V4      pVP                  P                  VP                  V R7      P                  V4      pV '       d=   VP
                  P                  ^8X  g   Q hVP
                  P                  ^8X  g   Q hR# VP
                  P                  ^8X  g   Q hVP
                  P                  ^8X  g   Q hR# )r   rL   rM   rP   rp   N)r   r   r   rP   r   r   nlevels)rq   r  r%   	df_resultseries_results   &&   r   test_apply_result_typer    s     
C#sS1a&1	2B

3:
6<<SAIDDLL*L=CCCHM&&!+++""**a///&&!+++""**a///r   c                      \        R . ROR. RO/4      p V P                  R RR7      P                  R 4      pV P                  R RR7      P                  R 4      p\        P                  ! W4       R# )	r   rP   Frp   c                     V # r   r   r1   s   &r   r3   4test_result_order_group_keys_false.<locals>.<lambda>.  s    qr   c                 "    V P                  4       # r   r   r1   s   &r   r3   r  /  s
    r   N)rm   r   rm   r)  r   r  s      r   "test_result_order_group_keys_falser  *  s`     
CC3	4BZZZ.44[AFzz#%z0667IJH&+r   c                  6   R .^,          R.^,          ,           R.^,          ,           p \         P                  ! V 4      p\         P                  ! V RR7      p\        R\        \	        ^4      4      ^,          R\	        ^4      RV/4      p\        R\        \	        ^4      4      ^,          R\	        ^4      RV/4      pVP                  RRR	7      P                  R
 4      pVP                  RRR	7      P                  R 4      p\        P                  ! WV4       R# )z
2001-01-01z
2001-01-02z
2001-01-03UTC)tzr2   r  tFrp   c                 2    V R .,          P                  4       # r  r   ru   s   &r   r3   1test_apply_with_timezones_aware.<locals>.<lambda>;      "cU)..BRr   c                 2    V R .,          P                  4       # r  r   ru   s   &r   r3   r  <  r  r   N)	r   rE  r   r   r   r   r   r#   r$   )datesindex_no_tzindex_tzr  r  r   r   s          r   test_apply_with_timezones_awarer  3  s    NQ,!!33|nq6HHE""5)K%0H
S$uQx.1,c58S+N
OC
S$uQx.1,c58S(K
LCkk#%k0667RSGkk#%k0667RSG'+r   c           	      n   \        R . ROR. ROR. RO/4      p\        R^^.R^^<./\        ^X^c.R R7      R7      pVP                  R R7      pVP                  \        P
                  ^ R7      p\        P                  ! WB4       VP                  R R7      p\        W4      pV R8X  d
   \        pRpMR	pR
p\        P                  ! WgR7      ;_uu_ 4        \        W04      ! V!  pR	R	R	4       VP                  \        P
                  ^ R7      p\        P                  ! WB4       R	#   + '       g   i     LI; i)rL   rM   r   r,   r   r  r   corrwithz'DataFrameGroupBy.corrwith is deprecatedNr  match)c   r  r  X   r  r  )r   rm   r   r   r   r   )r  r   r  (   r  <   )r   r   r   r   r   r  r#   r$   r   r   assert_produces_warningr,  )	reduction_funcr%   r>   grpr?   argswarnmsg_s	   &        r   ;test_apply_is_unchanged_when_other_methods_are_called_firstr  A  s    
)#)	

B 	r1gsS"I&RH3'H ***
CYYrvvAY&F&+ ***
C">6D#7		#	#D	4	4C($/ 
5YYrvvAY&F&+ 
5	4s   D$$D4	c                  P   \        R . R
OR\        R^^
4      \        R^^
4      \        R^^
4      \        R^^
4      .R. RO/\        . RORR7      R7      p V P                  R R.4      pVP	                  R 4      pV P
                  . RO,          pVP                  4       p\        P                  ! V. RO,          4      Vn	        VP                  . ROR7      p\        P                  ! W#4       VP                  P                  ^,           F  p\        V4      \        J d   K  Q h	  R	# )r   rP   i  rQ   idxr,   r   c                 $    V P                  ^4      # r  r   r1   s   &r   r3   Rtest_apply_with_date_in_multiindex_does_not_convert_to_timestamp.<locals>.<lambda>w  s    r   rC   N)rL   rL   rL   rM   r   )ry  e   f   g   )r   rm   r   )r   rP   r  )r   r   r   r   r   r!  r   r   
from_framer   r  r#   r$   levelstype)r%   r  r?   r>   vals        r   @test_apply_with_date_in_multiindex_does_not_convert_to_timestampr  e  s	    
%T1b!T1b!T1b!T1b!	 		
 (u5
B **c3Z
 CYY*+Fwwy!H##%H**84E+FGHN}}%6}7H&+||""1%%CyD    &r   c           
      n   \        R . R
ORR\        P                  \        P                  RR./\        R4      R7      pVP	                  RV RR7      P                  R 4      pV '       d   VP                  4       R .,          MVR .,          P                  . RO,          p\        P                  ! W#4       R	# )colr   rL   rM   xxyxzr   F)dropnarq   c                     V # r   r   r1   s   &r   r3   5test_apply_dropna_with_indexed_same.<locals>.<lambda>  s    RSr   Nr   rm   r   r   r   )r   r   r   rm   r   )
r   r   r  r   r   r   r  r!  r#   r$   )r  r%   r?   r>   s   &   r   #test_apply_dropna_with_indexed_samer    s     
?c26626634	
 7m
B ZZ5ZAGGTF'-ryy{E7#2ug;3C3CO3TH&+r   zas_index, expectedr  r   r   c           	          \        R . ROR. ROR. RO/4      pVP                  R R.V R7      P                  R 4      p\        P                  ! W14       R# )rL   rM   r   r   c                     ^# r  r   r1   s   &r   r3   5test_apply_as_index_constant_lambda.<locals>.<lambda>      qr   Nr  )r   r   r   r   )r   r   r   r#   assert_equal)r   r>   r%   r?   s   &&  r   #test_apply_as_index_constant_lambdar    sH    & 
CsL#|L	MBZZc
XZ6<<[IFOOF%r   c            	         \        R . ROR. R	OR. R
O/\        ^4      R7      p V P                  R4      P                  R 4      p\	        \        ^^4      \
        P                  ! . RORR.R7      R R7      p\        P                  ! W4       R# )r   rP   rQ   r   c                 6    V P                   P                  4       # r   )r   
sort_indexr1   s   &r   r3   (test_sort_index_groups.<locals>.<lambda>  s    QSS^^-=r   Nr   r  r  )r   r   r   r  r   )r   r   r   rm   rm   )r   r   r   r   rm   r   )rm   r   )	r   r   r   r   r	   r   r   r#   rH   r  s      r   test_sort_index_groupsr    s}    		osOS/JAh
B ZZ_""#=>Fa$$4S$K
 H 6,r   c            
      R   \        R \        P                  ! RR^R7      R\        ^4      R\	        R4      /4      p V P                  V P                  V P                  P                  P                  .RR7      P                  R	 4      p\        P                  ! WR R.,          4       R
# )r   z
2010-01-0112h)rk  r  valsletr   Frp   c                 (    V P                   R ,          # ):r   NNrC  r1   s   &r   r3   ;test_positional_slice_groups_datetimelike.<locals>.<lambda>  s    affRjr   N)r   r   r  r   r   r   r  r   dtr   r#   r$   )r>   r?   s     r   )test_positional_slice_groups_datetimeliker    s    BMM,UAFE!H4=	
H 	x}}'',,-%  e !  &FF+;"<=r   c            	          \        R . ROR. ROR. RO/4      p V P                  R 4      pVRR.,          P                  R 4      p\        RRR.RRR./\        RR.R R	7      R
7      p\        P
                  ! W#4       R# )r   rL   rM   rP   rQ   c                 r    V P                  \        4      P                  4       V P                  4       ,
          # r   )r!   floatrY  rX  r1   s   &r   r3   7test_groupby_apply_shape_cache_safety.<locals>.<lambda>  s     AHHUO,?,?,AAEEG,Kr   rQ  r`  r   r,   r   N)rL   rL   rM   r)  r*  r8  )r%   r   r?   r>   s       r   %test_groupby_apply_shape_cache_safetyr    sz    	C#y#yI	JB	CBc
^!!"KLF	sCj#Sz*%c
2MH &+r   c            	      .   \         P                  ! R . ROR. R	OR. R
O/4      p V P                  R 4      RR.,          pVP                  R 4      p\        P
                  ! . RO. RO.R R.R7      p\        . ROVRR7      p\        P                  ! W$4       R# )rL   b1b2c                 >    V P                  4       P                  4       # r   )unstackrA  r1   s   &r   r3   3test_groupby_apply_to_series_name.<locals>.<lambda>  s    !9!9!;r   N)arraysrW   r   r  )rL   rM   rL   rM   )aaacr  ad)r  r  r  r  )rL   rL   rM   rM   rM   )r  r  r  r  r  )r   r   rm   r   r   )	r   	from_dictr   r   r   rF  r	   r#   rH   )r%   r  r?   expected_idxr>   s        r   !test_groupby_apply_to_series_namer    s    			%**	

B **S/4,
'CYY;<F)))+IJDkL o\HH6,r   c                 
   \        R . ROR. ROR^^\        P                  \        P                  ./4      pVP                  R V R7      pVP	                  R 4      pVP	                  R 4      p\
        P                  ! W44       R# )	r  r  z)r  c                 &    V P                  ^R4      # )r   r  )nlargest)grp_dfs   &r   r3   test_apply_na.<locals>.<lambda>  s    3(?r   c                 F    V P                  R RR7      P                  ^4      # )r  F)	ascending)r   r   r1   s   &r   r3   r    s    Q]]3%]%H%M%Ma%Pr   Nr  )r   r   rm   r   )r   r   r  r   r   r#   r$   )r  r%   dfgrpr?   r>   s   &    r   test_apply_nar    sm    		c<q!RVVRVV6LM
B JJuVJ,E[[?@F{{PQH&+r   c                     \        R . R	OR. R
OR\        P                  ! . RORR7      /4      P                  R R.4      P	                  R 4      p \        \        P                  ! ^RR7      .\        P                  ! ^RR7      ..R.\
        P                  ! RR.R R.R7      R7      p\        P                  ! W4       R# )rL   rM   r   rB  )unitc                 (    V P                   R,          # r   rN  rC  ru   s   &r   r3   8test_apply_empty_string_nan_coerce_bug.<locals>.<lambda>   s    "''"+r   r   r  Nr  )r  r  r  r  r   )r   r  )rm   r  )	r   r   r;   r   r   r   r   r#   r$   )r?   r>   s     r   &test_apply_empty_string_nan_coerce_bugr    s     	\%R^^Ls;	
 
#s		%	&  
..
%	&)D(EF$$gw%7SzJH
 &+r   index_valuesc                     \        R . ROR. RO/\        V 4      R7      p\        R. R	O/\        . R
OR R7      R7      pVP                  R 4      P                  R 4      p\        P
                  ! W4       R# )rL   rM   r   b_meanr,   c                 J    \        V R ,          P                  4       .R.R7      # )rM   r  r   )r	   r   ru   s   &r   r3   0test_apply_index_key_error_bug.<locals>.<lambda>  s    62c7<<>*8*=r   N)r  a2a3r)  )r   r  rQ  )r  r  r  )r   r   r   r   r#   r$   )r  r?   r>   s   &  r   test_apply_index_key_error_bugr  
  sy     #	
 L!F o	
 &S1	H ^^C &&=F &+r   zarg,idxc                     \        R V RV /VR7      pVP                  R RR7      P                  R 4      pVR.,          p\        P                  ! W44       R# )r  r  r   Frp   c                     V # r   r   r1   s   &r   r3   5test_apply_nonmonotonic_float_index.<locals>.<lambda>L  s    r   Nr   )argr  r%   r?   r>   s   &&   r   #test_apply_nonmonotonic_float_indexr     sP    V 
E3s+3	7BZZ%Z066{CF5'{H&+r   zargs, kwargsnumeric_onlyc                     V P                  R 4      pVP                  ! R.VO5/ VB pVP                  RR7      p\        P                  ! WE4       R# )r   r  T)r  N)r   r   r  r#   r$   )r%   r  kwargsr   r?   r>   s   &&&   r   test_apply_str_with_argsr  Q  sG     
CBXXe-d-f-Fvv4v(H&+r   r-   	some_namec                     \        ^^.V R7      pVP                  RR.RR7      P                  R 4      p\        ^^.V R7      p\        P                  ! W#4       R# )r   r,   rL   Frp   c                     V # r   r   r1   s   &r   r3   1test_result_name_when_one_group.<locals>.<lambda>^  r  r   N)r	   r   r   r#   rH   )r-   r  r?   r>   s   &   r   test_result_name_when_one_groupr#  Z  sS     !Qd
#C[[#s[6<<[IFq!f4(H6,r   z
method, opc                 (    V P                   R,          # r  )r  r   s   &r   r3   r3   g  s    RYYr]r   c                 6    V R ,          P                   ^ ,          # )rM   rC  r%  s   &r   r3   r3   h  s    RW\\!_r   c                     \        R . R. /4      pVP                  R RR7      pVP                  p\        W@4      ! V4      p\	        . RR\        . RR R7      R7      p\        P                  ! WV4       R# )rL   rM   Trp   float64r7  )r-   r  r   N)r   r   rM   r,  r	   r   r#   rH   )methodopempty_dfr   r   r?   r>   s   &&     r   test_empty_dfr,  d  sp     #r3+,H			#$		/BDDEU#B'F
IU2YS-QH 6,r   c                      \        R . ROR. R	O/4      p V P                  R 4      p\        P                  ! \        RR7      ;_uu_ 4        VP                  R RR7       RRR4       R#   + '       g   i     R# ; i)
rL   rM   z(include_groups=True is no longer allowedr  c                 "    V P                  4       # r   r'  r1   s   &r   r3   %test_include_groups.<locals>.<lambda>  s
    1557r   T)include_groupsNro  r   r   r   )r   r   pytestraises
ValueErrorr   )r%   r   s     r   test_include_groupsr5  }  sU    	CC3	4B	CB	z)S	T	T
"48 
U	T	T	Ts   	A((A9	zfunc, valuec           	          \        ^ . RO^. RO^. RO/4      pVP                  ^ 4      pVP                  V 4      p\        W.\	        ^^.^ R7      R7      p\
        P                  ! WE4       R# )r   r,   r   Nro  r1  r~  )rj   r*   r%   r   r?   r>   s   &&    r   test_builtins_applyr7    s\     
Ay!Y9=	>B	ABXXd^FunE1a&q,ABH6,r   c                  J   \        R . ROR\        \        P                  ! ^4      RR7      R\        P
                  ! R^R7      /4      p R	 pV P                  R 4      P                  4       R.,          pV P                  R 4      P                  V4      R.,          p\        P                  ! W24       R
 pV P                  R 4      P                  V4      R.,          pVP                  4       p\        P                  VP                  R&   \        P                  ! W54       R pV P                  R 4      P                  V4      R.,          pVP                  4       p\        P                  VP                  R&   \        P                  ! W54       R pV P                  R 4      P                  V4      R.,          pV P                  R 4      P                  4       R.,          p\        P                  VP                  R&   \        P                  ! W54       R pV P                  R 4      P                  V4      pV P                  R 4      P                  4       R,          P                  4       p\        P                  VP                  R&   RVn        \        P                   ! W54       R# )r   TigerPonyrP   r  r  rQ   20130101r  c                 (    V P                   ^ ,          # r_   rC  r  s   &r   f_0*test_inconsistent_return_type.<locals>.f_0  s    xx{r   c                 N    V P                   R 8X  d   R# V P                  ^ ,          # )r9  Nr-   r!  r=  s   &r   f_1*test_inconsistent_return_type.<locals>.f_1  s    88wxx{r   c                 N    V P                   R 8X  d   R# V P                  ^ ,          # r:  NrA  r=  s   &r   f_2*test_inconsistent_return_type.<locals>.f_2      88vxx{r   c                 N    V P                   R 8X  d   R# V P                  ^ ,          # rE  rA  r=  s   &r   f_3*test_inconsistent_return_type.<locals>.f_3  rH  r   c                 p    V P                   R 8X  d   R# V P                  ^ ,          P                  R,          # )r:  NrQ   )r-   r!  r=  r=  s   &r   f_4*test_inconsistent_return_type.<locals>.f_4  s(    88vxx{s##r   N)r9  r9  r9  LambrO  r:  r:  )r   r	   r   rb  r   r  r   firstr   r#   r$   rU   r  r=  NaTr-   rH   )	r%   r>  r>   r?   rB  r  rF  rJ  rM  s	            r   test_inconsistent_return_typerR    s    
L		!G4z15	

B zz#$$&u-HZZ_""3'.F&+
 ZZ_""3'.FAVVAEE'N&$
 ZZ_""3'.FAFFAEE&M&$
 ZZ_""3'.F


3&AFFAEE&M&$$
 ZZ_""3'F


3$))+AFFAEE&MAF6%r   c                  b   \        R R.^,          R.^,          ,           RR.^,          R.^,          ,           R.^,          ,           R.^,          ,           R.^,          ,           R.^,          ,           R.^,          ,           R\        P                  P                  ^4      P	                  ^d^R	7      /4      p R
 p\        R \        R4      R\        R4      R. RO/4      P                  R R.4      R,          pV P                  R 4      P                  V4      p\        P                  ! W24       R# )cat1rL   rM   cat2r   r   r  r   r  r  c                     V P                  4       p V P                  P                  R R7      V R&   V P                  R4      R,          P	                  4       # )rX  )r)  rankrU  )rU   r  rW  r   rX  r1   s   &r   r   (test_nonreducer_nonstransform.<locals>.f  sB    FFHEEJJeJ,&	yy (,,..r   aaaabbbcdefcderW  N)r  r   g      @rQ  r   r  rQ  )r   r   r   r   r)  r   r   r   r   r#   rH   )r%   r   r>   r?   s       r   test_nonreducer_nonstransformr[    s%    
SEAI	)SEAIeaieai eai eai	
 eai eai 299((+44Sr4B
	

B/
 DODO7	
 i !&*H ZZ%%a(F6,r   c            	        a \         P                  P                  ^*R7      p \        RV P	                  ^
^RR7      R. RO/4      p/ oV3R lpVP                  R4      P                  V4       VP                  ^ ^.^ .3,          pVP                  ^^.^ .3,          p\        P                  ! S^ ,          V4       \        P                  ! S^,          V4       R# )	*   )seedr   r  rP   c                 @   < V P                  4       S\        S4      &   R # r   )rU   r   )r2   r   s   &r   addstore/test_groupby_apply_store_copy.<locals>.addstore  s    FFHc%jr   N)r   )r   rm   r   rm   )
r   r   r   r   normalr   r   r!  r#   r$   )rngr%   r`  expected_out_0expected_out_1r   s        @r   test_groupby_apply_store_copyrf    s    
))

R

(C	B.	

B E% JJsO(#WWaVaS[)NWWaVaS[)N%(N3%(N3r   )r   r   r   rm   r   )rL   rL   rL   rM   r   r)  r  )r   r   r   r   )r   r   r   rm   rm   r   r   rm   rg  )r   r   r   r   r   r  )
r   r   r   r   r   rm   r   r  r   r   )
r   r   rm   rm   rm   r   r   r   rm   r   r^  )rm   rm   r   )GH2936zGH7739 & GH10519GH10519GH2656GH12155GH20084GH21417)r   r   r   rm   )r   r   rm   r   r   ))rL   r   )rL   r   )rM   r   )r   r   r  )r   r   r   )rm   rm   r   )rL   rM   Nr  )rm   rm   )rQ  r   r  )皙?333333?皙?)rm  ro  rn  )r   r   r   )rm  r+  ro  )aggskew)rp  kurt)rp  prod)rp  r  )jr   r   numpyr   r2  pandas.errorsr   pandasr   r   r   r   r	   r
   pandas._testing_testingr#   pandas.tests.groupbyr   r'   r@   rK   markparametrizer   r   rk   r   r   r   r   r   r   r   r   r  r  r  r  r  r+  r2  r@  rG  rO  rc  rh  rm  r|  r  r  r  r  r  r  r  r  r  r  r  r  r#  r-  r1  todayrB   r9  rG  rR  rh  rO  interval_rangeperiod_ranger   rp  r  r  r  r  r  r  r  r  r  r  r"   r  r  r  r  r  r  r  r  r  r  r#  r,  r5  rY  rX  r  r7  rR  r[  rf  r   r   r   <module>r     s~  
   (    85 %,P#-L 	C#/HI	JIV	CsL9	:QFC	C!:	$	C13aA	BQFK	C+U4FG	H)Tl+77 		
 
Iy)?	CaVL!$	)  <)$=<)$X.,$:  
$#	22,(%,P$ 3F2&0
@@,-9 e}5, 6,B,,,.,,: u60 70 -$.; --4,6,"C894
,   |V\:VaVaV	
 |V\:VaVaV	
..- ';!78, 9,- ^^x~~',,.0@0E0E0GH
,	
,#-L eT]3( 4(,-( 
DK(
!Q
4=>	11 	$1v1v&67	)QFQF+;<	+ff-=>8Aq\Aq!Q<(	

 J!fq!fAA/0	
 -! -- &9;V%WX- Y-,$ 	CD e}50 6	
0 ,,!,H!<,  I&6Ff0U	
 Aj44ff5EcSVZX	
"&#"&-">	,-(,,* )_)EF, G,* 
	

	

	
3%(R,S(R, D62,nd=S8T)UV, W, +t!45- 6- 	*+	,-
-
-9 #qC8c1X(FG- H-;&|-D4r   