File: //usr/local/lib/python3.10/dist-packages/numpy/fft/tests/__pycache__/test_helper.cpython-310.pyc
o
���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 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 �F/usr/local/lib/python3.10/dist-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 �freqs�shiftedr 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% �
shift_dim0�
shift_dim1�shift_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 ]-}td�D ]&}d D ]!}tj�||�}t t
� ||�|||�� t t
� ||�|||�� q.q*q$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 }|du rtt|��}nt|t�r|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# �tmpr2 r �kr �p2�mylist�r/ r- r. r0 r r �original_fftshiftZ s
z>TestFFTShift.test_equal_to_original.<locals>.original_fftshiftc s� �| �}|j }|du rtt|��}nt|t�r|f}|}|D ] }|j| }||d d }�� ||�� |�f�}�|||�}q |S )z( How ifftshift was implemented in v1.14 Nr r r1 r8 r= r r �original_ifftshiftj s
z?TestFFTShift.test_equal_to_original.<locals>.original_ifftshift� )r r Nr$ r! )N)
�
numpy.corer- r. r/ r0 r4 r r �randr r r r )r r> r? �i�j� axes_keyword�inpr r= r �test_equal_to_originalV s � ����z#TestFFTShift.test_equal_to_originalN)�__name__�
__module__� __qualname__r r r' r, rG r r r r r s
-r c @ � e Zd Zdd� ZdS )� TestFFTFreqc C �p 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 �fftfreqr �r r r r r r � � zTestFFTFreq.test_definitionN�rH rI rJ r r r r r rL � � rL c @ rK )� TestRFFTFreqc C rM )N)r r r r r
r )r r r r r
r( rN )r r �rfftfreqr rP r r r r � rQ z TestRFFTFreq.test_definitionNrR r r r r rT � rS rT c @ rK )�
TestIRFFTNc C s2 t j�d�\}}|d| }d}tj||d� d S )N)r r@ � � y �?)r
r" )r r r �irfftn)r �ar�ai�ar# r r r �test_not_last_axis_success� s z%TestIRFFTN.test_not_last_axis_successN)rH rI rJ r] r r r r rV � rS rV ) �__doc__�numpyr �
numpy.testingr r r r rL rT rV r r r r �<module> s }