gX<ddlmZddlZddlmZmZmZddlZddl Z ddl m Z m Z m Z dejejjzZejddd Zejd dd d Zejeddd dgegZejde e gejdejejejejgdZejde e gejdejejejgdZ GddZ!GddZ"dS)) factorialN)assert_allclose assert_equalassert_array_less)AAAFloaterHormannInterpolatorBarycentricInterpolatorg@numi i)basermethoddtypectj|jdzdz}|tur|dz}tjd}tjdd|}||tj|}|ddd |}t||tj||||j |ksJ|tur@|j j |ksJ|jj |ksJ|jj |jj ksJ|jj |ksJ|j tj|d ksJ|j tj|d ksJ|j tj|d ksJdS) Ng?dld wr r rsizertol?)npfinfoepsrrandom default_rnglinspacesinuniformastyperrrsupport_pointssupport_valueserrorsrealweightspoles result_typeresiduesroots)rrrrngzrz2s `/granary/edev/quantum/lib/python3.11/site-packages/scipy/interpolate/tests/test_bary_rational.pytest_dtype_preservationr2(s 8E?? $ & ,D +++   )   1 2 2C B'''Aq"&))A R % % , ,U 3 3BAAbEE26"::D1111 1R55;%     }}%....%....x~---- 9?e # # # # 7799?bnUB77 7 7 7 7 ::<< r!:!: : : : : 7799?bnUB77 7 7 7 7 7 7ctjd|}|||}|jjtj|dksJ|t uro|jjtj|dksJ|jjtj|dksJ|jjtj|dksJ| jtj|dksJ| jtj|dksJ| jtj|dksJ||jtj|dksJdS)Nrr?r) raranger(rr*rr$r%r&r)r+r,)rrr.r/s r1test_integer_promotionr7As[ "E"""Aq! A 9?bnUC88 8 8 8 8 }}%s)C)CCCCC%s)C)CCCCCx~s!;!;;;;; 7799?bnUB77 7 7 7 7 ::<< r!:!: : : : : 7799?bnUB77 7 7 7 7 1Q44:s33 3 3 3 3 3 3r3c eZdZdZejjdZdZdZ dZ dZ dZ dZ d Zd Zejd d d dfdddfdddfdddfdddfdddfdddfgdZdZejdddd d!d"d#d$d%d&g d'Zd(Zejjd)Zd*S)+TestAAActjtd5tdgddgdddn #1swxYwYtjtd5tdgdggdgdggdddn #1swxYwYtjtd5ttjgdgdddn #1swxYwYtjt 5tdgdgddddn #1swxYwYtjtd 5tdgdgd ddddS#1swxYwYdS) Nz same sizematchrr z1-Dfiniter5 max_termsgreaterr )pytestraises ValueErrorrrinf TypeErrorselfs r1test_input_validationzTestAAA.test_input_validationSs ]:[ 9 9 9   aV                   ]:U 3 3 3 ( ( !qc aS1#J ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ]:X 6 6 6   1#                   ]9 % % ) ) aSC ( ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ]:Y 7 7 7 ( ( aSB ' ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sW<AA"BB B -CCC:DD"DE##E'*E'ctjtd5ttt jtdddddS#1swxYwYdS)Nz AAA failedr;r r>)rAwarnsRuntimeWarningr UNIT_INTERVALrexprFs r1test_convergence_errorzTestAAA.test_convergence_error_s \. = = = D D  rvm44 C C C C D D D D D D D D D D D D D D D D D Ds/AAActjt}tt|}t |t|t t |tjtjtj|tj sJ|j j }tt|d}|j j |ksJdS)NatolgMbP?r) rrMrLrrTOLrnanisfiniterDr$r)rGfr/m1s r1test_expzTestAAA.test_expfs F= ! !  q ! !-((!#6666QQrvYY'''{11RV99%%%%%   "  qt , , ,$r))))))r3cztjtjtz}t t|}t |t|dt zdt tjtj| ddt tjtj| dz dt tjtj| dksJdS) Nrgz]ʂ>rQrrgLa㧝=rP?vIh%<=) rtanpirLrrrRminabsr,r)r+rGrUr/s r1test_tanzTestAAA.test_tanrs F25=( ) )  q ! !-((!"s(HHHHrvaggii001115AAAArvaggii#o6677EEEEvbfQZZ\\**++e333333r3cZtjddg}tjddg}t||d}t|||tt|dt|dt|d t|j |t|j |t|j d d gt|j ddgtjgd }tjgd }t||d}t|||tttj |tj d dgttj |tj ddgttj |tj ddgt|j |t|j |t|j gdt|j gddS)Nrr r[rrPrZg?gUUUUUU?g;f?)rr rc)r rrgȢ.fs!6!a%==A1H- -r3rrP)rrrstandard_normalrrrR)rGr-r.rUr/s r1 test_log_funczTestAAA.test_log_funcsi##$788    & &c.A.A%.H.H)H H . . . 11Q44LL!aadd------r3ctjdd}t|tj|}t |dtjdddS)Nr r g)\(?V瞯r rwrPrir y) rr rrM flatnonzeror_r)rr+rxryrz)rGrr/iis r1 test_residueszTestAAA.test_residuess Ks + + + 26!99q= ! ! ^BF17799--4 5 5 R(!%8888 Fem11!444 5 5 ^C R 011D8 9 9 R('>>>>>>r3zfunc,atol,rtolc6tj|dzdzS)NrZy{Gz?rr_rs r1zTestAAA.sC%)@)@r3g-a=gHz>c6tjdd|z z S)Nr g?)rr!rs r1rzTestAAA.s4!8 )=)=r3vIh%L=c6tjd|dzz S)Nr rcrrMrs r1rzTestAAA.sAqD ):):r3gG@ X=rc6tjd|dzzS)Nircrrs r1rzTestAAA.sQT ):):r3gvIh%l=c6tjdd|z z S)Ng333333?rrs r1rzTestAAA.sS1W )>)>r3+=cBddtjd|dzzzz S)Nr rrZrrs r1rzTestAAA.s"AbfS!c'].C.C,C)Dr3c0tj|dz S)Ngffffff?rrs r1rzTestAAA.sD)9)9r3ư>c tjd5|t}dddn #1swxYwYtt t |t t|||dS)Nr)dividerY)rrPTSrrrL)rGfuncrQrrUs r1test_basic_functionszTestAAA.test_basic_functionss[ ) ) )  S A               ?M44 +>+>??DD4 1 1 1 1 1 1s 377cd}tt|t}ttj||zddd}tt|t}t|ddtttj dtj ztz}ttj tj |dd dd }tt|t}t|d |d zd ddS) Nc0|dz|dzz|dz|dzzz S)Nr rcrss r1rUz,TestAAA.test_poles_zeros_residues..fs%aCAaC=QqSQqSM2 2r3r-q=rPc$dd|zz d|dz z zS)Nrcrrqy@rrss r1rUz,TestAAA.test_poles_zeros_residues..fsa!e9q!b&z) )r3rrg?c|dz |dzz S)Ny@@rcrrss r1rUz,TestAAA.test_poles_zeros_residues..fsL1q5) )r3ry) rrLrrsumr)r,r+prodr!r]rfr_r`s r1test_poles_zeros_residuesz!TestAAA.test_poles_zeros_residuessi 3 3 3  qq// 0 0qwwyy17799455sGGGG * * *  qq// 0 0 ))++Rd;;;;  rvbh}&<== > >qwwyy 1 12226%HHHH * * *  qq// 0 0 ! QWWYYq\1EFFFFFFr3rc*tj|SN)r zeros_likerss r1rzTestAAA.s a(8(8r3c|Srrrss r1rzTestAAA.sAr3c d|zS)Nrrrss r1rzTestAAA.s QSTUQUr3c|dz|zS)Nrcrrss r1rzTestAAA.s1qr3c|dz|zS)Nrrrss r1rzTestAAA.sAqD1Hr3cdd|zz S)Nr g?rrss r1rzTestAAA.s37 r3cddd|zzz S)Nr rrrss r1rzTestAAA.sq!bd(|r3cdd|z|dzzz S)Nr rrcrrss r1rzTestAAA.s1q51a4<(8r3cdd|dzzz S)Nr g)\(?rrrss r1rzTestAAA.sAtaQRd{Or3c ttt|tt|tddS)NrrP)rrrLr)rGrs r1 test_polynomials_and_reciprocalsz(TestAAA.test_polynomials_and_reciprocalssS ?M44 +>+>??DDS  / / / / / /r3ctjtjdddtjzzd}t |tjtj|zdz }t tjtj| ddgdd dS) NrZ.@r r rcr)r r rrg1E2>r) rrMr r]rr\rrfr_r)r{s r1 test_spiralzTestAAA.test_spirals F2;tS3ru9_$??? @ @ 26"%'!)$$ % %qwwyy 1 1222A26 4PPPPPPr3cVtjtjdddtjzzd}t jt 5t|tjtj|zdz ddd }dddn #1swxYwYtj tj | d k}t jt 5| d ksJ dddn #1swxYwYtj tj | d k|ksJt||tjtj|zdz d ddS)NrrZrr r rcr<F)rr?clean_uprr gAfc=gAfc=rY)rrMr r]rArJrKrr\rr_r+rr)rGr.r/ n_spuriouss r1test_spiral_cleanupzTestAAA.test_spiral_cleanups F2;tS3ru9_$??? @ @\. ) ) P PArvbeAgai((qBOOOA P P P P P P P P P P P P P P PVBF1::<<005899 \. ) ) % %::<<1$$$$$ % % % % % % % % % % % % % % %vbfQZZ\\**U233j@@@@!bfRU1WQY//e%HHHHHHs$7BBB6DD"%D"N)__name__ __module__ __qualname__rHrAmark thread_unsaferNrWrargrnrur|rr parametrizerrrrrrr3r1r9r9Rs ( ( ( [DDD * * *444***B000...HHH 999??? [-@@%N==udK::GQG::E1E>>qIDDeTR994F HII11II1 GGG& [V88++~~002D2D335K5K88:S:SUVV // VV / QQQ  [ I I I I Ir3r9c eZdZdZdZdZejddgdfdgdfd gd fd gd fd gdfgdZ ejde ddZ dZ dZ ejdgdejdddgdZdZdZdS)TestFloaterHormanncdd|dzzz S)Nr rcr)rGr.s r1rungezTestFloaterHormann.runge s!ad(|r3cVdtj||zzt|zS)Nr )rr6r)rGnds r1scalezTestFloaterHormann.scale s%billQ&')A,,66r3ctjtd5tdggdgddddn #1swxYwYtjtd5tdgdddddn #1swxYwYtjtd5tdgddgddggddddn #1swxYwYtjtd5ttjgdgddddn #1swxYwYtjtd 5tdgdgd dddn #1swxYwYtjtd 5tdgdgd dddn #1swxYwYtjt 5tdgdgd ddddS#1swxYwYdS) Nz`x`r;rrz`y` dimensionr r=z`d`r r)rArBrCrrrDrErFs r1test_ivzTestFloaterHormann.test_ivs ]:U 3 3 3 8 8 &uqcQ 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 ]:U 3 3 3 4 4 &sA 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 ]:[ 9 9 9 C C &saVaV,< B B B B C C C C C C C C C C C C C C C ]:X 6 6 6 ; ; &x! : : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ]:U 3 3 3 7 7 &sQC2 6 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 ]:U 3 3 3 7 7 &sQC2 6 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 ]9 % % 8 8 &sQC3 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s{>AA$BB B*CCC6D!!D%(D%E((E,/E,F//F36F3G55G9<G9z d,expectedr) r r r r r r r r r r r r ) r rcrcrcrcrcrcrcrcrcr rc) r rrrrrrrrrr r) r rrrrrrrr r) r rq rrrrrqr ctjd}t|d|z|}t|j||j|z|dddS)NrrrrwrrQ)rr6rrr(ravelrr)rGrexpectedrr/s r1test_uniform_gridz$TestFloaterHormann.test_uniform_grid!st IbMM &q#a%1 5 5 5 ))$**QVQ*?*??" 0 0 0 0 0 0r3rrctjddd}tjd}|ddd}||}|d|dz }t |||}d||dzzz}t||||d | t||||dS) Nrr 3lM +RYr rrrg|=rY) rr rrr"rrrr) rGrrr-xxyhr/tols r1 test_rungezTestFloaterHormann.test_runge/s K1b ! !i##$677 [[AD[ ) ) JJqMM aD1Q4K &q!q 1 1 1QqSk"tzz"~~EDDDDQQqTT4::a==)))))r3c tjdd}||dzz}t|tj|d}tjddd}||dzz}t ||tj|d dS) Nr r r rr r rr)rr rr!r)rGrr.r/rzzs r1 test_complexzTestFloaterHormann.test_complex>s KA   "H &q"&))r : : : [QD ) ) ) "R%Z"rvbzz666666r3c$tjddd}tjddd}tj|}t|||jdz }t ||}t ||||dddS)Nrr rirrr)rr r!rrr r)rGrrrr/ps r1test_polyinterpz"TestFloaterHormann.test_polyinterpFs K1b ! ! [At $ $ F1II &q!qvax 8 8 8 #Aq ) )"qquu5u======r3y_shape))rc)rcrr )r rqrxx_shaper)rrc tjdd}tjtjtj|t t dt|dz|j|z}t||}tj d}| |}tjtjtj|t t |j t||j z|j|z}||} | j|j|zksJt| |ddS)Nrr lwcBMYrr)rr broadcast_to expand_dimsr!tuplerangelenshaperrrndimr) rGrrrrr/r-ryyrrs r1test_trailing_dimz$TestFloaterHormann.test_trailing_dimOs3 K1   O N26!99eE!S\\A5E,F,F&G&G H H Gg    'q! , ,i##$BCC ZZ ! ! _ N26"::uU27CLL27) rr rr!r]r_subtractouterr,r6r^r)rGrr/errs r1 test_zeroszTestFloaterHormann.test_zerosds K23 ' ' ' &q"&q// : :fR[&&qwwyy")B--@@AAEE1EMM#t$$$$$r3c"tjdd}t|d|dzz }|}|jdk|jdkztj|jdkz}tj|dksJdS)Nr r rcrr)rr rr)r'r_imagr)rGrr/rmasks r1 test_no_polesz TestFloaterHormann.test_no_polesks KA   &q!AqD& 1 1 GGII" 1-&1HIvd||q      r3N)rrrrrrrArrrrrrrrrrrr3r1rr s777888" [\ - - -. - - -. - - -. - - -. 4 4 45 ,000 [S%%)),, * *-, *777>>> [Y(G(G(GHH [Z3):;;++<;IH+&%%%!!!!!r3r)#mathrnumpyr numpy.testingrrrrArxscipy.interpolaterrr rfloat64rrRr rLlogspacer concatenaterrfloat32 complex64rjr2int16int32int64r7r9rrr3r1rs4JJJJJJJJJJ VVVVVVVVVV HBHRZ $$ Bt,,, bk#qrs+++bns44R4yj1#s+,,C)C#DEE2:rz2<"WXX88YXFE8.C)C#DEE28RXrx"@AA 4 4BAFE 4tItItItItItItItIng!g!g!g!g!g!g!g!g!g!r3