o ֭jX<@s*ddlmZddlZddlmZmZmZddlZddl Z ddl m Z m Z m Z deejjZejdddd Zejd dd d d Zeeddd dgegZejde e gejdejejejejgddZejde e gejdejejejgddZ GdddZ!GdddZ"dS)) factorialN)assert_allclose assert_equalassert_array_less)AAAFloaterHormannInterpolatorBarycentricInterpolatorg@numi i)baser methoddtypecCs.t|jdd}|tur|d9}tjd}tjdd|d}||t|}|jdddd |}t ||t||d||j |ksHJ|t urf|j j |ksTJ|jj |ks\J|jj |jj ksfJ|jj |ksnJ|j t|d ks{J|j t|d ksJ|j t|d ksJdS) Ng?dld wr r rsizertol?)npfinfoepsrrandom default_rnglinspacesinuniformastyperrrsupport_pointssupport_valueserrorsrealweightspoles result_typeresiduesroots)rrrrngzrz2r/g/var/www/html/CropPilot/venv/lib/python3.10/site-packages/scipy/interpolate/tests/test_bary_rational.pytest_dtype_preservation(s" r1cCstjd|d}|||}|jjt|dksJ|tur@|jjt|dks(J|jjt|dks4J|jjt|dks@J| jt|dksMJ| jt|dksZJ| jt|dksgJ||jt|dkstJdS)Nrr?r) raranger&rr(rr"r#r$r'r)r*)rrr,r-r/r/r0test_integer_promotionAs r4c @s:eZdZddZejjddZddZddZ d d Z d d Z d dZ ddZ ddZddZejdddddfddddfddddfddd dfd!dd"dfd#dddfd$dd%dfgd&d'Zd(d)Zejd*d+dd,dd-dd.dd/dd0dd1dd2dd3dg d4d5Zd6d7Zejjd8d9Zd:S);TestAAAcCsFtjtddtdgddgWdn1swYtjtddtdgdggdgdggWdn1s>wYtjtddttjgdgWdn1s]wYtttdgdgddWdn1s{wYtjtd dtdgdgd dWddS1swYdS) Nz same sizematchrr z1-Dfiniter2 max_termsgreaterr )pytestraises ValueErrorrrinf TypeErrorselfr/r/r0test_input_validationSs "zTestAAA.test_input_validationcCsHtjtddttttddWddS1swYdS)Nz AAA failedr6r r9)r<warnsRuntimeWarningr UNIT_INTERVALrexprAr/r/r0test_convergence_error_s"zTestAAA.test_convergence_errorcCsvtt}tt|}t|t|tdt|tjtjt|tj s&J|j j }tt|dd}|j j |ks9JdS)NatolgMbP?r) rrGrFrrTOLrnanisfiniter?r"r)rBfr-m1r/r/r0test_expfs  zTestAAA.test_expcCsttjt}tt|}t|t|dtddttt| dddttt| ddtdtt| dksGJdS) Nrgz]ʂ>rJrrgLa㧝=rI?vIh%<=) rtanpirFrrrKminabsr*r'r)rBrNr-r/r/r0test_tanrs  " zTestAAA.test_tancCsptddg}tddg}t||dd}t|||tdt|dt|dt|d t|j |t|j |t|j d d gt|j ddgtgd }tgd }t||dd}t|||tdtt |t d dgtt |t ddgtt |t ddgt|j |t|j |t|j gdt|j gddS)Nrr rSrrIrRg?gUUUUUU?g;f?)rr rZ)r rrgȢ.frrI)rrrstandard_normalrrrK)rBr+r,rNr-r/r/r0 test_log_funcs zTestAAA.test_log_funccCs<tdd}t|tj|}t|dtjddddS)Nr r g)\(?V瞯r rkrIr_r y) rrrrG flatnonzerorWr'rr)rlrmrn)rBrur-iir/r/r0 test_residuesszTestAAA.test_residueszfunc,atol,rtolcCst|ddS)NrRy{Gz?rrWrur/r/r0zTestAAA.g-a=gHz>cCtdd|S)Nr g?)rrr|r/r/r0r}r~vIh%L=cCstd|dS)Nr rZrrGr|r/r/r0r}r~gG@ X=rcCstd|dS)NirZrr|r/r/r0r}r~gvIh%l=cCr)Ng333333?rr|r/r/r0r}r~+=cCsddtd|dS)Nr rrRrr|r/r/r0r}scCst|dS)Ngffffff?r{r|r/r/r0r}sư>cCsVtjdd |t}Wdn1swYttt|tt|||ddS)Nrr)dividerQ)rrtPTSrrrF)rBfuncrJrrNr/r/r0test_basic_functionss   zTestAAA.test_basic_functionscCsdd}tt|t}tt||ddddd}tt|t}t|dddttt dtj t}tt t |d d ddd d}tt|t}t|d |d d dddS)NcSs |d|d|d|dS)Nr rZr/rgr/r/r0rNs z,TestAAA.test_poles_zeros_residues..fr-q=rIcSsdd|d|dS)NrZrrey@r/rgr/r/r0rNsrrwg?cSs|d|dS)Ny@@rZr/rgr/r/r0rNry) rrFrrsumr'r*r)prodrrUr]rWrXr/r/r0test_poles_zeros_residuess "&z!TestAAA.test_poles_zeros_residuesrcCs t|SN)r zeros_likergr/r/r0r}s cCs|Srr/rgr/r/r0r}scCsd|S)Nrr/rgr/r/r0r}scC |d|S)NrZr/rgr/r/r0r} cCr)Nrr/rgr/r/r0r}rcCs dd|S)Nr g?r/rgr/r/r0r}rcCsddd|S)Nr rr/rgr/r/r0r}cCsdd||dS)Nr rrZr/rgr/r/r0r}scCsdd|dS)Nr g)\(?rr/rgr/r/r0r}rcCs$ttt|tt|tdddS)NrrI)rrrFr)rBrr/r/r0 test_polynomials_and_reciprocalss z(TestAAA.test_polynomials_and_reciprocalscCshttjdddtjdd}t|ttj|d}ttt| ddgdd d dS) NrR.@r r rZr)r r rrg1E2>r) rrGrrUrrTrr]rWr'ror/r/r0 test_spirals .zTestAAA.test_spiralcCsttjdddtjdd}ttt|ttj|dddd d }Wdn1s1wYt t | d k}tt| d ksPJWdn1sZwYt t | d k|ksoJt ||ttj|dd dddS)NrrRrr r rZr<F)rr:clean_uprr gAfc=gAfc=rQ)rrGrrUr<rDrErrTrrWr)rr)rBr,r- n_spuriousr/r/r0test_spiral_cleanups  $  (zTestAAA.test_spiral_cleanupN)__name__ __module__ __qualname__rCr<mark thread_unsaferHrPrYr^rdrjrprvrz parametrizerrrrrr/r/r/r0r5RsF   !              r5c @seZdZddZddZddZejddgd fd gd fd gd fdgdfdgdfgddZ ejde dddZ ddZ ddZ ejdgdejddd gd!d"Zd#d$Zd%d&Zd'S)(TestFloaterHormanncCsdd|dS)Nr rZr/)rBr,r/r/r0runge rzTestFloaterHormann.rungecCsdt||t|S)Nr )rr3r)rBndr/r/r0scale rhzTestFloaterHormann.scalecCstjtddtdggdgddWdn1swYtjtddtdgdddWdn1s;wYtjtddtdgddgddggddWdn1s`wYtjtddttjgdgddWdn1swYtjtd dtdgdgd dWdn1swYtjtd dtdgdgd dWdn1swYtttdgdgd dWddS1swYdS) Nz`x`r6rrz`y` dimensionr r8z`d`r r)r<r=r>rrr?r@rAr/r/r0test_ivs* "zTestFloaterHormann.test_ivz d,expectedr) r r r r r r r r r r r r ) r rZrZrZrZrZrZrZrZrZr rZ) r rrrrrrrrrr r) r rrrrrrrr r) r re rrrrrer cCsDtd}t|d||d}t|j||j||ddddS)NrrrrkrrJ)rr3rrr&ravelrr)rBrexpectedrur-r/r/r0test_uniform_grid!s  z$TestFloaterHormann.test_uniform_gridrrc Cstddd}tjd}|jdddd}||}|d|d}t|||d}d||d}t||||d |d t||||dS) Nrr 3lM +RYr rrrg|=rQ) rrrrr rrrr) rBrrur+xxyhr-tolr/r/r0 test_runge/s  zTestFloaterHormann.test_rungecCsdtdd}||d}t|t|dd}tjdddd}||d}t||t|dd dS) Nr r r rr r rr)rrrrr)rBrur,r-rzzr/r/r0 test_complex>s   zTestFloaterHormann.test_complexcCs`tddd}tddd}t|}t|||jdd}t||}t||||ddddS)Nrr rirrr)rrrrrrr)rBrurrr-pr/r/r0test_polyinterpFs   z"TestFloaterHormann.test_polyinterpy_shape))rZ)rZrr )r rerxx_shaper)rrc Cstdd}ttt|ttdt|d|j|}t ||}tj d}| |}ttt|tt|j t||j |j|}||} | j|j|ksXJt | |dddS)Nrr lwcBMYrr)rr broadcast_to expand_dimsrtuplerangelenshaperrrndimr) rBrrrurr-r+ryyrrr/r/r0test_trailing_dimOs "   &z$TestFloaterHormann.test_trailing_dimcCsZtjdddd}t|ttj|}ttj|t dj dd}t |ddS)Nrrrr r)axisgh㈵>) rrrrrUrWsubtractouterr*r3rVr)rBrur-errr/r/r0 test_zerosds&zTestFloaterHormann.test_zeroscCs`tdd}t|d|d}|}|jdk|jdk@t|jdk@}t|dks.JdS)Nr r rZrr)rrrr'r%rWimagr)rBrur-rmaskr/r/r0 test_no_polesks $z TestFloaterHormann.test_no_polesN)rrrrrrr<rrrrrrrrrrr/r/r/r0r s*          r)#mathrnumpyr numpy.testingrrrr<rlscipy.interpolaterrrrfloat64rrKrrFlogspacer concatenaterrfloat32 complex64r`r1int16int32int64r4r5rr/r/r/r0s&   8