
    HR-e,                         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                            dd d g          d	             ZdS )
    N)convolveconvolve_fft)Gaussian2DKernel)NDDatac                     t          j        d          } d| d<   t          |           }t          d          }t	          ||          }t           j        d dd df         \  }}|d         t          j        d|dz
  dz  |dz
  dz  z   z            z  }t           j                            ||d	           t          ||          }t           j                            ||d	           d S )
N   r	         r   r	   g      r      gư>)atol)
npzerosr   r   r   mgridexptestingassert_allcloser   )arrnddtest_kernelresultxyexpectedresultfs           Nlib/python3.11/site-packages/astropy/convolution/tests/test_convolve_nddata.pytest_basic_nddatar      s    
(8

CCI
++C"1%%Kc;''F8CRC"HDAqd|bfTa!e\QUqL-H%IJJJHJvxd;;;3,,GJwt<<<<<    convfuncc                      t          | dddS NinterpolateT)nan_treatmentnormalize_kernel)r   argss    r   <lambda>r(      s    h
 
 
 r   c                      t          | dddS r"   )r   r&   s    r   r(   r(   "   s    l
 
 
 r   c                    t          j        d          }dx|d<   x|d<   x|d<   |d<   d|d<   t          |          }|d	k     }d
|d<   t          ||          }|                                }t           j        |d<   t          |          }t          d          } | ||          } | ||          }	 | ||          }
t          j        |	|
          sJ t          j        ||
          rJ t          j        ||	          rJ t          j        t          j        |j	                            t          j        t          j        |j	                            k    sJ d S )Nr   g?)   r   )   r   )r   r+   )r   r,   g      ?r   r   T)maskr
   )
r   r   r   copynanr   allclosesumisnandata)r    r   ndd_baser-   ndd_maskarrnanndd_nanr   result_base
result_nanresult_masks              r   test_masked_nddatar;      s_    (8

C477CI7D	7CID	CIc{{H7DDJc%%%HXXZZF6F4LVnnG"1%%K(8[11K';//J(8[11K;z;/////{;44444{;
33333 6"(8=))**bfRXgl5K5K.L.LLLLLLLr   )numpyr   pytestastropy.convolution.convolver   r   astropy.convolution.kernelsr   astropy.nddatar   r   markparametrizer;    r   r   <module>rD      s         ? ? ? ? ? ? ? ? 8 8 8 8 8 8 ! ! ! ! ! != = =" 	
 	
	
 	
	
 
M M
 
M M Mr   