%PDF- %PDF-
| Direktori : /usr/lib64/python3.9/site-packages/numpy/fft/tests/__pycache__/ |
| Current File : //usr/lib64/python3.9/site-packages/numpy/fft/tests/__pycache__/test_pocketfft.cpython-39.opt-1.pyc |
a
z[yc2 � @ s� d dl Zd dlZd dlmZ d dlmZmZmZ d dl Z d dl
Z
dd� ZG dd� d�ZG dd � d �Z
ej�d
ejejejejg�ej�ddd
g�ej�dejjejjejjejjejjejjg�dd� ���ZG dd� d�ZdS )� N)�random)�assert_array_equal�
assert_raises�assert_allclosec C sN t | �}dtj t�|�| }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�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, r. c C s| t d�dt d� }t|tjjtjj||d�|d�dd� tjtdd�� tjjg |d� W d � n1 sn0 Y d S )Nr'