σ
ίΘ[c           @` s§   d  d l  m Z m Z m Z m Z d  d l Z d  d l Z d d l m Z m	 Z	 d d l
 m Z d d l m Z d   Z e j j d	 d
   d   g  d    Z d S(   i    (   t   absolute_importt   divisiont   print_functiont   unicode_literalsNi   (   t   convolvet   convolve_fft(   t   Gaussian2DKerneli   (   t   NDDatac          C` sΩ   t  j d  }  d |  d	 <t |   } t d  } t | |  } t  j d  d  d  d  f \ } } | d
 t  j d | d d | d d  } t  j j | | d d t	 | |  } t  j j | | d d d  S(   Ni   i   i   g      ΰΏi   t   atolgν΅ χΖ°>(   i   i   (   i   i   (   i   i   (
   t   npt   zerosR   R   R   t   mgridt   expt   testingt   assert_allcloseR   (   t   arrt   nddt   test_kernelt   resultt   xt   yt   expectedt   resultf(    (    sM   lib/python2.7/site-packages/astropy/convolution/tests/test_convolve_nddata.pyt   test_basic_nddata   s    
%/u   convfuncc          G` s   t  d d d t |   S(   Nt   nan_treatmentu   interpolatet   normalize_kernel(   R   t   True(   t   args(    (    sM   lib/python2.7/site-packages/astropy/convolution/tests/test_convolve_nddata.pyt   <lambda>   s    c          G` s   t  d d d t |   S(   NR   u   interpolateR   (   R   R   (   R   (    (    sM   lib/python2.7/site-packages/astropy/convolution/tests/test_convolve_nddata.pyR       s    c         C` sW  t  j d
  } d | d <| d <| d <| d <d | d <t |  } | d k  } t | d <t | d | } | j   } t  j | d <t |  } t d	  } |  | |  } |  | |  }	 |  | |  }
 t  j |	 |
  sε t  t  j | |
  sώ t  t  j | |	  st  t  j	 t  j
 | j   t  j	 t  j
 | j   k sSt  d  S(   Ni   gΙ?i   i   i   g      ψ?i    t   maski   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   R	   R
   R   R   t   copyt   nanR   t   allcloset   AssertionErrort   sumt   isnant   data(   t   convfuncR   t   ndd_baseR   t   ndd_maskt   arrnant   ndd_nanR   t   result_baset
   result_nant   result_mask(    (    sM   lib/python2.7/site-packages/astropy/convolution/tests/test_convolve_nddata.pyt   test_masked_nddata   s$    "

(   t
   __future__R    R   R   R   t   pytestt   numpyR	   R   R   t   kernelsR   t   nddataR   R   t   markt   parametrizeR-   (    (    (    sM   lib/python2.7/site-packages/astropy/convolution/tests/test_convolve_nddata.pyt   <module>   s   "	