Current File : //lib64/python3.9/site-packages/numpy/fft/tests/__pycache__/test_helper.cpython-39.pyc
a
z[yc � @ 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 sd g d�}g d�}t t�|�|� t t�|�|� g d�}g d�}t t�|�|� t t�|�|� d S )N� r � � � � ������������) r r r
r r r r r r
�
r r r r r
���r r r
r )
r r r r
r r r r r r
�r r �fftshift� ifftshift)�self�x�y� r �A/usr/lib64/python3.9/site-packages/numpy/fft/tests/test_helper.py�test_definition
s z TestFFTShift.test_definitionc C s2 dD ](}t j�|f�}tt�t�|��|� qd S )N)r r
� �d �� )�np�randomr r r r )r �nr r r r � test_inverse s zTestFFTShift.test_inversec C s� g d�g d�g d�g}g d�g d�g d�g}t tj|dd�|� t tj|d d�tj|d
d�� t tj|dd�|� t tj|d d�tj|d
d�� t t�|�|� t t�|�|� d S ) N)r r r )r r
r )r r
r )r r r
)r r r )r r r
�r r ��axesr �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! Nr )r r% Z
shift_dim0Z
shift_dim1Zshift_dim_bothr r r �test_uneven_dims) s< ����z TestFFTShift.test_uneven_dimsc s� ddl m�m�m� m� d
� ���fdd� }d � ���fdd� }td�D ]Z}td�D ]L}d D ]B}tj�||�}t t
� ||�|||�� t t
� ||�|||�� q\qTqHdS ) z\ Test that the new (>=v1.15) implementation (see #10073) is equal to the original (<=v1.14) r )�asarray� concatenate�arange�takeNc s~ �| �}|j }|du r$tt|��}nt|t�r4|f}|}|D ]<}|j| }|d d }�� ||�� |�f�}�|||�}q<|S )z& How fftshift was implemented in v1.14Nr r ��ndim�list�range�
isinstance�int�shape� r r# �tmpr. r �kr Zp2Zmylist�r+ r) r* r, r r �original_fftshiftZ s
z>TestFFTShift.test_equal_to_original.<locals>.original_fftshiftc s� �| �}|j }|du r$tt|��}nt|t�r4|f}|}|D ]@}|j| }||d d }�� ||�� |�f�}�|||�}q<|S )z( How ifftshift was implemented in v1.14 Nr r r- r4 r7 r r �original_ifftshiftj s
z?TestFFTShift.test_equal_to_original.<locals>.original_ifftshift� )r r Nr$ r! )N)N)
Z
numpy.corer) r* r+ r, r0 r r Zrandr r r r )r r8 r9 �i�jZ axes_keyword�inpr r7 r �test_equal_to_originalV s � �z#TestFFTShift.test_equal_to_originalN)�__name__�
__module__� __qualname__r r r& r( r> r r r r r s
-r c @ s e Zd Zdd� ZdS )� TestFFTFreqc C sp g d�}t dt�d� |� t dt t�dt� |� g d�}t dt�d� |� t dt t�dt� |� d S )Nr r r �
)r r Zfftfreqr �r r r r r r � s zTestFFTFreq.test_definitionN�r? r@ rA r r r r r rB � s rB c @ s e Zd Zdd� ZdS )� TestRFFTFreqc C sp g d�}t dt�d� |� t dt t�dt� |� g d�}t dt�d� |� t dt t�dt� |� d S )N)r r r r r
r )r r r r r
r' rC )r r Zrfftfreqr rD r r r r � s z TestRFFTFreq.test_definitionNrE r r r r rF � s rF c @ s e Zd Zdd� ZdS )�
TestIRFFTNc C s2 t j�d�\}}|d| }d}tj||d� d S )N)r r: � � 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)r? r@ rA rL r r r r rG � s rG ) �__doc__Znumpyr Z
numpy.testingr r r r rB rF rG r r r r �<module> s }
Mini Shell