File: //usr/local/lib/python3.10/dist-packages/numpy/fft/tests/__pycache__/test_pocketfft.cpython-310.pyc
o
���g_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�pi�arange�reshape�sum�exp)�x�L�phase� r �I/usr/local/lib/python3.10/dist-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 ]1}ttj�tj�|d|� ��|d|� dd� ttj�tj�|d|� �|�|d|� dd� qd S )Ni � �?r r g�-���q=��atol)r �ranger r r �ifft�irfft�rfft)r �maxlenr �xr�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_fftr2 �Nr0 r3 r5 c C s~ t d�dt d� }t|tjjtjj||d�|d�dd� tjtdd�� tjjg |d� W d � d S 1 s8w Y d S )Nr. r# r1 r/ r$ z!Invalid number of FFT data points)�match)r r r r r'