
    IR-e@
                     n   d dl Zd dlZd dlmZ d dlmZmZ d dlm	Z	 ej
                            e	 d          d             Zej
                            e	 d          d             Zej
                            e	 d          d	             Zej
                            e	 d          d
             ZdS )    N)convolve_models_fft)Const1DConst2D)	HAS_SCIPYzrequires scipy)reasonc                     t                      } t                      }t          | |dd          }|j        J |j        J  |d          }|                                t          j        dg                                          k    sJ |j        J |j        J |                                 |j        J |j        J d S )N   {Gz?r   g      ?)r   r   _kwargs_convolutionallnparrayclear_cache)m1m2modelresultss       Glib/python3.11/site-packages/astropy/modeling/tests/test_convolution.pytest_clear_cacher      s    	B	BB66E=   %%%eAhhG;;==BHcUOO//111111=$$$)))	=   %%%%%    c                      t                      } t                      }t          | |dd          } |d          }|j        dk    sJ t          j        ddd          } ||          }|j        |j        k    sJ d S )Nr	   r   r   r   r
   r   皙?)r   r   shaper   arange)r   r   r   r   xs        r   test_input_shape_1dr       s    	B	BB66EeAhhG=D    
	"aAeAhhG=AG######r   c                     t                      } t                      }t          | |dd          } |dd          }|j        dk    sJ t          j        ddd          } ||d          }|j        |j        k    sJ  |d|          }|j        |j        k    sJ t          j        ||          } || }|j        |d         j        k    sJ |j        |d         j        k    sJ d S )Nr	   r	   r   r   r   r
   r   r   )r   r   r   r   r   meshgrid)r   r   r   r   r   grids         r   test_input_shape_2dr%   -   s    	B	BB(:DAAEeAqkkG=D    
	"aAeAqkkG=AG####eAqkkG=AG####;q!DeTlG=DGM))))=DGM))))))r   c                     t                      } t                      }t          | |dd          }t          j        ddd          }t          j        ddd          }t          j        ||          }t          j        ||          }t          j        dg          df|                    d          k    sJ t          j         |j        | d	         t          j        |d	         |d         gd
          j	        k              sJ  |j        | d         |d	         j
        k    sJ t          j         |j        | d	         t          j        |d	         |d         gd
          j	        k              sJ  |j        | d         |d	         j
        k    sJ t          j        t          d          5  |                    |d	         |d                    d d d            d S # 1 swxY w Y   d S )Nr"   r   r
   r   r      r   r   )r(   r
   zValues have differing shapes)match)r   r   r   r   r#   r   _convolution_inputsr   reshapeTr   pytestraises
ValueError)r   r   r   r   ygrid0grid1s          r   test__convolution_inputsr3   C   s4   	B	BB(:DAAE
	"aA
	"aAK1EK1E HaSMM4 E$=$=a$@$@@@@@ 6!!5)!,:uQxq*G446	7     %5$e,Q/58>AAAA6!!5)!,:uQxq*G446	7     %5$e,Q/58>AAAA 
z)H	I	I	I 6 6!!%(E!H5556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s   .#GG"%G")numpyr   r-   astropy.convolutionr   astropy.modeling.modelsr   r   "astropy.utils.compat.optional_depsr   markskipifr   r    r%   r3    r   r   <module>r;      s2        3 3 3 3 3 3 4 4 4 4 4 4 4 4 8 8 8 8 8 8 	M*:;;& & <;&$ 	M*:;;$ $ <;$ 	M*:;;* * <;** 	M*:;;6 6 <;6 6 6r   