
    0PhK                     D    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dS )zTest fast_dict.    N)assert_allcloseassert_array_equal)IntFloatDictargminc                     t           j                            d          } t          j        |                     dd                              t           j                            }|                     t          |                    }t          ||          }t          ||          D ]\  }}||         |k    sJ t          |          t          |          k    sJ |                    dd           |d         dk    sJ t          |          t          |          dz   k    sJ t          d          D ]}|                    |d	z   d
           |d         d
k    sJ d S )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          b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/sklearn/utils/tests/test_fast_dict.pytest_int_float_dictr!   	   sD   
)


"
"C9S[[2[..55bg>>??DXXc$ii  FT6""A$''  
Uvq66SYYHHS#S6S====q66SYY]""""4[[    	T3T7c>>>>>>    c                      t          j        dt           j                  } t          j        dt           j                  }t	          | |          }t          |          dk    sJ d S )Nr   dtype)r   r   )r   aranger   float64r   r   )r   r   r   s      r    test_int_float_dict_argminr(      sZ    9S(((DYs"*---FT6""A!99r"   c                  d   t          j        g dt           j                  } t          j        g dt           j                  }t	          | |          }|                                \  }}|j        | j        k    sJ |j        |j        k    sJ t          ||            t          ||           d S )N)r         r$   )         )	r   arrayr   r'   r   	to_arraysr%   r   r   )keys_in	values_inr   keys_out
values_outs        r    test_to_arraysr5   #   s     hyyy000G"*55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 9 9 9 9 9 9 9  $  + + + + +r"   