U
|
P ±d« ã @ sd d Z ddlZddlmZ ddlmZmZ G dd dZG dd dZG 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 úRd:\z\workplace\vscode\pyvenv\venv\Lib\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À 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 Ú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 mmm m d
|
fdd }d fdd }tdD ]Z}tdD ]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 }|dkr$tt|}nt|tr4|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 Úp2Zmylist©r) r' r( r* r r Úoriginal_fftshiftZ s
|
|
z>TestFFTShift.test_equal_to_original.<locals>.original_fftshiftc s | }|j }|dkr$tt|}nt|tr4|f}|}|D ]@}|j| }||d d } || |f}|||}q<|S )z( How ifftshift was implemented in v1.14 Nr r r+ r2 r6 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* r. r r Zrandr r r r )r r7 r8 ÚiÚjZ axes_keywordÚinpr r6 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 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©r> r? r@ r r r r r rA s rA 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% rB )r r Zrfftfreqr rC r r r r s z TestRFFTFreq.test_definitionNrD r r r r rE s rE c @ s e Zd Zdd ZdS )Ú
|
TestIRFFTNc C s2 t j d¡\}}|d| }d}tj||d d S )N)r r9 é é 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? r@ rK r r r r rF s rF ) Ú__doc__Únumpyr Z numpy.testingr r r r rA rE rF r r r r Ú<module> s }
|
