ó
î&]\c           @   sR   d  d l  m Z d  d l m Z d  d l Z d  d l m Z d „  Z	 d „  Z
 d S(   iÿÿÿÿ(   t   assert_array_equal(   t   raisesN(   t   linear_sum_assignmentc       	   C   s¼  xµd d d g d d d g d d d g g d d d g f d d d d g d d d d g d d d d	 g g d d d g f d
 d
 d g d d d g d d d g g d
 d d g f d
 d
 d d g d d d d g d d d d
 g g d
 d d g f g  g  g g  f g D]´ \ }  } t  j |  ƒ }  t |  ƒ \ } } t | t  j | ƒ ƒ t | |  | | f ƒ |  j }  t |  ƒ \ } } t | t  j | ƒ ƒ t t  j | ƒ t  j |  | | f ƒ ƒ q Wd  S(   Ni  i–   iÂ  iX  i,  iá   i   i   i   i
   i   i	   i   i   i   (   t   npt   arrayR   R    t   sortt   T(   t   cost_matrixt   expected_costt   row_indt   col_ind(    (    sB   lib/python2.7/site-packages/scipy/optimize/tests/test_hungarian.pyt   test_linear_sum_assignment   s8    		c          C   s  t  t t d d d g ƒ d d d g d d d g g }  t t |  ƒ t t j |  ƒ ƒ ƒ t t |  ƒ t t j |  ƒ ƒ ƒ t j d ƒ } t t | j t j	 ƒ ƒ t | ƒ ƒ t  t t | j t
 ƒ ƒ t j | d d <t  t t | ƒ t j d ƒ } t j | d d <t  t t | ƒ d  S(   Ni   i   i   i   i   i   i    (   t   assert_raisest
   ValueErrorR   R    R   t   asarrayt   matrixt   identityt   astypet   boolt   strt   nant   inf(   t   Ct   I(    (    sB   lib/python2.7/site-packages/scipy/optimize/tests/test_hungarian.pyt+   test_linear_sum_assignment_input_validation7   s    (   t   numpy.testingR    t   pytestR   R   t   numpyR   t   scipy.optimizeR   R   R   (    (    (    sB   lib/python2.7/site-packages/scipy/optimize/tests/test_hungarian.pyt   <module>   s
   	+