
    Q3j                     z    d Z ddlZddlZddlZddlmZ d Zej                  j                  dg d      d        Z
y)z;
Testing for the utility function _get_n_samples_bootstrap
    N)_get_n_samples_bootstrapc                     d\  } }}t        | ||      | k(  sJ d\  } }}t        | ||      |k(  sJ d}d\  } }}t        j                  t        |      5  t        | ||      t	        || z        k(  sJ 	 d d d        d\  } }}t        j                  t        |      5  t        | ||      dk(  sJ 	 d d d        d}t
        j                  j                  d	      }d
d|j                  d
      }}} t        j                  t        |      5  t        | ||      t	        ||j                         z        k(  sJ 	 d d d        t        j                         5  t        j                  d       d\  } }}t        | ||      |k(  sJ dd|j                  d      }}} t        | ||      t	        ||j                         z        k(  sJ 	 d d d        y # 1 sw Y   ]xY w# 1 sw Y   -xY w# 1 sw Y   xY w# 1 sw Y   y xY w)N)
   Nnot_used)r      r   z?.+the number of samples.+low number.+max_samples.+as an integer)r   gQ?N)match)r   g{Gz?N   zI.+the total sum of sample weights.+low number.+max_samples.+as an integerr   r   g?)sizeerror)d      Nr   g      ?)r   pytestwarnsUserWarningintnprandomdefault_rnguniformsumwarningscatch_warningssimplefilter)	n_samplesmax_samplessample_weightwarning_msgwarning_msg_with_weightsrngs         R/DATA/.local/lib/python3.12/site-packages/sklearn/ensemble/tests/test_bootstrap.pytest_get_n_samples_bootstrapr!      s   ,@)I{M#I{MJiWWW ->)I{M KG;VV TK,:)I{M	k	5'	;NRU)#S
 
 	
 
 
6
 -;)I{M	k	5'	;NRSSSS 
6 	T  ))


"C,.S[[b[5IM{I	k)A	B'	;NRU-++--S
 
 	
 
 
C 
	 	 	"g&0=-	;$Y]K	
 14S#++3+:O;	'	;NRU-++--S
 
 	
 
 
#	") 
6	5 
6	5 
C	B 
#	"s1   GG-G)A0G5GG&)G25G>r   )Nr	   r   i  g?g      ?g      ?c                     d}t         j                  j                  d      }|j                  dd|      }|j	                         }t        || |      }t        || d       }| ||k7  sJ y ||k(  sJ y )Nr   r      r   )r   r   RandomStaterandintr   r   )r   r   r   r   n_samples_repeatedn_bootstrap_weightedn_bootstrap_repeateds          r    6test_n_samples_bootstrap_repeated_weighted_equivalencer)   ?   s     I
))


"CKK1i0M&**,3; 4K #';;;;#';;;;    )__doc__r   numpyr   r   sklearn.ensemble._bootstrapr   r!   markparametrizer)    r*   r    <module>r1      sE       @,
^ (IJ< K<r*   