Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/fft/tests/__pycache__/test_pocketfft.cpython-37.pyc
Назад
B �5�g2 � @ 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�piZarange�floatZreshape�sum�exp)�x�LZphase� r �T/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/fft/tests/test_pocketfft.py�fft1 s r c @ s e Zd Zdd� ZdS )�TestFFTShiftc C s t ttjjdddgd� d S )Nr � � 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 |�}xptd|�D ]b}ttj�tj�|d|� ��|d|� dd� ttj�tj�|d|� �|�|d|� dd� q,W d S )Ni y �?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� y �?g���ư>)r �backward)�norm�orthog >@�forward)r r r r r �sqrt)r r r r r �test_fft$ s zTestFFT1D.test_fftr) )Nr( r* r+ c C sh t d�dt d� }t|tjjtjj||d�|d�dd� tjtdd�� tjjg |d� W d Q R X d S )Nr'