
    -iC	                         S SK r S SKJr  S SKrS SKJr  S SKJrJr  S SK	J
r
  S r\R                  R                  S/ SQ5      S	 5       rS
 rg)    N)chain)config_context)gen_even_slicesget_chunk_n_rows)assert_array_equalc            	          [        S5      n [        [        [        SS5       Vs/ s H  oU   PM	     sn6 5      n[	        X5        g s  snf )N
      )rangelistr   r   r   )
some_rangeslicejoined_ranges      T/var/www/html/venv/lib/python3.13/site-packages/sklearn/utils/tests/test_chunking.pytest_gen_even_slicesr      sD    rJrST?UV?Ue5 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                    [         R                  " 5          [         R                  " S[        5        [	        U UUS9nS S S 5        WU:X  d   e[        U5      [        U5      L d   e[        US9   [         R                  " 5          [         R                  " S[        5        [	        XS9nS S S 5        XC:X  d   e[        U5      [        U5      L d   e S S S 5        g ! , (       d  f       N= f! , (       d  f       NI= f! , (       d  f       g = f)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%	QF ' !!!F|tH~--- 
7	6 
#	" '& 
7	6s/   'C/C>%C-*(C>
C*-
C;	7C>>
Dc                     Sn SnSnSnSn[         R                  " [        US9   [        U UUS9nSSS5        WU:X  d   e[	        U5      [	        U5      L d   e[        US9   [         R                  " [        US9   [        XS9nSSS5        XS:X  d   e[	        U5      [	        U5      L d   e SSS5        g! , (       d  f       N= f! , (       d  f       NI= f! , (       d  f       g= f)	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%	QF 7!!!F|tH~---	 
7	6 
3	2 76 
7	6s/   B8"C<
C	(C8
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   