Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/fft/tests/__pycache__/test_helper.cpython-37.pyc
Назад
B �5�g � @ sd d Z ddlZddlmZ ddlmZmZ G dd� d�ZG dd� d�ZG d d � d �Z G d d � d �Z dS ) zfTest functions for fftpack.helper module Copied from fftpack.helper by Pearu Peterson, October 2005 � N)�assert_array_almost_equal)�fft�pic @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestFFTShiftc C s� ddddddddd g }dddd dddddg }t t�|�|� t t�|�|� dddddd dddd g }d dddd dddddg }t t�|�|� t t�|�|� d S ) Nr � � � � ���������������)r r �fftshift� ifftshift)�self�x�y� r �Q/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/fft/tests/test_helper.py�test_definition s z TestFFTShift.test_definitionc C s6 x0dD ](}t j�|f�}tt�t�|��|� qW d S )N)r r � �d �� )�np�randomr r r r )r �nr r r r � test_inverse s zTestFFTShift.test_inversec C s� dddgdddgddd gg}d ddgdddgdddgg}t tj|d d �|� t tj|dd �tj|d d �� t tj|d d �|� t tj|dd �tj|d d �� t t�|�|� t t�|�|� d S ) Nr r r r r r r r r )r r )�axes)r )r r r r )r �freqsZshiftedr r r �test_axes_keyword s z TestFFTShift.test_axes_keywordc C sv ddgddgddgg}ddgddgddgg}t tj|dd�|� t tj|dd�|� t tj|dd�|� t tj|dgd�|� ddgddgddgg}t tj|dd�|� t tj|dd�|� ddgddgddgg}t tj|d d�|� t tj|d d�|� t tj|ddgd�|� t tj|ddgd�|� t tj|d d�|� t tj|d d�|� t t�|�|� t t�|�|� d S ) z1 Test 2D input, which has uneven dimension sizes r r r r r � )r )r )r r N)r r r r )r r Z shift_dim0Z shift_dim1Zshift_dim_bothr r r �test_uneven_dims) s4 z TestFFTShift.test_uneven_dimsc s� ddl m�m�m� m� d � ���fdd� }d � ���fdd� }xntd�D ]b}x\td�D ]P}xJd D ]B}tj�||�}t t � ||�|||�� t t � ||�|||�� qbW qXW qJW dS ) z\ Test that the new (>=v1.15) implementation (see #10073) is equal to the original (<=v1.14) r )�asarray� concatenate�arange�takeNc s� �| �}|j }|dkr$tt|��}nt|t�r4|f}|}xD|D ]<}|j| }|d d }�� ||�� |�f�}�|||�}q>W |S )z& How fftshift was implemented in v1.14Nr r )�ndim�list�range� isinstance�int�shape) r r �tmpr' r �kr �p2�mylist)r% r# r$ r& r r �original_fftshiftZ s z>TestFFTShift.test_equal_to_original.<locals>.original_fftshiftc s� �| �}|j }|dkr$tt|��}nt|t�r4|f}|}xH|D ]@}|j| }||d d }�� ||�� |�f�}�|||�}q>W |S )z( How ifftshift was implemented in v1.14 Nr r )r' r( r) r* r+ r, ) r r r- r' r r. r r/ r0 )r% r# r$ r& r r �original_ifftshiftj s z?TestFFTShift.test_equal_to_original.<locals>.original_ifftshift� )r r N)r )r r )N)N) Z numpy.corer# r$ r% r& r) r r Zrandr r r r )r r1 r2 �i�jZ axes_keywordZinpr )r% r# r$ r& r �test_equal_to_originalV s z#TestFFTShift.test_equal_to_originalN)�__name__� __module__� __qualname__r r r r" r6 r r r r r s -r c @ s e Zd Zdd� ZdS )� TestFFTFreqc C s� ddddddddd g }t d t�d � |� t d t t�d t� |� dddddd dddd g }t d t�d � |� t d t t�d t� |� d S ) Nr r r r r r r r r r r � )r r Zfftfreqr )r r r r r r � s zTestFFTFreq.test_definitionN)r7 r8 r9 r r r r r r: � s r: c @ s e Zd Zdd� ZdS )� TestRFFTFreqc C s~ dddddg}t dt�d� |� t dt t�dt� |� ddddddg}t dt�d� |� t dt t�dt� |� d S ) Nr r r r r r r! r; )r r Zrfftfreqr )r r r r r r � s z TestRFFTFreq.test_definitionN)r7 r8 r9 r r r r r r< � s r< c @ s e Zd Zdd� ZdS )� TestIRFFTNc C s2 t j�d�\}}|d| }d}tj||d� d S )N)r r3 � � y �?)r )r )r r r Zirfftn)r �arZai�ar r r r �test_not_last_axis_success� s z%TestIRFFTN.test_not_last_axis_successN)r7 r8 r9 rB r r r r r= � s r= ) �__doc__�numpyr Z numpy.testingr r r r r: r< r= r r r r �<module> s }
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка