o
    tBh                     @   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*+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*+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 dd}||}t| ||dd ttdd d|_| |}t||j t|jd W d    n1 s9w   Y  t| ||dd d S )	N   
      g|=)atolTonlyr      )	nparangereshaper   r   r   number_callsr   return_value)funcnp_funcmockxanswery r.   s/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/scipy/fft/tests/test_backend.pytest_backend_call-   s   r0   z
func, mockc                 C   s   t dd}tjtdd | |dd W d    n1 s w   Y  ttdd( d	|_| |dd}t	||j
 t	|jd
 t	|jd
 d d W d    d S 1 sVw   Y  d S )Nr   r   zprecomputed plan)matchfoo)planTr    r   r"   r3   )r#   r$   r%   pytestraisesNotImplementedErrorr   r   r&   r   r'   	last_args)r(   r*   r+   r-   r.   r.   r/   test_backend_planK   s   "r8   )1numpyr#   	scipy.fftscipyr   r   scipy.fft.testsr   numpy.testingr   r   r4   fnamesr   r   r	   r
   r   r   r   r   r   r   r   r   hffthfft2hfftnihfftihfft2ihfftnr   r   r   r   r   r   r   r   np_funcsfuncsmocksmarkparametrizezipr0   
plan_funcs
plan_mocksr8   r.   r.   r.   r/   <module>   sj    		

