
    {KgC	                         d dl Z d dlmZ d dlZd dlmZ d dlmZmZ d dl	m
Z
 d Zej                  j                  dg d      d	        Zd
 Zy)    N)chain)config_context)gen_even_slicesget_chunk_n_rows)assert_array_equalc            	          t        d      } t        t        t        dd      D cg c]  }| |   	 c}       }t	        | |       y c c}w )N
      )rangelistr   r   r   )
some_rangeslicejoined_ranges      e/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/sklearn/utils/tests/test_chunking.pytest_gen_even_slicesr      sH    rJrST?UV?Ue
5 1?UVWXLz<0  Ws   A)	row_bytes
max_n_rowsworking_memoryexpected))   N   r   )r   NgG?  )r   Nr     )r   Nr   r   )r   N   i   )r      r   r   )i   Nr   r   c                    t        j                         5  t        j                  dt               t	        | ||      }d d d        |k(  sJ t        |      t        |      u sJ t        |      5  t        j                         5  t        j                  dt               t	        | |      }d d d        ||k(  sJ t        |      t        |      u sJ 	 d d d        y # 1 sw Y   xY w# 1 sw Y   >xY w# 1 sw Y   y xY w)Nerrorr   r   r   r   r   r   )warningscatch_warningssimplefilterUserWarningr   typer   )r   r   r   r   actuals        r   test_get_chunk_n_rowsr'      s     
	 	 	"g{3!!)
 
# X<4>)))	~	6$$&!!';7%	jQF ' !!!F|tH~--- 
7	6 
#	" '& 
7	6s/   )C1C7(C+.'C7C(+C4	0C77D c                     d} d}d}d}d}t        j                  t        |      5  t        | ||      }ddd       |k(  sJ t	        |      t	        |      u sJ t        |      5  t        j                  t        |      5  t        | |      }ddd       ||k(  sJ t	        |      t	        |      u sJ 	 ddd       y# 1 sw Y   xY w# 1 sw Y   >xY w# 1 sw Y   yxY w)	z<Check that warning is raised when working_memory is too low.i  Nr   zICould not adhere to working_memory config. Currently 1MiB, 2MiB required.)matchr   r   r    )pytestwarnsr$   r   r%   r   )r   r   r   r   warn_msgr&   s         r   test_get_chunk_n_rows_warnsr-   1   s    IJNH 	T  
k	2!!)
 
3 X<4>)))	~	6\\+X6%	jQF 7!!!F|tH~---	 
7	6 
3	2 76 
7	6s/   C(CC'CCC	CC$)r!   	itertoolsr   r*   sklearnr   sklearn.utils._chunkingr   r   sklearn.utils._testingr   r   markparametrizer'   r-        r   <module>r6      sL       " E 51 =..&.r5   