
    I j1                     >    S SK rS SKrS SKJr  S rS rS r	S r
S rg)    Nc                     [         R                  " / SQ/ SQ/ SQS.5      n U R                  SS9n[         R                  " SS/0[         R                  " S	/5      S
9n[        R
                  " X5        U R                  SSS9n[         R                  " SSS/0[         R                  " S	S/5      S
9n[        R
                  " X5        g )N      r   r   r   r         r   r	         ABC)r   sum)foor   
   r   index)r   min)r   barr   r   )pd	DataFrameaggIndextmassert_frame_equaldfresultexpecteds      /root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/apply/test_frame_apply_relabeling.pytest_agg_relabelr#      s    	L|,O	PB VVV%F||S2$Krxx/@AH&+ VV,V7F||S2q'N"((E5>2JKH&+    c            
      j   [         R                  " / SQ/ SQ/ SQS.5      n U R                  SSSSS	S
S9n[         R                  " S[        R                  S[        R                  S[        R                  /[        R                  S[        R                  S[        R                  [        R                  /[        R                  [        R                  [        R                  [        R                  [        R                  S/S.[         R
                  " / SQ5      S9n[        R                  " X5        g )Nr   r   r
   r   )r   r   )r   mean)r   r   )r   max)r   r'   r   r   )r   r   catdatfgg      @      ?       @g      @      @      @r   )r   r   r   npnanr   r   r   r   s      r"   ,test_agg_relabel_multi_columns_multi_methodsr3      s    	L|,O	PBVV

  F ||rvvsBFFC8&&#rvvsBFFBFF;&&"&&"&&"&&"&&#>	

 hh=>H &+r$   c            
         [         R                  " / SQ/ SQ/ SQS.5      n U R                  S[        R                  4SS[
        4S9n[         R                  " S/ SQ0[         R                  " / SQ5      S	9n[        R                  " X5        U R                  S[
        4S
[        R
                  4S
[        4SS
[        R                  4S
S 4S9n[         R                  " S[        R                  [        R                  [        R                  [        R                  [        R                  /[        R                  SS[        R                  SS/[        R                  [        R                  [        R                  S[        R                  [        R                  /S.[         R                  " / SQ5      S	9n[        R                  " X5        g )Nr   r   r
   r   r   )r   r&   r   r   r)   )      ?r6   r-   r   r   r(   c                     [        U 5      $ )N)r   )xs    r"   <lambda>4test_agg_relabel_partial_functions.<locals>.<lambda><   s    3q6r$   )r   r   r)   r*   r+   kkr-   r/   g      $@r0   )r   r   r   r1   r&   r   r   r   r   r'   r   r2   r   s      r"   "test_agg_relabel_partial_functionsr<   -   s@   	L|,O	PBVVbggMSzVJF||	obhh/D&EH &+VV#J"&&M#J-!"  F ||rvvrvvrvvrvvrvv>&&#sBFFD#6&&"&&"&&#rvvrvv>	

 hh>?H &+r$   c            
         [         R                  " SS/SS/S.5      n U R                  [         R                  " SS5      [         R                  " SS5      [         R                  " SSS	9[         R                  " SS
S9S9n[         R                  " S/ SQ0[         R                  " / SQ5      S9n[
        R                  " X5        U R                  [         R                  " SS5      [         R                  " SS
S	9[         R                  " SS
S	9S9n[         R                  " S[        R                  S/[        R                  S[        R                  /S.[         R                  " / SQ5      S9n[
        R                  " X5        g )Nr   r   r   )r   r   r   r   r   count)columnaggfuncr'   )r@   )r   r   r)   fft)r   r   r   r   r   r   r5   g        r-   r.   )	r   r   r   NamedAggr   r   r   r1   r2   r   s      r"   test_agg_namedtuplerC   I   s(   	QF!Q0	1BVVKKU#KKU#KKsG4KKU+	  F ||	l288,H#IH &+VVKKU#KKsE2KKsE2  F
 ||BFFC RVV'<=hh,-H &+r$   c                      [         R                  R                  R                  S5      n Sn[        R
                  " X5        g )Nr   )Fr   NN)r   coreapplyreconstruct_funcr   assert_equal)r    r!   s     r"   test_reconstruct_funcrI   d   s-     WW]]++E2F)HOOF%r$   )numpyr1   pandasr   pandas._testing_testingr   r#   r3   r<   rC   rI    r$   r"   <module>rO      s'      , ,,,8,6&r$   