3
Od                 @   s  d dl Zd dlZd dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
 d dlZd Zejjejjejjejjejjejjejjejjejjejjejjejjejjejejejjejejejej ej!ej"ej#ej$ej%ej&fZ'ejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejj ejj!ejj"ejj#ejj$ejj%ejj&fZ(ejejejejejejejejejejejejejejejejejejejej ej!ej"ej#ej$ej%ej&fZ)ej*j+de,e(e'e)dd Z-ejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjejjfZ.ejejejejejejejejejejejejejejejejejejfZ/ej*j+de,e.e/dd Z0dS )!    N)set_backend)
_pocketfft)mock_backend)assert_allcloseassert_equalfftfft2fftnifftifft2ifftnrfftrfft2rfftnirfftirfft2irfftndctidctdctnidctndstidstdstnidstnzfunc, np_func, mockc             C   s~   t jdjd
}||}t| ||dd ttdd, d|_| |}t||j t|jd	 W d Q R X t| ||dd d S )N   
      g|=)ZatolT)onlyr      )r   r   )	nparangereshaper   r   r   number_callsr   return_value)funcZnp_funcmockxZanswery r)   P/var/www/html/virt/lib64/python3.6/site-packages/scipy/fft/tests/test_backend.pytest_backend_call-   s    r+   z
func, mockc             C   s   t jdjd}tjtdd | |dd W d Q R X ttdd	D d
|_| |dd}t	||j
 t	|jd t	|jd d d W d Q R X d S )Nr   r   r   zprecomputed plan)matchZfoo)planT)r   r   r   r-   )r   r   )r    r!   r"   pytestZraisesNotImplementedErrorr   r   r#   r   r$   Z	last_args)r%   r&   r'   r(   r)   r)   r*   test_backend_planK   s    r0   )r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )1Znumpyr    Z	scipy.fftZscipyr   r   Zscipy.fft.testsr   Znumpy.testingr   r   r.   fnamesr   r   r	   r
   r   r   r   r   r   r   r   r   ZhfftZhfft2ZhfftnZihfftZihfft2Zihfftnr   r   r   r   r   r   r   r   Znp_funcsfuncsZmocksmarkZparametrizezipr+   Z
plan_funcsZ
plan_mocksr0   r)   r)   r)   r*   <module>   sd        