
    Owg                        d dl Zd dlZd dlmZ d dlmc mZ d dl	Z
d dlmZ d Zd Zej                   j#                  de ej$                  eej&                        g      d        Zej                   j#                  d ej*                  g d	ej,                  
      d ej*                  g dej,                  
      f ej*                  g d	ej,                  
      d ej*                  g dej,                  
      f ej*                  g dej,                  
      d ej*                  ddgej,                  
      f ej*                  g dej,                  
      d ej*                  ddgej,                  
      fg      d        Zy)    N)libc                 x    t        |       t        | j                  j                  d   j                        k(  sJ yNr   )len_mgrblocksmgr_locsxs    \/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/groupby/test_bin_groupby.pyassert_block_lengthsr      s/    q6Sq)223333    c                 @    | j                         j                          yr   )cumsummaxr
   s    r   
cumsum_maxr      s    HHJNNr   func)marksc                 4   t        j                  g dg dg dd      }|j                  ddg      j                  |       }t        j                  dddgit         j                  j                  d	gd	d
ggddg            }t        j                  ||       y )N)ar   r   )r   br   )   r   r   )ABCr   r   r   r   r   r   )names)index)pd	DataFramegroupbyagg
MultiIndexfrom_producttmassert_frame_equal)r   dfresultexpecteds       r   test_mgr_locs_updatedr)      s     
O/	R	SBZZc
#''-F||	q!fmm((3%#s)<S#J(OH &(+r   zbinner,closed,expected)r         	   dtypeleft)      r+   right)r*   r+   r+   )r   r*   r+   r0   r1   r*   r+   c                     t        j                  g dt         j                        }t        j                  || |      }t        j                  ||       y )N)r   r0   r*      r1   r+   r-   )closed)nparrayint64r   generate_bins_dt64r$   assert_numpy_array_equal)binnerr5   r(   valuesr'   s        r   test_generate_binsr=   )   s=    , XX(9F##FF6BF1r   )numpyr6   pytestpandas._libsr   pandas.util._test_decoratorsutil_test_decoratorstdpandasr   pandas._testing_testingr$   r   r   markparametrizeparamskip_array_manager_invalid_testr)   r7   r8   r=    r   r   <module>rM      s^      ) )  

 
)1S1ST
,
,  BHH\2BHHYbhh/	
 BHH\2BHHYbhh/	

 
)288	,fhbhh1vRXX6VWBHHYbhh/BHHaV288,	
*2+*2r   