U
    _»{jL  ã                   @   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s”t‚t|ƒt|ƒd ks¬t‚tdƒD ]}| |d	 d
¡ q´|d d
ksÚt‚d S )Nr   éd   é
   )Úsizeéx   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/bwave/python/venv/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   