a BCCfL<у@s,dZddlmZmZddlmZddlmZddlZddl Z ddlZddlm Z ddlmZmZddlmZe ejаd бgZejjZgd вZddДZd dДZeddНGddДdГГZeddНGddДdГГZeddgdНGddДdГГZeddddgdНGdd Дd ГГZeddddgdНGd!d"Дd"ГГZdS)#zМIncludes test functions for fftpack.helper module Copied from fftpack.helper by Pearu Peterson, October 2005 Modified for Array API, 2023 щ)┌ next_fast_len┌_init_nd_shape_and_axes)┌assert_equal)┌raisesN)┌array_api_compatible)┌xp_assert_close┌SCIPY_DEVICE)┌fft┌skip_if_array_api)щщщщщщщ щ щiШicCstD]}tt|Г|ГqdS)N)┌_5_smooth_numbersrr)┌nйr·W/var/www/html/django/DPS/env/lib/python3.9/site-packages/scipy/fft/tests/test_helper.py┌test_next_fast_lensrcCsМ|}|dkrdsJВt|dГ\}}|dkr,q2|}qtd|ddГD]"}t||Г\}}|dkr^qB|}qFqB|dksИJd|Ыd|Ыd|ЫЭГВdS) NrFrrщzx=z is not z-smooth, remainder=)┌divmod┌range)┌xrZx_orig┌q┌r┌drrr┌_assert_n_smooth s r T)Znp_onlyc@sNeZdZddДZddДZddДZejje j dkded d НddДГZd dДZ dS)┌TestNextFastLencCs\tjаdбddД}|ГD]<}t|Г}t|dГ|t|dГksBJВt|dГ}t|dГqdS)Ni╥csstddГEdHdVdS)NriшiАy)rrrrr┌nums<sz0TestNextFastLen.test_next_fast_len..numsщFTr)┌np┌random┌seedrr )┌selfr"r┌mrrrr9s z"TestNextFastLen.test_next_fast_lencCsLtjtjtjtjtjtjg}|D]&}|dГ}t|Г}t|tt |ГГГq dS)Ni90) r$Zint16Zint32Zint64Zuint16Zuint32Zuint64rr┌int)r'ZITYPESZityprZtestNrrr┌test_np_integersHs z TestNextFastLen.test_np_integerscCsJddddddddddd d ddd dЬ}|абD]\}}tt|dГ|Гq,dS)Nrrrr rrrщщщiщщ@ )rrrr rrщrщr+r,щi¤r.r/Tй┌itemsrrйr'Zhamsr┌yrrr┌testnext_fast_len_smallOs ■z'TestNextFastLen.testnext_fast_len_smalllz$Hamming Numbers too large for 32-bitT)┌reasonr┌strictcCsddddddddddd d ddd dddddddddddddddЬ}|абD]\}}tt|dГ|ГqFdS)Nщ└╟hiАДщ+Йы2iB3ьА ╖lАsЎ'╖ьа>х9∙l ьf╪nl>Uk▒oь°qЕ. l╪xQ+t0 ьв!═ldopL?═ьэЕhRьЙ2 f;┘ьь`71Y!lрЕ("ь ь ╦єX l )r:i┴╟hi└=ur;i,Йы2r<lА ╖r=lб>х9∙r>lf╪nr?l°qЕ. r@lв!═rAlьЕhRrBlИ2 f;┘rCl rDl`71Y!rEl l ╦єX rFl! ╦єX Tr3r5rrr┌testnext_fast_len_bigWs6фz%TestNextFastLen.testnext_fast_len_bigcCs,tdddНdksJВtdddНdks(JВdS)Nr#T)┌realщr0F)┌targetrH)r)r'rrr┌test_keyword_args|sz!TestNextFastLen.test_keyword_argsN)┌__name__┌ __module__┌__qualname__rr*r7┌pytest┌markZxfail┌sys┌maxsize┌ ValueErrorrGrKrrrrr!6s■ "r!)Zcpu_onlyc@sleZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZddДZddДZ ddДZdS)┌Test_init_nd_shape_and_axesc CsF|аdб}d}d}d}g}t|||Г\}}||ks6JВ||ksBJВdS)Nr rй┌asarrayrй r'┌xpr┌shape┌axesZshape_expectedZ axes_expectedZ shape_resZaxes_resrrr┌test_py_0d_defaultsДs z/Test_init_nd_shape_and_axes.test_py_0d_defaultsc CsF|аdб}d}d}d}g}t|||Г\}}||ks6JВ||ksBJВdS)Ng@rrUrWrrr┌test_xp_0d_defaultsСs z/Test_init_nd_shape_and_axes.test_xp_0d_defaultsc CsL|аgdвб}d}d}d}dg}t|||Г\}}||ksJВ||ksJJВdS)Nйrrrrr йrrrrr й┌zerosrrWrrr┌test_xp_5d_defaults╘sz/Test_init_nd_shape_and_axes.test_xp_5d_defaultsc CsR|аgdвб}gdв}d}d}gdв}t|||Г\}}||ksBJВ||ksNJВdS)Nrd)rщ rirr )rrrrr rerfrWrrr┌test_xp_5d_set_shapeсsz0Test_init_nd_shape_and_axes.test_xp_5d_set_shapec CsR|аgdвб}d}gdв}d}gdв}t|||Г\}}||ksBJВ||ksNJВdS)Nrd)r rr)r rrrfrWrrr┌test_xp_5d_set_axesюsz/Test_init_nd_shape_and_axes.test_xp_5d_set_axesc CsV|аgdвб}gdв}gdв}d}gdв}t|||Г\}}||ksFJВ||ksRJВdS)Nrd)rrir)rrr)rrrrfrWrrr┌test_xp_5d_set_shape_axes√sz5Test_init_nd_shape_and_axes.test_xp_5d_set_shape_axescCs<|аdб}t|dddН\}}|dks(JВ|gdвks8JВdS)N)rrr r)rrrйrYrZr]rf)r'rXrrYrZrrr┌test_shape_axes_subsets z2Test_init_nd_shape_and_axes.test_shape_axes_subsetcCsШ|аdб}ttddНП*t|dddgddggdНWdГn1sF0YttddНП"t|dgdвdНWdГn1sД0Yttd dНП t|ddgdНWdГn1s└0Yttd dНП t|dd gdНWdГn1s№0YttddНП"t|dddgdНWdГn1Рs<0Yttd dНП*t|ddgddggddНWdГn1РsД0Yttd dНП"t|gdвddНWdГn1Рs─0YttddНП.t|аgdвбgdвdgdНWdГn1Рs0YttddНП t|dgddНWdГn1РsN0YttddНПt|d ddНWdГn1РsК0YdS)Nrz-axes must be a scalar or iterable of integers)┌matchrrr rm)gЁ?g@g@g@z$axes exceeds dimensionality of inputщ■ zall axes must be uniquerz.shape must be a scalar or iterable of integerszBwhen given, axes and shape arguments have to be of the same length)rrrrr]z1invalid number of data points $\[0\]$ specifiedz2invalid number of data points $\[-2\]$ specified)rg┌ assert_raisesrSr)r'rXrrrr┌test_errorssF 80 . . 2:2 & 0 z'Test_init_nd_shape_and_axes.test_errorsN)rLrMrNr[r\r^rarbrcrhrjrkrlrnrrrrrrrTБs rTZtorchz1torch.fft not yet implemented by array-api-compat)┌reasonsc@s,eZdZddДZddДZddДZddДZd S) ┌TestFFTShiftcCs||аgdвб}|аgdвб}ttа|б|Гttа|б|Г|аgdвб}|аgdвб}ttа|б|Гttа|б|ГdS)N) чrrrr щ№ щ¤ rpri) g└rwrprirrrrr ) rurrrr щ√ rvrwrpri) g└rvrwrprirrrrr йrVrr ┌fftshift┌ ifftshift)r'rXrr6rrr┌test_definitionBszTestFFTShift.test_definitioncCs8dD].}|аtjа|fбб}ttаtа|бб|ГqdS)N)rr rщdщ╙)rVr$r%rr r{rz)r'rXrrrrr┌test_inverseLszTestFFTShift.test_inversecCs└|аgdвgdвgdвgб}|аgdвgdвgdвgб}ttj|ddН|Гttj|d dНtj|d dНГttj|ddН|Гttj|d dНtj|d dНГttа|б|Гttа|б|ГdS)N)rurr)rr rv)rwrpri)gЁ┐rwrp)rrr)rvrr йrrйrZrйrry)r'rX┌freqsZshiftedrrr┌test_axes_keywordQs zTestFFTShift.test_axes_keywordcCsж|jddgddgddgg|jdН}|jddgddgddgg|jdН}ttj|ddН|Гttj|ddН|Гttj|d dН|Гttj|dgdН|Г|jddgddgddgg|jdН}ttj|ddН|Гttj|ddН|Г|jddgddgddgg|jdН}ttj|d dН|Гttj|d dН|Гttj|ddgdН|Гttj|ddgdН|Гttj|ddН|Гttj|ddН|Гttа|б|Гttа|б|ГdS)z1 Test 2D input, which has uneven dimension sizes rrrrr rй┌dtyperБrВrАN)rV┌float64rr rzr{)r'rXrГZ shift_dim0Z shift_dim1Zshift_dim_bothrrr┌test_uneven_dims\sT¤№¤№¤№¤№zTestFFTShift.test_uneven_dimsN)rLrMrNr|rrДrИrrrrrt>s rtZarray_api_strictZcupyz+fft not yet implemented by array-api-strictz0cupy.fft not yet implemented by array-api-compatc@seZdZddДZdS)┌TestFFTFreqcCs(t}z0|jgdв|j|dН}|jgdв|j|dН}Wn:tyn|jgdв|jdН}|jgdв|jdН}Yn0|jdtjd|dН|jdН}t||Г|jd|jtjd|j|dН|jdН}t||Г|jdtjd|dН|jdН}t||Г|jd|jtjd|j|dН|jdН}t||ГdS)N) rrrrr rvrwrpriйrЖ┌device) rrrrr rxrvrwrprirЕrйrXr)rrVrЗ┌ TypeErrorr Zfftfreqr┌piйr'rXrЛrZx2r6rrrr|Пs* ( (zTestFFTFreq.test_definitionNйrLrMrNr|rrrrrЙКsrЙc@seZdZddДZdS)┌TestRFFTFreqcCs(t}z0|jgdв|j|dН}|jgdв|j|dН}Wn:tyn|jgdв|jdН}|jgdв|jdН}Yn0|jdtjd|dН|jdН}t||Г|jd|jtjd|j|dН|jdН}t||Г|jdtjd|dН|jdН}t||Г|jd|jtjd|j|dН|jdН}t||ГdS)NrerК)rrrrr rrЕrrМr)rrVrЗrНr ZrfftfreqrrОrПrrrr|лs ( (zTestRFFTFreq.test_definitionNrРrrrrrСжsrС)┌__doc__Zscipy.fft._helperrrZ numpy.testingrrOrrq┌numpyr$rQZscipy.conftestrZscipy._lib._array_apirrZscipyr rPZusefixturesZ pytestmarkr rrr r!rTrtrЙrСrrrr┌sFJ= J