o iX<@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)rrrrngzrZz2r.d/home/kim/smarthome/.venv/lib/python3.10/site-packages/scipy/interpolate/tests/test_bary_rational.pytest_dtype_preservation(s" r0cCstjd|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.r/test_integer_promotionAs r3c @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-Dfiniter1 max_termsZgreaterr )pytestraises ValueErrorrrinf TypeErrorselfr.r.r/test_input_validationSs "zTestAAA.test_input_validationcCsHtjtddttttddWddS1swYdS)Nz AAA failedr5r r8)r:warnsRuntimeWarningr UNIT_INTERVALrexpr?r.r.r/test_convergence_error_s"zTestAAA.test_convergence_errorcCsvtt}tt|}t|t|tdt|tjtjt|tj s&J|j j }tt|dd}|j j |ks9JdS)NatolgMbP?r) rrErDrrTOLrnanisfiniter=r"r)r@fr-m1r.r.r/test_expfs  zTestAAA.test_expcCsttjt}tt|}t|t|dtddttt| dddttt| ddtdtt| dksGJdS) Nrgz]ʂ>rHrrgLa㧝=rG?vIh%<=) rtanpirDrrrIminabsr*r'r)r@rLr-r.r.r/test_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 rQrrGrPg?gUUUUUU?g;f?)rr rX)r rrgȢ.frrG)rrrZstandard_normalrrrI)r@r+r,rLr-r.r.r/ test_log_funcs zTestAAA.test_log_funccCs<tdd}t|tj|}t|dtjddddS)Nr r g)\(?V瞯r rgrGr]r y) rrrrEZ flatnonzerorUr'rr)rhrirj)r@rqr-iir.r.r/ test_residuesszTestAAA.test_residueszfunc,atol,rtolcCst|ddS)NrPy{Gz?rrUrqr.r.r/zTestAAA.g-a=gHz>cCtdd|S)Nr g?)rrrwr.r.r/rxryvIh%L=cCstd|dS)Nr rXrrErwr.r.r/rxrygG@ X=rcCstd|dS)NirXr|rwr.r.r/rxrygvIh%l=cCrz)Ng333333?r|rwr.r.r/rxry+=cCsddtd|dS)Nr rrPr|rwr.r.r/rxscCst|dS)Ngffffff?rvrwr.r.r/rxsư>cCsVtjdd |t}Wdn1swYttt|tt|||ddS)Nrn)dividerO)rrpPTSrrrD)r@funcrHrrLr.r.r/test_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 rXr.rdr.r.r/rLs z,TestAAA.test_poles_zeros_residues..fr}-q=rGcSsdd|d|dS)NrXrrby@r.rdr.r.r/rLsrrsg?cSs|d|dS)Ny@@rXr.rdr.r.r/rLry) rrDrrsumr'r*r)prodrrSr[rUrVr.r.r/test_poles_zeros_residuess "&z!TestAAA.test_poles_zeros_residuesrcCs t|SN)rZ zeros_likerdr.r.r/rxs cCs|Srr.rdr.r.r/rxscCsd|S)Nrr.rdr.r.r/rxscC |d|S)NrXr.rdr.r.r/rx cCr)Nrr.rdr.r.r/rxrcCs dd|S)Nr g?r.rdr.r.r/rxrcCsddd|S)Nr rr.rdr.r.r/rxcCsdd||dS)Nr rrXr.rdr.r.r/rxscCsdd|dS)Nr g)\(?rr.rdr.r.r/rxrcCs$ttt|tt|tdddS)Nr{rG)rrrDr)r@rr.r.r/ test_polynomials_and_reciprocalss z(TestAAA.test_polynomials_and_reciprocalscCshttjdddtjdd}t|ttj|d}ttt| ddgdd d dS) NrP.@r r rXr)r r rrg1E2>r) rrErrSrrRrr[rUr'rkr.r.r/ test_spirals .zTestAAA.test_spiralcCsttjdddtjdd}ttt|ttj|dddd d }Wdn1s1wYt t | d k}tt| d ksPJWdn1sZwYt t | d k|ksoJt ||ttj|dd dddS)NrrPrr r rXr<F)rr9clean_upr~r gAfc=gAfc=rO)rrErrSr:rBrCrrRrrUr)rr)r@r,r-Z n_spuriousr.r.r/test_spiral_cleanups  $  (zTestAAA.test_spiral_cleanupN)__name__ __module__ __qualname__rAr:markZ thread_unsaferFrNrWr\rarfrlrrru parametrizerrrrrr.r.r.r/r4RsF   !              r4c @seZdZddZddZddZejddgd fd gd fd gd fdgdfdgdfgddZ ejde dddZ ddZ ddZ ejdgdejddd gd!d"Zd#d$Zd%d&Zd'S)(TestFloaterHormanncCsdd|dS)Nr rXr.)r@r,r.r.r/runge rzTestFloaterHormann.rungecCsdt||t|S)Nr )rr2r)r@ndr.r.r/scale rezTestFloaterHormann.scalecCstjtddtdggdgddWdn1swYtjtddtdgdddWdn1s;wYtjtddtdgddgddggddWdn1s`wYtjtddttjgdgddWdn1swYtjtd dtdgdgd dWdn1swYtjtd dtdgdgd dWdn1swYtttdgdgd dWddS1swYdS) Nz`x`r5rrz`y` dimensionr r7z`d`r r)r:r;r<rrr=r>r?r.r.r/test_ivs* "zTestFloaterHormann.test_ivz d,expectedr) r r r r r r r r r r r r ) r rXrXrXrXrXrXrXrXrXr rX) r rrrrrrrrrr r) r rrrrrrrr r) r rb rrrrrbr cCsDtd}t|d||d}t|j||j||ddddS)NrrrrgrrH)rr2rrr&Zravelrr)r@rexpectedrqr-r.r.r/test_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|=rO) rrrrr rrrr) r@rrqr+xxyhr-Ztolr.r.r/ test_runge/s  zTestFloaterHormann.test_rungecCsdtdd}||d}t|t|dd}tjdddd}||d}t||t|dd dS) Nr r r rr r rr)rrrrr)r@rqr,r-rzzr.r.r/ test_complex>s   zTestFloaterHormann.test_complexcCs`tddd}tddd}t|}t|||jdd}t||}t||||ddddS)Nrr rirrr)rrrrrrr)r@rqrrr-pr.r.r/test_polyinterpFs   z"TestFloaterHormann.test_polyinterpy_shape))rX)rXrr )r rbrxx_shaper)rrc Cstdd}ttt|ttdt|d|j|}t ||}tj d}| |}ttt|tt|j t||j |j|}||} | j|j|ksXJt | |dddS)Nrr lwcBMYrr)rrZ broadcast_toZ expand_dimsrtuplerangelenshaperrrndimr) r@rrrqrr-r+ryyZrrr.r.r/test_trailing_dimOs "   &z$TestFloaterHormann.test_trailing_dimcCsZtjdddd}t|ttj|}ttj|t dj dd}t |ddS)Nrrrr r)Zaxisgh㈵>) rrrrrSrUsubtractouterr*r2rTr)r@rqr-errr.r.r/ test_zerosds&zTestFloaterHormann.test_zeroscCs`tdd}t|d|d}|}|jdk|jdk@t|jdk@}t|dks.JdS)Nr r rXrr)rrrr'r%rUimagr)r@rqr-rmaskr.r.r/ test_no_polesks $z TestFloaterHormann.test_no_polesN)rrrrrrr:rrrrrrrrrrr.r.r.r/r s*          r)#mathrnumpyrZ numpy.testingrrrr:rhZscipy.interpolaterrrrZfloat64rrIrrDZlogspacerZ concatenaterrZfloat32Z complex64r^r0Zint16Zint32Zint64r3r4rr.r.r.r/s&   8