%PDF- %PDF-
Mini Shell

Mini Shell

Direktori : /usr/lib64/python3.9/site-packages/numpy/fft/tests/__pycache__/
Upload File :
Create Path :
Current File : //usr/lib64/python3.9/site-packages/numpy/fft/tests/__pycache__/test_pocketfft.cpython-39.opt-1.pyc

a

z[yc2�@s�ddlZddlZddlmZddlmZmZmZddl	Z	ddl
Z
dd�ZGdd�d�ZGdd	�d	�Z
ej�d
ejejejejg�ej�ddd
g�ej�dejjejjejjejjejjejjg�dd����ZGdd�d�ZdS)�N)�random)�assert_array_equal�
assert_raises�assert_allclosecCsNt|�}dtjt�|�|}t�|��dd�|}tj|t�|�dd�S)Ny��������axis)�len�npZpiZarangeZreshape�sumZexp)�x�LZphase�r�D/usr/lib64/python3.9/site-packages/numpy/fft/tests/test_pocketfft.py�fft1src@seZdZdd�ZdS)�TestFFTShiftcCstttjjgd�d�dS)N)r��r)r�
ValueErrorr�fft)�selfrrr�
test_fft_nszTestFFTShift.test_fft_nN)�__name__�
__module__�__qualname__rrrrrrsrc@s�eZdZdd�Zdd�Zej�dd�dd��Zd	d
�Z	dd�Z
d
d�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zej�d!ejjejjejjejjg�d"d#��Zd$d%�Zej�d&ejejejej g�d'd(��Z!d)S)*�	TestFFT1Dc	Cs�d}t|�dt|�}t|�}td|�D]b}ttj�tj�|d|���|d|�dd�ttj�tj�|d|��|�|d|�dd�q*dS)Ni��?rrg�-���q=��atol)r�rangerrr�ifft�irfft�rfft)r�maxlenr
Zxr�irrr�
test_identitys&��zTestFFT1D.test_identitycCs�td�dtd�}tt|�tj�|�dd�tt|�tjj|dd�dd�tt|�t�d�tjj|dd�dd�tt|�dtjj|d	d�dd�dS)
N�r���ư>r�backward��norm�ortho�>@�forward)rrrrr�sqrt�rr
rrr�test_fft$s��zTestFFT1D.test_fftr+�Nr)r,r.cCs|td�dtd�}t|tjjtjj||d�|d�dd�tjtdd�� tjjg|d�Wd�n1sn0YdS)Nr'rr*r(rz!Invalid number of FFT data points)�match)rrrrr!�pytestZraisesr)rr+r
rrr�	test_ifft-s��zTestFFT1D.test_ifftcCs�td�dtd�}ttjjtjj|dd�dd�tj�|�dd�ttj�|�tjj|dd	�dd�ttj�|�t�d
�tjj|dd	�dd�ttj�|�dtjj|d
d	�dd�dS�N�r'�rrrrr(rr)r*�Xr,���@r.)rrrr�fft2r/r0rrr�	test_fft28s����zTestFFT1D.test_fft2cCs�td�dtd�}ttjjtjj|dd�dd�tj�|�dd�ttj�|�tjj|dd	�dd�ttj�|�t�d
�tjj|dd	�dd�ttj�|�dtjj|d
d	�dd�dSr6)rrrrr!�ifft2r/r0rrr�
test_ifft2Cs����zTestFFT1D.test_ifft2cCs�td�dtd�}ttjjtjjtjj|dd�dd�dd�tj�|�dd�ttj�|�tjj|d	d
�dd�ttj�|�t�d�tjj|dd
�dd�ttj�|�d
tjj|dd
�dd�dS�N�r'r8�
rrrrrr(rr)r*�pr,�p�@r.)rrrr�fftnr/r0rrr�	test_fftnNs&����zTestFFT1D.test_fftncCs�td�dtd�}ttjjtjjtjj|dd�dd�dd�tj�|�dd�ttj�|�tjj|d	d
�dd�ttj�|�t�d�tjj|dd
�dd�ttj�|�d
tjj|dd
�dd�dSr?)rrrrr!�ifftnr/r0rrr�
test_ifftnZs&����zTestFFT1D.test_ifftnc	Cs�td�}|jd|jfD]�}dD]>}ttjj|||d�d|dd�tjj|||d�dd�q ttjj||d�tjj||d	d�dd�ttjj||d�t�|�tjj||d
d�dd�ttjj||d�|tjj||dd�dd�qdS)Nr'rr2��nr+rr(r)rIr)r,r.)r�sizerrrr#r/)rr
rIr+rrr�	test_rfftfs& ����zTestFFT1D.test_rfftcCs�td�}t|tj�tj�|��dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�dS)Nr'r(rr)r*r,r.)rrrrr"r#r0rrr�
test_irfftws"������zTestFFT1D.test_irfftcCs�td�}ttj�|�dd�dd�ftj�|�dd�ttj�|�tjj|dd�dd�ttj�|�t�d�tjj|dd�dd�ttj�|�d	tjj|d
d�dd�dS)Nr7�r(rr)r*r9r,r:r.)rrrrr;�rfft2r/r0rrr�
test_rfft2�s.���zTestFFT1D.test_rfft2cCs�td�}t|tj�tj�|��dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�dS)Nr7r(rr)r*r,r.)rrrrZirfft2rNr0rrr�test_irfft2�s"������zTestFFT1D.test_irfft2cCs�td�}ttj�|�dd�dd�dd�ftj�|�dd�ttj�|�tjj|dd�dd�ttj�|�t�d�tjj|dd�dd�ttj�|�d	tjj|d
d�dd�dS)Nr@�r(rr)r*rBr,rCr.)rrrrrD�rfftnr/r0rrr�
test_rfftn�s4���zTestFFT1D.test_rfftncCs�td�}t|tj�tj�|��dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�dS)Nr@r(rr)r*r,r.)rrrr�irfftnrRr0rrr�test_irfftn�s"������zTestFFT1D.test_irfftncCs�td�dtd�}t�td�|td�f�}t�||ddd���f�}ttj�|�tj�|�dd�ttj�|�tjj|dd�dd�ttj�|�t�d	�tjj|d
d�dd�ttj�|�dtjj|dd�dd�dS)
N�rrrr(rr)r*r'r,r-r.)rr�concatenate�conjrr�hfftr/�rr
Zx_hermrrr�	test_hfft�s���zTestFFT1D.test_hfftcCs�td�dtd�}t�td�|td�f�}t�||ddd���f�}t|tj�tj�|��dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|d	d�d	d�dd�t|tjjtjj|d
d�d
d�dd�dS)NrVrrrr(rr)r*r,r.)rrrWrXrr�ihfftrYrZrrr�
test_ihfft�s2���������zTestFFT1D.test_ihfft�opcCsPtd�}gd�}|D]6}|t�||��}t�|||d�|�}t||dd�qdS)Nr@))rrr)rrr)rrr)rrr)rrr)rrr��axesr(r)rrZ	transposer)rr^r
r`�aZop_trZtr_oprrr�	test_axes�szTestFFT1D.test_axesc	Cs�td�}tj�|�}|jd}tjjtjjftjjtjjftjj	tjj
fg}|D]Z\}}|jd|jfD]@}dD]6}||||d�}||||d�}t|tj�|�dd�qpqhqPdS)Nr'rr2rHr(r)rrZlinalgr+rJrr!r#r"r\rYr)	rr
Zx_normrIZ
func_pairsZforwZbackr+�tmprrr�test_all_1d_norm_preserving�s
��z%TestFFT1D.test_all_1d_norm_preserving�dtypecCsNtd��|�}ttj�tj�|��|dd�ttj�tj�|��|dd�dS)Nr'r(r)r�astyperrrr!r"r#)rrer
rrr�test_dtypes�szTestFFT1D.test_dtypesN)"rrrr&r1r4�mark�parametrizer5r<r>rErGrKrLrOrPrSrUr[r]rrrDrFrRrTrbrdZhalfZsingleZdoubleZ
longdoublergrrrrrs2
	






�
�rre�order�Fznon-contiguousrcCs0tj�d�}|�ddd�j|dd�}dt�t�|j��t�|j	�j
}|dkr\t�|�}n"|ddd	�}t�|ddd	��}|j
�d
�r�td�D],}|||d�}|||d�}	t||	||d
�q�nj|j
�d��r&gd�}
|j
�d�r�|
�gd��|
D],}|||d�}|||d�}	t||	||d
�q�nt��dS)N�*���
F)�copyg @rkrrrr)rZrtol)r;rD))rr)rr)rrrD))r)r)rNr_)rrZRandomStateZrandrfr/Zlog2rJZfinforeZepsZasfortranarrayZascontiguousarrayr�endswithr r�extendr)rerjr�rng�XZ_tol�Yr	ZX_resZY_resr`�axrrr�test_fft_with_order�s*$rwc@s<eZdZdZdZdd�Zdd�Zdd�Zd	d
�Zdd�Z	d
S)�TestFFTThreadSafe�)i ��cs|�fdd��t������}���fdd�t|j�D�}dd�|D�dd�|D�t|j�D]}t�jdd�|d	�q^dS)
Ncs|��|��dS�N)Zput)�args�q)�funcrr�workersz.TestFFTThreadSafe._test_mtsame.<locals>.workercsg|]}tj���fd��qS))�targetr|)�	threading�Thread)�.0r%)r|r}rrr�
<listcomp>s�z2TestFFTThreadSafe._test_mtsame.<locals>.<listcomp>cSsg|]}|���qSr)�start�r�r
rrrr��cSsg|]}|���qSr)�joinr�rrrr�r��)�timeoutz6Function returned wrong value in multithreaded context)�queueZQueuer �threadsr�get)rr~r|Zexpected�tr%r)r|r~r}rr�_test_mtsames��zTestFFTThreadSafe._test_mtsamecCs(t�|j�dd}|�tjj|�dS�Nry)r�ones�input_shaper�r�rrarrrr1%szTestFFTThreadSafe.test_fftcCs(t�|j�dd}|�tjj|�dSr�)rr�r�r�rr!r�rrrr5)szTestFFTThreadSafe.test_ifftcCs t�|j�}|�tjj|�dSr{)rr�r�r�rr#r�rrrrK-szTestFFTThreadSafe.test_rfftcCs(t�|j�dd}|�tjj|�dSr�)rr�r�r�rr"r�rrrrL1szTestFFTThreadSafe.test_irfftN)
rrrr�r�r�r1r5rKrLrrrrrxsrx)Znumpyrr4Znumpy.randomrZ
numpy.testingrrrr�r�rrrrhriZfloat32Zfloat64Z	complex64Z
complex128rr;rDr!r=rFrwrxrrrr�<module>s,Q���

Zerion Mini Shell 1.0