ó
Ê½÷Xc           @   sH   d  d l  m Z m Z d  d l Z d  d l m Z d „  Z d „  Z d S(   iÿÿÿÿ(   t   assert_array_equalt   assert_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   /tmp/pip-build-7oUkmx/scipy/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 |  ƒ ƒ ƒ d  S(   Ni   i   i   i   i   i   (   R   t
   ValueErrorR   R    R   t   asarrayt   matrix(   t   C(    (    sB   /tmp/pip-build-7oUkmx/scipy/scipy/optimize/tests/test_hungarian.pyt+   test_linear_sum_assignment_input_validation6   s    (	   t   numpy.testingR    R   t   numpyR   t   scipy.optimizeR   R   R   (    (    (    sB   /tmp/pip-build-7oUkmx/scipy/scipy/optimize/tests/test_hungarian.pyt   <module>   s   	+