o
    iK                     @   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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randomZRandomStateuniquerandintZastypeintpZrandlenr   zipappendrange)rngkeysvaluesdkeyvaluei r   \/home/kim/smarthome/.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   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s'J |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    