https://t.me/RX1948
Server : Apache
System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64
User : dh_edsupp ( 6597262)
PHP Version : 8.2.26
Disable Function : NONE
Directory :  /lib/python3/dist-packages/numpy/fft/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Current File : //lib/python3/dist-packages/numpy/fft/tests/__pycache__/test_pocketfft.cpython-310.pyc
o

6��a2�@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_allclosecCsRt|�}dtjt�|�t|�}t�|��dd�|}tj|t�|�dd�S)Ny��������axis)�len�np�pi�arange�float�reshape�sum�exp)�x�L�phase�r�@/usr/lib/python3/dist-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__rrrrrrsrc@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]1}ttj�tj�|d|���|d|�dd�ttj�tj�|d|��|�|d|�dd�qdS)Ni��?rrg�-���q=��atol)r�rangerrr�ifft�irfft�rfft)r�maxlenr�xr�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�rrrrr�test_fft$s�
�zTestFFT1D.test_fftr2�Nr0r3r5cCs~td�dtd�}t|tjjtjj||d�|d�dd�tjtdd��tjjg|d�Wd�dS1s8wYdS)Nr.r#r1r/r$z!Invalid number of FFT data points)�match)rrrrr'�pytest�raisesr)rr2rrrr�	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.�r#rrrr/r$r0r1�Xr3���@r5)rrrr�fft2r6r7rrr�	test_fft28����
�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�dSr>)rrrrr'�ifft2r6r7rrr�
test_ifft2CrEzTestFFT1D.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.r@�
r#rrrrr/r$r0r1�pr3�p�@r5)rrrr�fftnr6r7rrr�	test_fftnN�&���
�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�dSrH)rrrrr'�ifftnr6r7rrr�
test_ifftnZrOzTestFFT1D.test_ifftnc	Cs�td�}|jd|jfD]g}dD]}ttjj|||d�d|dd�tjj|||d�dd�qttjj||d�tjj||d	d�dd�ttjj||d�t�|�tjj||d
d�dd�ttjj||d�|tjj||dd�dd�qdS)Nr.rr9��nr2rr/r$)rSr0r3r5)r�sizerrrr)r6)rrrSr2rrr�	test_rfftfs( �����zTestFFT1D.test_rfftcC�td�}t|tj�tj�|��dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�t|tjjtjj|dd�dd�dd�dS)Nr.r/r$r0r1r3r5)rrrrr(r)r7rrr�
test_irfftw�"�����
�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)Nr?�r/r$r0r1rAr3rBr5)rrrrrC�rfft2r6r7rrr�
test_rfft2�s.��
�zTestFFT1D.test_rfft2cCrV)Nr?r/r$r0r1r3r5)rrrr�irfft2rZr7rrr�test_irfft2�rXzTestFFT1D.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)NrI�r/r$r0r1rKr3rLr5)rrrrrM�rfftnr6r7rrr�
test_rfftn�s4��
�zTestFFT1D.test_rfftncCrV)NrIr/r$r0r1r3r5)rrrr�irfftnr_r7rrr�test_irfftn�rXzTestFFT1D.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�r#rrr/r$r0r1r.r3r4r5)rr�concatenate�conjrr�hfftr6�rr�x_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)Nrcr#rrr/r$r0r1r3r5)rrrdrerr�ihfftrfrgrrr�
test_ihfft�s2��������
�zTestFFT1D.test_ihfft�opcCsPtd�}gd�}|D]}|t�||��}t�|||d�|�}t||dd�q
dS)NrI))rrr)rrr)rrr)rrr)rrr)rrr��axesr/r$)rr�	transposer)rrlrrn�a�op_tr�tr_oprrr�	test_axes�s�zTestFFT1D.test_axesc	Cs�td�}tj�|�}|jd}tjjtjjftjjtjjftjj	tjj
fg}|D]-\}}|jd|jfD] }dD]}||||d�}||||d�}t|tj�|�dd�q8q4q(dS)Nr.rr9rRr/r$)rr�linalgr2rTrr'r)r(rjrfr)	rr�x_normrS�
func_pairs�forw�backr2�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))rr{rrrr�test_dtypes�s"zTestFFT1D.test_dtypesN)"rr r!r-r8r;�mark�parametrizer=rDrGrNrQrUrWr[r]r`rbrirkrrrMrPr_rarsrz�half�single�double�
longdoubler}rrrrr"s4
	






�
�r"r{�order�Fznon-contiguousrcCs.tj�d�}|�ddd�j|dd�}dt�t�|j��t�|j	�j
}|dkr.t�|�}n|ddd	�}t�|ddd	��}|j
�d
�rbtd�D]}|||d�}|||d�}	t||	||d
�qIdS|j
�d�r�gd�}
|j
�d�ry|
�gd��|
D]}|||d�}|||d�}	t||	||d
�q{dSt��)N�*���
F)�copyg @r�rrrr)r%�rtol)rCrM))rr)rr)rrrM))r)r)rNrm)rr�RandomState�randr|r6�log2rT�finfor{�eps�asfortranarray�ascontiguousarrayr�endswithr&r�extendr)r{r�r�rng�X�_tol�Yr	�X_res�Y_resrn�axrrr�test_fft_with_order�s.$��r�c@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)�put)�args�q)�funcrr�workersz.TestFFTThreadSafe._test_mtsame.<locals>.workercsg|]}tj���fd��qS))�targetr�)�	threading�Thread)�.0r,)r�r�r�rr�
<listcomp>s�z2TestFFTThreadSafe._test_mtsame.<locals>.<listcomp>cS�g|]}|���qSr)�start�r�rrrrr��cSr�r)�joinr�rrrr�r��)�timeoutz6Function returned wrong value in multithreaded context)�queue�Queuer&�threadsr�get)rr�r��expected�tr,r)r�r�r�r�r�_test_mtsames���zTestFFTThreadSafe._test_mtsamecCs(t�|j�dd}|�tjj|�dS�Nry)r�ones�input_shaper�r�rrprrrr8%�zTestFFTThreadSafe.test_fftcC�(t�|j�dd}|�tjj|�dSr�)rr�r�r�rr'r�rrrr=)r�zTestFFTThreadSafe.test_ifftcCs t�|j�}|�tjj|�dSr�)rr�r�r�rr)r�rrrrU-szTestFFTThreadSafe.test_rfftcCr�r�)rr�r�r�rr(r�rrrrW1r�zTestFFTThreadSafe.test_irfftN)
rr r!r�r�r�r8r=rUrWrrrrr�sr�)�numpyrr;�numpy.randomr�
numpy.testingrrrr�r�rrr"r~r�float32�float64�	complex64�
complex128rrCrMr'rFrPr�r�rrrr�<module>s.Q���

https://t.me/RX1948 - 2025