U
    ½mœd0  ã                   @   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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   ú[/home/sam/Atlas/atlas_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   )Zdtype)r   r   )r	   Zaranger   Zfloat64r   r   r   )r   r   r   r   r   r   Útest_int_float_dict_argmin   s    
r   )Ú__doc__Únumpyr	   Zsklearn.utils._fast_dictr   r   r   r   r   r   r   r   Ú<module>   s   