o
    tBh0                     @   s0   d Z ddlZddlmZmZ dd Zdd ZdS )z Test fast_dict.
    N)IntFloatDictargminc                  C   s   t jd} t | jdddt j}| t|}t	||}t
||D ]\}}|| |ks1J q%t|t|ks<J |dd |d dksJJ t|t|d ksVJ tdD ]
}||d	 d
 qZ|d d
ksmJ d S )Nr   d   
   )sizex   g      @   i  i  g      @iL  )nprandomRandomStateuniquerandintastypeintprandlenr   zipappendrange)rngkeysvaluesdkeyvaluei r   y/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/utils/tests/test_fast_dict.pytest_int_float_dict   s   
r   c                  C   s>   t jdt jd} t jdt jd}t| |}t|dksJ d S )Nr   )dtype)r   r   )r	   aranger   float64r   r   )r   r   r   r   r   r   test_int_float_dict_argmin   s   
r"   )__doc__numpyr	   sklearn.utils._fast_dictr   r   r   r"   r   r   r   r   <module>   s
    