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 : |
o 6��a2 � @ 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 sR t | �}dtj t�|�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�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'