a
    hK                     @   sH   d Z ddlZddlmZmZ ddlmZmZ dd Z	dd Z
d	d
 ZdS )zTest fast_dict.    N)assert_allcloseassert_array_equal)IntFloatDictargminc                  C   s   t jd} t | jdddt j}| t|}t	||}t
||D ]\}}|| |ksJJ qJt|t|ksxJ |dd |d dksJ t|t|d ksJ tdD ]}||d	 d
 q|d d
ksJ d S )Nr   d   
   )sizex   g      @   i  i  g      @iL  )nprandomZRandomStateuniquerandintZastypeintpZrandlenr   zipappendrange)rngkeysvaluesdkeyvaluei r   ^/var/www/html/assistant/venv/lib/python3.9/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   Zaranger   float64r   r   )r   r   r   r   r   r   test_int_float_dict_argmin   s    
r!   c                  C   sv   t jg dt jd} t jg dt jd}t| |}| \}}|j| jksNJ |j|jks^J t||  t|| d S )N)r
         r   )         )	r   arrayr   r    r   Z	to_arraysr   r   r   )Zkeys_inZ	values_inr   Zkeys_outZ
values_outr   r   r   test_to_arrays#   s    

r(   )__doc__numpyr   Znumpy.testingr   r   Zsklearn.utils._fast_dictr   r   r   r!   r(   r   r   r   r   <module>   s   