
    .i              
          d dl Zd dlZd dlmZ d dlZd dlmZ	 d Z
d Zej                  j                  dee
g      d        Zej                  j                  dg dd	g d
fg ddg dfg dd	ddgfg ddddg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    c/var/www/app/trading-bot/venv/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   funcc                 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         	   left)      r*   right)r)   r*   r*   )r   r)   r*   r-   r.   r)   r*   c                 >   t        j                  g dt         j                        }t        j                  |t        j                  | t         j                        |      }t        j                  |t         j                        }t        j                  ||       y )N)r   r-   r)      r.   r*   )dtype)closed)nparrayint64r   generate_bins_dt64r#   assert_numpy_array_equal)binnerr3   r'   valuesr&   s        r   test_generate_binsr;   (   se    , XX(9F##rxx0F xx1H1r   )numpyr4   pytestpandas._libsr   pandasr   pandas._testing_testingr#   r   r   markparametrizer(   r;    r   r   <module>rE      s        

 

,
,  	
 	

 
FQF#F	
*2+*2r   