ó
‡ˆ\c           @   s^   d  Z  d d l Z d d l m Z m Z d d l m Z d d l m	 Z	 d „  Z
 d „  Z d S(   s    Test fast_dict.
iÿÿÿÿN(   t   IntFloatDictt   argmin(   t   assert_equal(   t   xrangec          C   s(  t  j j d ƒ }  t  j |  j d d d ƒj t  j ƒ ƒ } |  j t | ƒ ƒ } t	 | | ƒ } x. t
 | | ƒ D] \ } } t | | | ƒ qp Wt t | ƒ t | ƒ ƒ | j d d ƒ t | d d ƒ t t | ƒ t | ƒ d ƒ x( t d ƒ D] } | j | d	 d
 ƒ qõ Wt | d d
 ƒ d  S(   Ni    id   t   sizei
   ix   g      @i   iÐ  iè  g      @iL  (   t   npt   randomt   RandomStatet   uniquet   randintt   astypet   intpt   randt   lenR    t   zipR   t   appendR   (   t   rngt   keyst   valuest   dt   keyt   valuet   i(    (    sA   lib/python2.7/site-packages/sklearn/utils/tests/test_fast_dict.pyt   test_int_float_dict
   s    *c          C   sV   t  j d d t  j ƒ}  t  j d d t  j ƒ} t |  | ƒ } t t | ƒ d ƒ d  S(   Nid   t   dtypei    (   i    i    (   R   t   arangeR   t   float64R    R   R   (   R   R   R   (    (    sA   lib/python2.7/site-packages/sklearn/utils/tests/test_fast_dict.pyt   test_int_float_dict_argmin   s    (   t   __doc__t   numpyR   t   sklearn.utils.fast_dictR    R   t   sklearn.utils.testingR   t   sklearn.externals.six.movesR   R   R   (    (    (    sA   lib/python2.7/site-packages/sklearn/utils/tests/test_fast_dict.pyt   <module>   s   	