Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/numpy/fft/tests/__pycache__/test_pocketfft.cpython-39.pyc
Ðазад
a e�h_2 � @ s� d dl Zd dlZd dlmZ d dlmZmZmZm Z d dl Z d dlZdd� ZG dd� d�Z G dd � d �Zej�d ejejejejg�ej�ddd g�ej�dejjejjejjejjejjejjg�dd� ���Zejje dd�G dd� d��ZdS )� N)�random)�assert_array_equal� assert_raises�assert_allclose�IS_WASMc C sN t | �}dtj t�|�| }t�|��dd�| }tj| t�|� dd�S )Ny � ����� ��axis)�len�np�piZarangeZreshape�sum�exp)�x�LZphase� r ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/site-packages/numpy/fft/tests/test_pocketfft.py�fft1 s r c @ s e Zd Zdd� ZdS )�TestFFTShiftc C s t ttjjg d�d� d S )N)r � � r )r � ValueErrorr �fft)�selfr r r � test_fft_n s zTestFFTShift.test_fft_nN)�__name__� __module__�__qualname__r r r r r r s r c @ s� e Zd Zdd� Zdd� Zej�dd�dd� �Zd d � Z dd� Z d d� Zdd� Zdd� 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 C s� d}t |�dt |� }t |�}td|�D ]b}ttj�tj�|d|� ��|d|� dd� ttj�tj�|d|� �|�|d|� dd� q*d S )Ni � �?r r g�-���q=��atol)r �ranger r r �ifft�irfft�rfft)r �maxlenr Zxr�ir r r � test_identity s &��zTestFFT1D.test_identityc C s� t d�dt d� }tt|�tj�|�dd� tt|�tjj|dd�dd� tt|�t�d� tjj|dd�dd� tt|�d tjj|d d�dd� d S ) N� r ���ư>r! �backward��norm�ortho� >@�forward)r r r r r �sqrt�r r r r r �test_fft$ s ��zTestFFT1D.test_fftr. �Nr, r/ r1 c C s| t d�dt d� }t|tjjtjj||d�|d�dd� tjtdd�� tjjg |d� W d � n1 sn0 Y d S )Nr* r r- r+ r! z!Invalid number of FFT data points)�match)r r r r r$ �pytestZraisesr )r r. r r r r � test_ifft- s ��zTestFFT1D.test_ifftc C s� t d�dt d� }ttjjtjj|dd�dd�tj�|�dd� ttj�|�tjj|dd �dd� ttj�|�t�d � tjj|dd �dd� ttj�|�d tjj|d d �dd� d S �N�r* � r r r r r+ r! r, r- �X r/ � ��@r1 )r r r r �fft2r2 r3 r r r � test_fft28 s ����zTestFFT1D.test_fft2c C s� t d�dt d� }ttjjtjj|dd�dd�tj�|�dd� ttj�|�tjj|dd �dd� ttj�|�t�d � tjj|dd �dd� ttj�|�d tjj|d d �dd� d S r9 )r r r r r$ �ifft2r2 r3 r r r � test_ifft2C s ����zTestFFT1D.test_ifft2c C s� t d�dt d� }ttjjtjjtjj|dd�dd�dd�tj�|�dd� ttj�|�tjj|d d �dd� ttj�|�t�d� tjj|dd �dd� ttj�|�d tjj|dd �dd� d S �N�r* r; � r r r r r r+ r! r, r- �p r/ � p�@r1 )r r r r �fftnr2 r3 r r r � test_fftnN s &����zTestFFT1D.test_fftnc C s� t d�dt d� }ttjjtjjtjj|dd�dd�dd�tj�|�dd� ttj�|�tjj|d d �dd� ttj�|�t�d� tjj|dd �dd� ttj�|�d tjj|dd �dd� d S rB )r r r r r$ �ifftnr2 r3 r r r � test_ifftnZ s &����zTestFFT1D.test_ifftnc C s� t d�}|jd|j fD ]�}dD ]>}ttjj|||d�d |d d � tjj|||d�dd� q ttjj||d�tjj||d d�dd� ttjj||d�t�|� tjj||d d�dd� ttjj||d�| tjj||dd�dd� qd S )Nr* r r5 ��nr. r r+ r! )rL r, r/ r1 )r �sizer r r r&