
    bMh                        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                            de ej        eej                  g          d             Ze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dS )    N)libc                 t    t          |           t          | j        j        d         j                  k    sJ dS Nr   )len_mgrblocksmgr_locsxs    e/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/groupby/test_bin_groupby.pyassert_block_lengthsr      s2    q66Sq)23333331    c                 R    |                                                                   dS r   )cumsummaxr
   s    r   
cumsum_maxr      s    HHJJNN1r   func)marksc                 @   t          j        g dg dg dd          }|                    ddg                              |           }t          j        dddgit           j                            d	gd	d
ggddg                    }t          j        ||           d S )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     
OOO///			RR	S	SBZZc
##''--F|	q!fm((3%#s)<S#J(O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        ||           d S )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=   )   sQ    , X(((999F#FF6BBBF1111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
              ) ) ) ) ) ) ) ) )            
  
 
)1STTT 
, 
, 
,  BH\\\222BHYYYbh///	
 BH\\\222BHYYYbh///	

 
)))28	,	,	,fhbh1vRX6V6V6VWBHYYYbh///BHaV28,,,	
 *2 2+ *2 2 2r   