
    0PhC	                         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                            dg d          d	             Zd
 ZdS )    N)chain)config_context)gen_even_slicesget_chunk_n_rows)assert_array_equalc            	          t          d          t          t          fdt          dd          D                        } t	          |            d S )N
   c                      g | ]
}|         S  r   ).0slice
some_ranges     a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/sklearn/utils/tests/test_chunking.py
<listcomp>z(test_gen_even_slices.<locals>.<listcomp>   s    VVVe
5 1VVV       )rangelistr   r   r   )joined_ranger   s    @r   test_gen_even_slicesr      sV    rJVVVVrST?U?UVVVWXXLz<00000r   )	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   t          j                    5  t          j        dt                     t	          | ||          }d d d            n# 1 swxY w Y   ||k    sJ t          |          t          |          u sJ t          |          5  t          j                    5  t          j        dt                     t	          | |          }d d d            n# 1 swxY w Y   ||k    sJ t          |          t          |          u sJ 	 d d d            d S # 1 swxY w Y   d S )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33!!)
 
 

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 X<<4>>))))	~	6	6	6 . .$&& 	R 	R!';777%	jQQQF	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R !!!!F||tH~~-----. . . . . . . . . . . . . . . . . .sG   -AAAD$,CDC 	 D#C 	$+DD!$D!c                      d} d}d}d}d}t          j        t          |          5  t          | ||          }ddd           n# 1 swxY w Y   ||k    sJ t	          |          t	          |          u sJ t          |          5  t          j        t          |          5  t          | |          }ddd           n# 1 swxY w Y   ||k    sJ t	          |          t	          |          u sJ 	 ddd           dS # 1 swxY w Y   dS )	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_warnsr2   1   s   IJNH 	T  
k	2	2	2 
 
!!)
 
 

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 X<<4>>))))	~	6	6	6 . .\+X666 	R 	R%	jQQQF	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R 	R!!!!F||tH~~-----	. . . . . . . . . . . . . . . . . .sG   AA	A	D$C6DC	D	C	
+DD
D)r&   	itertoolsr   r/   sklearnr   sklearn.utils._chunkingr   r   sklearn.utils._testingr   r   markparametrizer,   r2   r   r   r   <module>r9      s            " " " " " " E E E E E E E E 5 5 5 5 5 51 1 1 =   . . .&. . . . .r   