
    -iK                     B    S r SSKrSSKJrJr  SSKJrJr  S r	S r
S rg)zTest fast_dict.    N)assert_allcloseassert_array_equal)IntFloatDictargminc                  V   [         R                  R                  S5      n [         R                  " U R	                  SSS9R                  [         R                  5      5      nU R                  [        U5      5      n[        X5      n[        X5       H  u  pEX4   U:X  a  M   e   [        U5      [        U5      :X  d   eUR                  SS5        US   S:X  d   e[        U5      [        U5      S-   :X  d   e[        S5       H  nUR                  US	-   S
5        M     US   S
:X  d   eg )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          U/var/www/html/venv/lib/python3.13/site-packages/sklearn/utils/tests/test_fast_dict.pytest_int_float_dictr!   	   s    
))


"C99S[[2[.55bgg>?DXXc$i FT"A$'
v (q6SYHHS#S6S==q6SY]"""4[	T3 T7c>>    c                      [         R                  " S[         R                  S9n [         R                  " S[         R                  S9n[	        X5      n[        U5      S:X  d   eg )Nr   dtype)r   r   )r   aranger   float64r   r   )r   r   r   s      r    test_int_float_dict_argminr(      sF    99S(DYYs"**-FT"A!9r"   c                  n   [         R                  " / SQ[         R                  S9n [         R                  " / SQ[         R                  S9n[	        X5      nUR                  5       u  p4UR                  U R                  :X  d   eUR                  UR                  :X  d   e[        X05        [        XA5        g )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W(A;;=H>>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"   