
    {KgK                     B    d Z ddlZddlmZmZ ddlmZmZ d Z	d Z
d Zy)zTest fast_dict.    N)assert_allcloseassert_array_equal)IntFloatDictargminc                  P   t         j                  j                  d      } t        j                  | j	                  dd      j                  t         j                              }| j                  t        |            }t        ||      }t        ||      D ]  \  }}||   |k(  rJ  t        |      t        |      k(  sJ |j                  dd       |d   dk(  sJ t        |      t        |      dz   k(  sJ t        d      D ]  }|j                  |d	z   d
        |d   d
k(  sJ y )Nr   d   
   )sizex   g      @   i  i  g      @iL  )nprandomRandomStateuniquerandintastypeintprandlenr   zipappendrange)rngkeysvaluesdkeyvalueis          f/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/sklearn/utils/tests/test_fast_dict.pytest_int_float_dictr!   	   s   
))


"C99S[[2[.55bgg>?DXXc$i FT6"A$'
Uv (q6SYHHS#S6S==q6SY]"""4[	T3 T7c>>    c                      t        j                  dt         j                        } t        j                  dt         j                        }t	        | |      }t        |      dk(  sJ y )Nr   dtype)r   r   )r   aranger   float64r   r   )r   r   r   s      r    test_int_float_dict_argminr(      sH    99S(DYYs"**-FT6"A!9r"   c                  z   t        j                  g dt         j                        } t        j                  g dt         j                        }t	        | |      }|j                         \  }}|j                  | j                  k(  sJ |j                  |j                  k(  sJ t        ||        t        ||       y )N)r         r$   )         )	r   arrayr   r'   r   	to_arraysr%   r   r   )keys_in	values_inr   keys_out
values_outs        r    test_to_arraysr5   #   s     hhy0G"**5IWi(A;;=Hj>>W]]***??j.....x)J	*r"   )__doc__numpyr   numpy.testingr   r   sklearn.utils._fast_dictr   r   r!   r(   r5    r"   r    <module>r;      s       = 9$+r"   