
\c           @  sq   d  d l  m Z m Z d  d l Z d  d l m Z d  d l m Z d  d l	 m
 Z
 d   Z d   Z d   Z d S(	   i(   t   print_functiont   divisionN(   t   inpaint(   t   testing(   t   assert_allclosec       
   C  s  t  j t  j t  j d d d   d  }  t  j |   } d | d d d   f <d | d d d   f <d | d d d   f <d |  t  j |  <t j |  |  } t  j d d d	 d
 d g d d d	 d d g d d d d d g d d d	 d
 d g d d d	 d
 d g g  } t	 | |  d  S(   Ni    i   i   i   i   i   g        g      ?g      ?g      ?g?g2?g]ߝ?g}?g	#m?g~4?g      ?(   i   i   (
   t   npt   tilet   squaret   linspacet
   zeros_liket   whereR   t   inpaint_biharmonict   arrayR   (   t   imgt   maskt   outt   ref(    (    sE   lib/python2.7/site-packages/skimage/restoration/tests/test_inpaint.pyt   test_inpaint_biharmonic_2d
   s    *c          C  s  t  j t  j t  j d d d   d  }  t  j |  |  j f  }  t  j |   } d | d d d   d  d   f <d | d d d   d  d   f <d | d d d   d  d   f <d |  t  j |  <t j	 |  |  } t  j t  j
 d d d	 d
 d g d d d	 d d g d d d d d g d d d	 d
 d g d d d	 d
 d g g  t  j
 d d d d d g d d d d d g d	 d	 d d d g d
 d
 d
 d
 d
 g d d d d d g g  f  } t | |  d  S(   Ni    i   i   i   i   i   g        g      ?g      ?g      ?gVm3?g>q?g?k#C34?gO4S?g6[*g?gjB?g      ?ggp?g/H<\?g"
'J?g cy6o?g)?gA'G^?(   i   i   (   R   R   R   R   t   dstackt   TR	   R
   R   R   R   R   (   R   R   R   R   (    (    sE   lib/python2.7/site-packages/skimage/restoration/tests/test_inpaint.pyt   test_inpaint_biharmonic_3d   s,    *!c       	   C  s   t  j g   t  j g   }  } t j t   t j |  |  Wd  QXt  j d  t  j d  }  } t j t   t j |  |  Wd  QXt  j j t  j d  d d d g d d g g }  t  j d	  } t j t	   t j |  |  Wd  QXd  S(
   Ni   i   i   R   i    (   i   i   (   i   i   (   i   i   (   i   i   (
   R   t   zerosR   t   raisest
   ValueErrorR   R   t   maR   t	   TypeError(   R   R   (    (    sE   lib/python2.7/site-packages/skimage/restoration/tests/test_inpaint.pyt   test_invalid_input6   s    3(   t
   __future__R    R   t   numpyR   t   skimage.restorationR   t   skimage._sharedR   t   skimage._shared.testingR   R   R   R   (    (    (    sE   lib/python2.7/site-packages/skimage/restoration/tests/test_inpaint.pyt   <module>   s   		