
    i1                     >    d dl Zd dlZd dlmZ d Zd Zd Z	d Z
d Zy)    Nc                     t        j                  g dg dg dd      } | j                  d      }t        j                  ddgit        j                  d	g      
      }t	        j
                  ||       | j                  dd      }t        j                  dddgit        j                  d	dg      
      }t	        j
                  ||       y )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      k/app/cer_product_mecsu/.venv/lib/python3.12/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            
      p   t        j                  g dg dg dd      } | j                  ddddd	d
      }t        j                  dt        j                  dt        j                  dt        j                  gt        j                  dt        j                  dt        j                  t        j                  gt        j                  t        j                  t        j                  t        j                  t        j                  dgdt        j
                  g d            }t        j                  ||       y )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            
         t        j                  g dg dg dd      } | j                  dt        j                  fddt
        f      }t        j                  dg dit        j                  g d      	      }t        j                  ||       | j                  dt
        fd
t        j
                  fd
t        fdd
t        j                  fd
d f      }t        j                  dt        j                  t        j                  t        j                  t        j                  t        j                  gt        j                  ddt        j                  ddgt        j                  t        j                  t        j                  dt        j                  t        j                  gdt        j                  g d      	      }t        j                  ||       y )Nr   r   r
   r   r   )r   r&   r   r   r)   )      ?r6   r-   r   r   r(   c                     t        |       S )N)r   )xs    r"   <lambda>z4test_agg_relabel_partial_functions.<locals>.<lambda><   s
    3q6 r$   )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;   -   sD   	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            	      $   t        j                  ddgddgd      } | j                  t        j                  dd      t        j                  dd      t        j                  dd	      t        j                  dd
            }t        j                  dg dit        j                  g d            }t        j                  ||       | j                  t        j                  dd      t        j                  dd
	      t        j                  dd
	            }t        j                  dt        j                  dgt        j                  dt        j                  gdt        j                  g d            }t        j                  ||       y )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_namedtuplerB   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                      t         j                  j                  j                  d      } d}t	        j
                  | |       y )Nr   )Fr   NN)r   coreapplyreconstruct_funcr   assert_equal)r    r!   s     r"   test_reconstruct_funcrH   d   s/     WW]]++E2F)HOOFH%r$   )numpyr1   pandasr   pandas._testing_testingr   r#   r3   r;   rB   rH    r$   r"   <module>rN      s'      , ,,,8,6&r$   