Qƅ15 d Z ddlZddlmZ ddlmZmZ G d d Z G d d Z G d d
Z G d d Z
dS )
zfTest functions for fftpack.helper module
Copied from fftpack.helper by Pearu Peterson, October 2005
N)assert_array_almost_equal)fftpic , e Zd Zd Zd Zd Zd Zd ZdS ) TestFFTShiftc 6 g d}g d}t t j | | t t j | | g d}g d}t t j | | t t j | | 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)selfxys o/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/fft/tests/test_helper.pytest_definitionz TestFFTShift.test_definition
s + + + + + +!#,q//1555!#-"2"2A666 / / / / / /!#,q//1555!#-"2"2A66666 c dD ]V}t j |f }t t j t j | | Wd S )N)r
r
d )nprandomr r r r )r nr s r test_inversezTestFFTShift.test_inverse s[ $ I IA !&&A %cmCLOO&D&Da H H H H I Ir c * g dg dg dg}g dg dg dg}t t j |d | t t j |d t j |d
t t j |d | t t j |d t j |d
t t j | | t t j | | 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 freqsshifteds r test_axes_keywordz TestFFTShift.test_axes_keyword s JJJ 5 <<JJJ7!#,u6"B"B"BGLLL!#,u1"="="="%,u4"@"@"@ B B B!#-f"E"E"EuMMM!#-a"@"@"@"%-d"C"C"C E E E "#,u"5"5w???!#-"8"8%@@@@@r c P ddgddgddgg}ddgddgddgg}t t j |d | t t j |d | t t j |d | t t j |dg | ddgddgddgg}t t j |d | t t j |d | ddgddgddgg}t t j |d | t t j |d | t t j |ddg | t t j |ddg | t t j |d
| t t j |d
| t t j | | t t j | | 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_dim1shift_dim_boths r test_uneven_dimsz TestFFTShift.test_uneven_dims) s: F
F
F
F
F
F
"#,u1"="="=zJJJ!#-
"C"C"CUKKK!#,u4"@"@"@*MMM!#-
!"E"E"EuMMM F
F
F
"#,u1"="="=zJJJ!#-
"C"C"CUKKK F
F
F
"#,u6"B"B"BNSSS!#-V"L"L"LeTTT!#,uAq6"B"B"BNSSS!#-aV"L"L"LeTTT "#,u4"@"@"@.QQQ!#-T"J"J"JERRR!#,u"5"5~FFF!#-"?"?GGGGGr c
ddl mm mm
d
fd }d
fd }t d D ]}t d D ]}dD ]|}t j || }t t j || ||| t t j || ||| }dS ) z\ Test that the new (>=v1.15) implementation (see #10073) is equal to the original (<=v1.14) r )asarray concatenatearangetakeNc ,
| }|j }| t t | }nt |t r|f}|}|D ]D}|j | }|dz dz } || | f } ||| }E|S )z& How fftshift was implemented in v1.14Nr
r ndimlistrange
isinstanceintshape
r r( tmpr: r kr# p2mylistr6 r4 r5 r7 s
r original_fftshiftz>TestFFTShift.test_equal_to_original..original_fftshiftZ s '!**C8D|E$KK((D#&&
wA
'
'IaL!e\$ffRmmVVBZZ%@AADFA&&Hr c 2
| }|j }| t t | }nt |t r|f}|}|D ]G}|j | }||dz dz z
} || | f } ||| }H|S )z( How ifftshift was implemented in v1.14 Nr
r r9 r@ s
r original_ifftshiftz?TestFFTShift.test_equal_to_original..original_ifftshiftj s '!**C8D|E$KK((D#&&
wA
'
'IaL!a%A %$ffRmmVVBZZ%@AADFA&&Hr )r r
Nr) r& )N)
numpy.corer4 r5 r6 r7 r< r! r" randr r r r ) r rE rG ij axes_keywordinpr6 r4 r5 r7 s @@@@r test_equal_to_originalz#TestFFTShift.test_equal_to_originalV s` AAAAAAAAAAAA $ r U UA2YY
U
U$> U UL )..A..C-cl3 .M.M.?.?\.R.RT T T .cmC .N.N.@.@l.S.SU U U U
U
U U Ur N)__name__
__module__ __qualname__r r$ r, r2 rO r r r r sn 7 7 7I I I
A A A+H +H +HZ/U /U /U /U /Ur r c e Zd Zd ZdS ) TestFFTFreqc v g d}t dt j d z | t dt z t j dt z | g d}t dt j d z | t dt z t j dt z | d S )Nr r r
)r r fftfreqr r r s r r zTestFFTFreq.test_definition s + + +!!CKNN"2A666!!B$s{1b'9'9"91=== / / /!"S[__"4a888!"R% B(;(;";Q?????r NrP rQ rR r rS r r rU rU s( @ @ @ @ @r rU c e Zd Zd ZdS ) TestRFFTFreqc v g d}t dt j d z | t dt z t j dt z | g d}t dt j d z | t dt z t j dt z | d S )N)r r
r r r
r )r r
r r r
r. rW )r r rfftfreqr rY s r r z TestRFFTFreq.test_definition s OO!!CLOO"3Q777!!B$s|Ar':':":A>>> !"S\"%5%5"5q999!"R% R(<(<"rl s
3 3 3 3 3 3 zU zU zU zU zU zU zU zUz@ @ @ @ @ @ @ @A A A A A A A A ! ! ! ! ! ! ! ! ! !r