U
    9%eL                     @   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sJtqJt|t|ksxt|dd |d dkstt|t|d ksttdD ]}||d	 d
 q|d d
kstd S )Nr   d   
   )sizex   g      @   i  i  g      @iL  )nprandomZRandomStateuniquerandintZastypeintpZrandlenr   zipAssertionErrorappendrange)rngkeysvaluesdkeyvaluei r   a/var/www/html/Darija-Ai-API/env/lib/python3.8/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:td S )Nr   dtype)r   r   )r   Zaranger   float64r   r   r   )r   r   r   r   r   r   test_int_float_dict_argmin   s    
r"   c                  C   sz   t jdddgt jd} t jdddgt jd}t| |}| \}}|j| jksRt|j|jksbtt||  t	|| d S )Nr
         r            )
r   arrayr   r!   r   Z	to_arraysr    r   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   