
\c           @   s)   d  d l  Z d  d l m Z d   Z d S(   iN(   t
   _hungarianc       	   C   s  d d d g d d d g d d d g g d f d d d d g d d d d	 g d d d d
 g g d f d d d g d d d g d d d g g d f d d d d g d d d d g d d d d g g d f g  g  g d f g }  x |  D] \ } } t  j |  } t |  } d } x. | D]& \ } } | | | f } | | 7} qW| | k sRt  t | j  } d } x. | D]& \ } } | | | f } | | 7} qnW| | k s t  q Wd  S(   Ni  i   i  iX  i,  i   iR  i   i   i   i  i
   i   i	   i   i   i   i   i   i    (   t   npt   arrayR    t   AssertionErrort   T(   t   matricest   cost_matrixt   expected_totalt   indexest
   total_costt   rt   ct   x(    (    sI   lib/python2.7/site-packages/sklearn/utils/tests/test_linear_assignment.pyt   test_hungarian
   s@    	(   t   numpyR   t    sklearn.utils.linear_assignment_R    R   (    (    (    sI   lib/python2.7/site-packages/sklearn/utils/tests/test_linear_assignment.pyt   <module>   s   