:j\hddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z mZmZmZmZmZmZddlmcmZddlmZddlZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z/ddl0m1cm.Z2ddl3m4Z4m5Z5m6Z6m7Z7m8Z8ddl9m:Z:m;Z;ddlmZ>ddl?m@cmAZBddlCmDZDdd lEmFZFdd lGmHZHdd lmIZJdd lmKZLdd lmMZMejjZOejjZPeeGddZQeedfGddZRdZSdZTdZUdZVdZWe fdZXdSdZYdde fdZZdZ[GddZ\eeGd d!Z]d"Z^dTd#Z_ejjd$d%d&gZaee Gd'd(ZbGd)d*ZcdUd+Zdd,Zed-ZfGd.d/Zgd0Zhee+Gd1d2Zid3Zjd4Zkd5ZlGd6d7Zmee*Gd8d9ZnGd:d;ZoGd<d=Zpd>Zqd?Zree<Gd@dAZsdBZtGdCdDZuGdEdFZvee=GdGdHevZwee=GdIdJevZxee>GdKdLZyee>GdMdNZzGdOdPZ{GdQdRZ|y)VN)xp_assert_equalxp_assert_closexp_default_dtype concat_1dmake_xp_test_casexp_ravel_xp_copy_to_numpyarray_namespace)raises)BSplineBPolyPPolymake_interp_splinemake_lsq_splinesplevsplrepsplprepsplder splantidersprootsplintinsert CubicSpline NdBSplinemake_smoothing_splineRegularGridInterpolator) _not_a_knot_augknt_woodbury_algorithm_periodic_knots_make_interp_per_full_matr) Fperiodic root_rati)generate_knots make_splrep make_splprep) AxisError)_run_concurrent_barrier) make_ndbspl)_fitpack) _bsplines)_dierckxcZeZdZdZdZdZdZdZdZdZ e dd e d d e d d dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZe dd dZd Zd!Zd"Zd#Zd$Zd%Zd&Z e!dd' d(Z"d)Z#d*Z$d+Z%e&jNjQd,e)d-d.d/Z*d0Z+d1Z,e&jNjQd2gd3e&jNjQd4e)d5d6Z-d7Z.d8Z/d9Z0e&jNjQd:gd;d<Z1e&jNjQd:gd;d=Z2d>Z3d?Z4d@Z5e&jNjme7jpdAk(dBe9CdDZ:yE)F TestBSplinec tttftfit ddgdgdt j d5tttfit dt jgdgddddtttfit dt jgdgdtttfit ddgdgdtttfit dgdggdgdtttfit gd dgdtttfit gd ddgd tttfit gd gd dtttfit gd gd dtttfit gdgdd d\}}|j||zdz|j}|jt jj|}t|||}t||jt||j||j k(sJy#1swYxYw)N??rtckignore)invalidrr0rr0r;r;)r?r2@@@)r2r2r2cubic@)r?rr0r0r;r=)r2r0r0 r=dtype) assert_raises TypeError ValueErrorr dictnperrstatenaninfarangefloat64asarrayrandomrr4r5r6)selfxpnr6r4r5bs z/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/scipy/interpolate/tests/test_bsplines.py test_ctorzTestBSpline.test_ctor5sy*-w 1!SbTQ/ 1 [[ * S *g RBFF tq1Q R Sj'NTQKB41-MNj'JTQGtq-IJj'MTaS1#J2$!-LM j'KTI!-JKj' <RH: < i O1\WM Oi K1\SI K j' B, a@ B1 IIac!e2::I . JJryy''* + Aq! 133133ACCxx; S Ss 3IIc2t}|j}t|j|dddt|j|ddd|j |dk(sJt jt5d|_dddy#1swYyxYw)NrV瞯<atolrtolr0r;foo) _make_random_splinetckrr4r5r6pytestr AttributeError)rUrXrbs rYtest_tckzTestBSpline.test_tckXs}  !eeSV%e<SV%e<ssc!f}}]]> * AE   s <B  Bct|jgd|jddgd}|jddd}|j|dk|jd|j d }t |||y) Nrr2r;rAr>rr3r; r2rGrB)r rSlinspacewhererRr)rUrVrXxxexpecteds rYtest_call_namespacezTestBSpline.test_call_namespacedso bjj, B70Cq I [[Ar "88BGRZZ"**Z%EsK"x(ct|jgd|jddgd}t|j|usJt|j|usJy)NrgrAr>rr3)r rSr r4r5rUrVrXs rY&test_attributes_have_correct_namespacez2TestBSpline.test_attributes_have_correct_namespacemsT bjj, B70Cq Iqss#r)))qss#r)))rnctjddd}tddgdgd}t||tj|dztgdddgd}t||tj |d kdd y) Nrr0rhrAr3)rffffff?r0r=r>rsrB)rMrir r ones_likerj)rUrkrXs rY test_degree_0zTestBSpline.test_degree_0rsu [[Ar " q!f *"r||B/#56 lq!f 2"rxxT 3<=rnc |jgd}|jgd}d}t|||}|jddd|j}t |||dt || z|dt |dz | zz|d t |d z | zzd t t|||f\}}} t||| |f} t |||j| d y) Nr<)r2r;r=r0r2rA2rGrrVr;+=r^) rSr rirRrB_012mapr r) rUrVr4r5r6rXxx_npt_npc_np splev_results rY test_degree_1zTestBSpline.test_degree_1{s JJ ' JJ{ #  Aq!  KKS"BJJK 7 aD aDqR 1Q4acb(9#9 9AaDqsrARc t}|j\}}}tj|||| dz d}t ||t ||||fdy)Nr0rwryrz)rarbrMrirrrUrXr4r5r6rks rYtest_rndm_splevzTestBSpline.test_rndm_splevsV  !%%1a [[1q!Aw +"uR!Q3%@rnctjjd}tj|jd}|jd}t ||}t |}|j |j}}tj|||| dz d}t||t||dy)Nr0Pryrz) rMrT RandomStatesortrr r4r6rirr) rUrngr}yrbrXr4r6rks rYtest_rndm_splrepzTestBSpline.test_rndm_splrepsii##D) GGCJJrN # JJrNQl SMssACC1 [[1q!Aw +"uR~E:rnc@t|}|j|j|_|j|j|j |j|j dz d|j }t|||j|y)Nrxr0drG)rartr5rir4r6rRr)rUrVrXrks rYtest_rndm_unityzTestBSpline.test_rndm_unityso 2 &ll133 [[QSS133tAv;2::[ F"r||B/0rnctjjd}d\}}tj|j|}|j|ddf}t |j ||f\}}t |||}|||| dz } }|| |z |j |jdzz} || jdk(sJy) Nrr=sizer0r=r>)r=r>rrr)rMrTrrr|rSr shape) rUrVrrWr6r4r5rXtmtprks rYtest_vectorizationzTestBSpline.test_vectorizationsii##D)1 GGCJJqM " JJQ1IJ &2::1v&1 Aq! 1q!AwB 27bjjI)>?? ?u{{o---rnc Xtjjd}d\}}tj|j||zdz}|j|}tj||j|dzf}t |||t |||}}|d|dz } tj |d| z |d| zd} t|| || dt|| t| |||fdt|| t| |||fdy) Nr)!r=r0r9rrwryrz) rMrTrrr_r rirr) rUrrWr6r4r5c_padrXb_paddtrks rY test_len_czTestBSpline.test_len_cs ii##D)1 GGCJJqs1u% & JJqMaAaC()1a#WQq%95 rUQqT\ [[1AbEBJ 3"uRyu5"uR!Q3%@"uR!UA7eDrnc t}|j\}}}|||| dz }}dD])}t|||g|||dz|dz g|dd+y)Nr0TF绽|=& .>Hz>r]rarbr) rUnum_parallel_threadsrXr4_r6rrextraps rYtest_endpointszTestBSpline.test_endpointsss  !%%1a1q!AwB# WF Ar2h/rEz2:6?dQU W Wrnc t}|j\}}}t|||dz| dz dz |||dz| dz dzdy)Nr0rrrzr)rUrrXr4rr6s rYtest_continuityzTestBSpline.test_continuitys\  !%%1a!AaC1+-.!AaC1+2E0F rnct|}|j\}}}|d|dz }|j|||z || dz |zd}|||k||| dz kz}t|||d|||dt t |||f\} } } |j t| | | |fd } t||d| y) Nrxr9rr0rwTrF)ext)rarbrirr|r rSr) rUrVrXr4r5r6rrkmaskxx_nprrrs rY test_extrapzTestBSpline.test_extraps 2 &%%1a rUQqT\ [[1AqbdGbL" 5!r b1aRT7l+ "T(5"T(6 8 1B1:>tTzz%dAA"FG "$/>rnct}|j\}}}|ddz |ddzg}||}tjtj|rJy)Nrr0r9)rarbrMallisnan)rUrXr4rr6rkyys rYtest_default_extrapzTestBSpline.test_default_extrapsY  !%%1adQh" " rU66"((2,''''rnc Rtjjd}tj|jd}|jd}t |j ||f\}}d}t |||d}|jd|dzz }|d |dz }|j|||z |||zd } ||| ||z ||||z zz} t t| ||f\} } } t||jk(rd nd }t|| |j t| | | |f| |j gd} ||| ||z ||||z zz} t|| d|| dt||jk(rd yd y)Nrr>r=periodicrrr0r9rw-q=gH׊>rz)r9r?r0TrygƠ>)rMrTrrr|rSr rrir rrRrr)rUrVrr4r5r6rXrWrrkxyxy_nprrr^s rYtest_periodic_extrapz TestBSpline.test_periodic_extrap sii##D) GGCJJqM " JJqM2::1v&1  Aq! 4 GGAJ!a%  rUQqT\ [[1AaD2Ir 2 qTR!A$Y1Q4!A$;/ / 1B1:>tT(, :u bE2::eED$?;<4 ZZ ( qTR!A$Y1Q4!A$;/ / bj ) bd #*2."**< CG rnct}|j\}}}tj|||f}t j |||| d}t ||||ddy)Nrryr])rarbr from_splinerMrir)rUrXr4r5r6pprks rY test_ppolyzTestBSpline.test_ppoly'sa  !%%1a   1ay ) [[1q!uc *"r"vE>rncjt}|j\}}}tj|d|dd}tj||f}t d|dzD])}t ||||f|}t||||d+t|||dztj|dy) Nrr9rwr0dernuryrz) rarbrMrirrangerr zeros_like)rUrXr4r5r6rkryds rYtest_derivative_rndmz TestBSpline.test_derivative_rndm/s  !%%1a [[1qub ) UU2q5\AaC= ;CrAq!9#.B B" E : ; "1 r}}R'8uErncd}gd}tjjd}tjdd|jdddf}t |||}tj gd}t |||dk7dz |||dk7dztj|d |d rJtj d d g}t ||dz d ||dzd tj d dg}tj||dz d ||dzd rJtj||dz d||dzdrJy)Nr;) r9r9rr0r0r=r>rrrrrrrr)r0r=r>rrrg2H@gη@r=r>r0r)rMrTrrr rSrallclose) rUr6r4rr5rXr}x0x1s rYtest_derivative_jumpsz!TestBSpline.test_derivative_jumps<sP  2ii##D) EE!Q 1 q!+ , Aq!  JJ| $!AF)e+,!AF)e+, .;;q{AgJ777ZZA "u*+"u*+ - ZZA ;;qe2Ab5jQ4GHHH;;qUq11QY13EFFFFrncupyzCuPy doesn't raisec tttjfit |j dgtttjfit |j gy)Nrr4)rIrKr basis_elementrLrSrUrVs rY$test_basis_element_invalid_too_shortz0TestBSpline.test_basis_element_invalid_too_shortVsJ j'"7"7S4"**aS/;RSj'"7"7R4"**R.;QRrnc |jddd}tj|jgd}t t ||j |jf\}}}|jt||||jf}t|||dt||jk(rdnd}t|||jt||j |tj|jgd }|jd d d |j }t|||j|dk||zd|z d zdy)Nr9r>rrr0r;r=rryrzrrGrr0r0r;rr;rhr0r@)rir rrSr|r r4r5rr6rrrRB_0123rj) rUrVrkrXrrrrr^s rYtest_basis_element_quadraticz(TestBSpline.test_basis_element_quadratic\s [[Q #  ! !BJJ|$< = 1BQSS>BtTzz%dACC/@"AB "|%8&r*bjj8Ud"rzz&*BJJzGdS  ! !BJJ|$< = [[Ar[ 4"aBB 3% Arnc t}|j\}}}tj|||| dz d}t ||t ||||dy)Nr0rryrz)rarbrMrir_sum_basis_elementsrs rYtest_basis_element_rndmz#TestBSpline.test_basis_element_rndmlsU  !%%1a [[1q!Aw +"22q!Q?eLrnct}|j\}}}|dz}t|||}t||jj|}t||jj |}t j|||| dz d}t||j||dt||j ||dy)Ny?@r0rryrz) rarbr r5realimagrMrir) rUrXr4r5r6ccb_reb_imrks rY test_cmplxzTestBSpline.test_cmplxrs  !%%1a (^ Ar1 q!##((A&q!##((A& [[1q!Aw +" DH59" DH59rnctj|jgd}|j||jsJy)Nr)r rrSrrOrps rYtest_nanzTestBSpline.test_nans5  ! !"**\": ;xx"&& """rnc td|}|j\}}}t|||}|j|||| dz d}t d|D].}|j }t |||||dd0y)Nr)r6rVr0rrr])rarbr rir derivativer) rUrVrXr4r5r6b0rkjs rYtest_derivative_methodz"TestBSpline.test_derivative_methods ! +%%1a Q1  [[1q!Aw +q! FA A Br1Iqu5u E Frnct|}|j\}}}|j|||| dz d}t|j j |||dd|j |||fd}|j ||fd}t|||}t|j j |||ddy)Nrxr0rryr]axisr;)rarbrirantiderivativerrr )rUrVrXr4r5r6rks rYtest_antiderivative_methodz&TestBSpline.test_antiderivative_methods 2 &%%1a [[1q!Aw +7((*557;"E 7 HHaAYQH ' HHaV!H $ Aq! 7((*557;"E 7rnc Vtj|jgd}tj|j ddddsJtj|j ddddsJtj|j ddddsJtj|j ddddsJtj|j ddddsJtj|j ddddsJtj|j ddd dsJtj|j ddd d dsJtj|j ddd ddsJtj|j ddd ddsJtj|j ddd t jddtt|jdsJy) Nr:rr0rryabs_tolr9r?TrF) r rrSmathisclose integrate_implrr|r rbrps rY test_integralzTestBSpline.test_integrals  ! !"**Y"7 8||AKK1-sEBBB||AKK1-xGGG||AKK1-tUCCC||AKK1-sEBBB||AKK1-xGGG||AKK1-tUCCC||AKKA.UCCC||AKKA4K@#uUUU||AKKA5KA3PUVVV||AKK25KA4QVWWW||AKK25KA!LLB4Equu0MN$)+ ++rnz(CuPy periodic extrapolation seems brokenc tj|jgd}d|_|j }|j|d|dz |j }t j|jdd|sJt j|jddd|zsJt j|jdd|sJt j|jd d d|zsJt|jd d |j|d |d z t|jd d |j|d|dz |dz|d z t|jdd|j|d|dz |dz|d z t|jd d|j|d|dz |dz|d z d|zzt|jdd|j|d|dz t|jdd|j|d|dz t|jdd|j|d|dz d|zz y)Nr:rr;rrGr9iiir?r=r0g+@rir>) r rrSrrrRrrrr)rUrVrXi period_ints rYtest_integral_periodicz"TestBSpline.test_integral_periodics[  ! !"**Y"7 8"   ZZ!qt 2::Z> ||AKK1-z:::||AKK1-rJ???||AKKB/<<<||AKKB/Z@@@ C- 1S6AcF?3 5 C+ 1Q4!A$;1#5##>? A Hf5 1Q4!A$;1#5##>? A C0 1Q4!A$;1#5##>Z#OP R  Ar*BJJqtad{,CD B,bjj1!.EF Ar* 1Q4!A$;Z#?@ Brncgd}t||}d|_tj|}dD]1\}}t |j |||j ||3y)Nr<r))r)rr)r )rrrrrr)rUr}rXprrs rYtest_integrate_ppolyz TestBSpline.test_integrate_ppolys`  q! $"   a 5 1FB AKKB/KKB/ 1 1rnctjgd}dD]C}|jdd|}t|tj sJ|j dk(rCJy)Nr:rrr0r)r rr isinstancerMndarrayndim)rUrXrress rYtest_integrate_0D_alwaysz$TestBSpline.test_integrate_0D_alwayssY  ! !) ,( !K++a +.BN)__name__ __module__ __qualname__rnrYBr's rnr,)rr0r;r;)r r __class__rr)rUr,rXs rYtest_subclassingzTestBSpline.test_subclassingsg   OOL ){{a||~''1,,,!++q000rnrrr>c d\}}|jdd||zdz}gd}|dz}|j||t|}tjj d}|j |j |} t|| ||} | jj||f|d|z||dzdzk(sJ|j d }| |j|d||jz||dzdzk(sJ| j dz | jfD]$} tttfit|| || &t|| ||jt|| ||jd t|| ||jt|| ||jd fD]} | j | j k(rJy) Nrrr0rrrr>rrrr)r4r5r6rr;)rirtuplerMrTrrSr r5rr"rIr'rLrrr) rUrrVrWr6r4shpos_axisrr5rXaxb1s rY test_axiszTestBSpline.test_axiss1 KK1a!eai ( !8 (A 2Yii##D) JJszzrz* + Aq!$ 'ssyyR\Ob(m;b!oMMMM ZZ "u{{b(mbhh6HQJKHHHHFF7Q;' :B )W : 1QR8 : : 1a.99;1a.99!<1a.==?1a.==a@B %B77aff$ $$  %rnc (d}|jgd}|jgdgdg}t|||d}t||dddf|}t||dddf|}t|d |j|d |d gy) Nr;)rr0r;r=r>rr)r9r;rr9)r;rr0r9rrr0rD)rSr rr)rUrVr6r4r5splspl0spl1s rY test_neg_axiszTestBSpline.test_neg_axiss  JJ, - JJ 6 7aAB'q!AqD'1%q!AqD'1%C"((DItCy+A"BCrncd}d}d}dD] }||||tdddD] }|||dd}tdd D] }|||d y ) a7 Splines with different boundary conditions are built on different types of vectors of knots. As far as design matrix depends only on vector of knots, `k` and `x` it is useful to make tests for different boundary conditions (and as following different vectors of knots). cBtjjd}tj|j |dzdz }|j |dzdz }|dk(r|d|d<t ||||}tj t|j|z dz }t|j|||}tj||j|j} t| |jz|d t|| d y ) zY To avoid repetition of code the following function is provided. r(rrr9rr6bc_typer0ryrzN)rMrTrr random_samplereyelenr4r design_matrixtoarrayrr5) rWr6rArr}rrr5 des_matr_def des_matr_csrs rYrun_design_matrix_testszHTestBSpline.test_design_matrix_bc_types..run_design_matrix_testss))''-C))!,r1B67A!!!$r)B.A*$u!%aaADs466{Q*+A074661a03L"00151244;GI  L466115 A L,U CrnrFr=clampednaturalrr; not-a-knotrrrN)r)rUrIrWr6bcs rYtest_design_matrix_bc_typesz'TestBSpline.test_design_matrix_bc_typess| D*  ( .B #Aq" - .q!Q 8A #Aq, 7 8 q! 6A #Aq* 5 6rnr)FTrdegreerc tjjd}|jd|dzz}tj|tj |}}|}tj tj|dz |dz |tj||d|dzztj|dz|dz|f}tjt||z dz } t|| ||} t| |tj||||jtj|dz |dz |dz|dzg}|s;tj t"5tj||||dddyt| |tj||||jy#1swYyxYw)z;Test that design_matrix(x) is equivalent to BSpline(..)(x).rrhr0r;rN)rMrTrrBaminamaxrrirCrDr rrErFarrayrcr rK) rUrrPrr}xminxmaxr6r4r5bsplines rY'test_design_matrix_same_as_BSpline_callz3TestBSpline.test_design_matrix_same_as_BSpline_call;sii##D)   bFQJ/ 0WWQZd  EE"++dQhq&9++dD!vz*:;++dQhq&9: ; FF3q6A:> "!Q;/ AJ--aA{CKKM HHdRi4#:tbyA Bz* <%%aA{; < <  %%aA{;CCE  < ryrz) rMrTrrrBrrr rEr4rFrr5) rUrrWr6r}rrrxcycrHs rYtest_design_matrix_x_shapesz'TestBSpline.test_design_matrix_x_shapesWsii##D)   GGC%%a(2-2 3   a 2 % *!!Q!,q! CA2AB2AB"00151244;GI  L46612E B  Crncxgd}tjd|dj}t|gdgdy)N)r2r2r2r@rArBrBrBr@r=)g?gmਪ?gK}\UU?r?ryrz)r rErFr)rUr4des_matrs rYtest_design_matrix_t_shapesz'TestBSpline.test_design_matrix_t_shapeshs7 ,((Q2::<;<" $rnctjjd}d}d}tj|j |dzdz }|j |dzdz }t |||}t t5tj||jddd|dddd}gd }gd }t t5tj|||dddy#1swYCxYw#1swYyxYw) Nrrhr=r?rrZr9r;)r?r2r@rArB@r) rMrTrrrBrrIrKr rEr4)rUrrWr6r}rrr4s rYtest_design_matrix_assertsz&TestBSpline.test_design_matrix_assertspsii##D)   GGC%%a(2-2 3   a 2 % *!!Q!,: & 6  ! !!TVVDbD\1 5 6  $  : & +  ! !!Q * + +  6 6  + +s(C, C8,C58DrA)rLrKrrMctjjd}tj|jd}|jd}|dk(r|d|d<t |||}t j ||}tjddd}t||||d t|||}t|j|jd y) Nrrrrr9rAr0r\rz) rMrTrrrr from_power_basisrirrr5) rUrArr}rcbrrkbspl_news rYtest_from_power_basisz!TestBSpline.test_from_power_basissii##D) GGCJJrN # JJrN j aDAbE Aw /''G< [[Ar "2Ru5%aG< 7rnctjjd}tj|jd}|jd|jddzz}|dk(r|d|d<t |||}t j ||}t||j|}t||j|}t|j|jd|jzzd y) Nrrr1rrr9rer\rz) rMrTrrrr rfrrrrr5) rUrArr}rrgr bspl_new_real bspl_new_imags rYtest_from_power_basis_complexz)TestBSpline.test_from_power_basis_complexsii##D) GGCJJrN # JJrNSZZ^b0 0 j aDAbE Aw /''G<*1affgF *1affgF  "}2F FUSrnctjgd}tjgd}tjt ||dd}t |j gddy) a} For x = [0, 1, 2, 3, 4] and y = [1, 1, 1, 1, 1] the coefficients of Cubic Spline in the power basis: $[[0, 0, 0, 0, 0],\$ $[0, 0, 0, 0, 0],\$ $[0, 0, 0, 0, 0],\$ $[1, 1, 1, 1, 1]]$ It could be shown explicitly that coefficients of the interpolating function in B-spline basis are c = [1, 1, 1, 1, 1, 1, 1] r<)r0r0r0r0r0rLre)r2r0r0r0r0r0r0r\rzN)rMrTr rfrrr5)rUr}rrs rYtest_from_power_basis_exmpz&TestBSpline.test_from_power_basis_exmpsO HH_ % HH_ %'' Aq)(L09; 7eDrncdtjddg}tjdg}|jd|jdtjddd}|jdt ||d}t ||tj |dzy)Nrr0rAFwriterhr3)rMrTsetflagsrir rrt)rUr4r5rkrXs rYtest_read_onlyzTestBSpline.test_read_onlys HHaV  HHcUO     [[Ar " %  a1 ""r||B/#56rncBt}fd}td||y)Nrxcp|j\}}}j|||| dz d}||y)Nr0i')rbri)rrXr4r6rkrVs rY worker_fnz/TestBSpline.test_concurrency..worker_fns9eeGAq!QqT1aRT7E2B bErnrh)rar()rUrVrXrws ` rYtest_concurrencyzTestBSpline.test_concurrencys 2 &  Iq1rncygwinz'threading.get_native_id not implemented)rr c nt}tjddd}||}tj}tj t |jd|dd|jj|jj}|j|ddtj t |jd|dd|jj|jj}|j|dd||_||_ t|||y) Nrr0rhr4z.datzw+)moderHrr5) rarMri threading get_native_idmemmapstrjoinr4rHrr5r)rUtmpdirrXrkrltidt_mmc_mms rY test_memmapzTestBSpline.test_memmaps ! [[Ar "R5%%'yyV[[1SE78t ssyy ;##QyyV[[1SE78t ssyy ;##Q"x(rnN);r(r)r*rZrermrqrurrskip_xp_backendsrrrrrrrrrrrrrrrrrrrrr rxfail_xp_backendsrrr$r.rcmark parametrizerr6r<rOrXr]r`rcrirmrortrxxfailsysplatformrdrr+rnrYr.r.2sM!F )* >D  $l+OPk*KLgk2 /3MQ /A ;1 .E&W? ( 6? FG4f%9:S;S A M :# F 7+,v&PQBRB4 1!1 [[VU2q\2%3%:D(6T [[],EF [[XuQx01G4C"$+$ [[Y)CD 8D 8 [[Y)CD TD TE& 7 2 [[  8 )  )rnr. insert_knotc>eZdZejj dgddZejj dgddZdZejj dgd d Z ejj d d d gdZ dZ dZ y ) TestInsertxval)r?r2rDr>@@c |jd|j}|j|dz}t||d}t |j t |j |jf}tt||}|j|}t|j |j|j dt|j |j|j d|j dz d||dk7r|n|dd} |j| d |dd|ddzzf} t|| || d|j|dz} t|| d} t||j|| fd d} | j|} t| j |j dt| j |j|j|j | j|j fd d||dk7r|n|dd} |j| d |dd|ddzzf} t| | | | dy) NrrGr=rZr\rzr0r9rr)rQrRsinrr r4r5r6r rrrrSconcatcosr)rUrrVr}rr9rbspl_1fspl_1rky1spl_y1spl_yyspl_yy1s rY test_insertzTestInsert.test_inserts IIarzzI * FF1IqL A+ '):355)A355I&s+,%FHH!5EBFHHYsuufQh,?!@uM!B%-QQsV YYC123B01 2Br7VVAY\#ArQ/#ArxxBax'@AF$$T* 5777 II HHcood+--v/A/A$/G/I/IJQRH S !B%-QQsV YYC123B01 2r GBKer;)rr;c |jd|j}|j|dz}t||d}t t |t |jt |j|jf|}|j||}t|j|j|jdt|j|j|jd|j dz d||d k7r|n|dd } |j| d |dd|dd zzf} t|| || dy) NrrGr=rZmr\rzr0r9r)rQrRrrr rr r4r5r6rrrSr) rUrrrVr}rr9rrrks rYtest_insert_multizTestInsert.test_insert_multis" IIarzzI * FF1IqL A+ (/1B3551I355QUV  a(FHH!5EBFHHYsuufQh,?!@uM!B%-QQsV YYC123B01 2Br7rnc6tjjd}d\}}|jtj|j ||zdz}|j|j |ddf}t |||}|j|j ||dz|| dz }|j|} |j|j ||dz|| dz d } t|| | | d y) N90rEr0rr=r;)lowhighrrrrr\rz) rMrT default_rngrSruniformr rr) rUrVrrWr6r4r5r9xvrrks rYtest_insert_randomzTestInsert.test_insert_random(sii##E*1 JJrwws{{!A{67 8 JJs{{Aq {2 3aA ZZ !A#Qr!tW = ># ZZ !A#Qr!tW2 F GBr7rnr) r皙?r@rB@@@gffffff@rc |jd|j}|j|dz}t||d\}}}t |j ||f\}}t |||d}|j|} t||jd\} } }t| j|j | d t| jd| d z |j | d| d z d tjjd j!d dd} |j | } t| | |j t#| | | |fd y)NrrGr=rZrrTperr\rzr0rrr)r)rQrRrrr|rSr rrrbrr4r5rMrTrrr)rUrrVr}rr4r5r6r9rtfcfrrks rYtest_insert_periodiczTestInsert.test_insert_periodic6s0 IIarzzI * FF1IqLA#1a2::1v&1aA:6#2swwD1 BBe<!A 2er!t9(=EJ %%d+333K ZZ b 2::eEBA;.G#HuUrnrNrc|jd|jdztjz}|j ||j |}}t ||d|zzd}||_t ||d}||_t ||d}||_d} |j| } |j| } |j| } t| j| jd t| j| jd t| j| jd| jzzd y) NrrGr;r1r=rZ @r\rz) rQrRrMpirrrrrrr4r5) rUrrVr}y_rey_imr9spl_respl_imrrspl_1respl_1ims rY test_complexzTestInsert.test_complexJs IIarzzI *Q . 6VVAYq d D2d7Na8%#Atq1(#Atq1( #$$R($$R(77R \!9Frncd}d}tjdg|dzzgdzdg|dzzz}tjt||z dz }t |||d}t t 5t||||fd dddt t 5|j|dddy#1swY3xYw#1swYyxYw) Nrr=rr0r;r=r>rrrrTr) rMrTonesrDr rIrKrr)rUrr6r4r5r9s rY+test_insert_periodic_too_few_internal_knotsz6TestInsert.test_insert_periodic_too_few_internal_knotsas  HHaS!A#Y-QqS 9 : GGCFQJN #aA:6 : & , 21ayd + ,: & OOB    , ,  s2B6C6B?C cd}tjdg|dzzgdzdg|dzzz}tjt||z dz }t |||}t t 5|jddddt t 5|jddddt t 5|jdddddy#1swY^xYw#1swYAxYw#1swYyxYw) Nr=rr0rrr9rr)rMrTrrDr rIrKr)rUr6r4r5r9s rYtest_insert_no_extrapz TestInsert.test_insert_no_extrapus  HHaS!A#Y-QqS 9 : GGCFQJN #aA : & OOB  : &  OOA  : & $ OOAO # $ $      $ $s$.CC)C5C&)C25C>) r(r)r*rcrrrrrrrrrr+rnrYrrs [[V%AB!=C!=H [[C88& 8 [[T$8VV" [[]T:,>?G@G, ( $rnrc dd}dD]S}t|}tt|D].\}}|||td|dzD]}||||dd0Uy)Nc |j\}}}tj|}tj|ddz d|dd|ddzz|ddzf}t t ||||f||||||d|d|j y) Nrrrr0r9zder = z k = )r^r_err_msg)rbrMuniquerrrr6) rXrrr^r_r4r5r6r}s rY check_splevz,test_knots_multiplicity..check_splevs%%1a IIaL EE!A$s(C122A/2s: ;a!QC0!As)se6!##.G Irnr0r;r=r>rrZr0r)rryry)ra enumerate_make_multiplesr)rr6rXrr5rs rYtest_knots_multiplicityrssI6 ! $q12 6EAr A Q!} 6B3u5 6 66rncb|dk(r|||cxkr ||dzkrdSdS|||z||k(rd}n)|||z |||z||z z t||dz ||z}|||zdz||dzk(rd}||zS|||zdz|z |||zdz||dzz z t||dz |dz|z}||zS)zw Naive way to compute B-spline basis functions. Useful only for testing! computes B(x; t[i],..., t[i+k+1]) rr0r2r?_naive_Br}r6rr4c1c2s rYrrs  Avda(!AaC&(s1(c11v1~ !A$h1Q3!A$ '(1ac1a*@ @1Qx1QqS6  G!AhlQqs1uX!A#. /(1ac1Q32J J Grnck(rn|jdz cxkr dzksJJk\rjdz ksJtfdtddzDS)z= Naive B-spline evaluation. Useful only for testing! r0rc3RK|]}|z t|z z ywNr).0rr5rr6r4r}s rY z_naive_eval..s-F1q1vAqsA..Fs$') searchsortedrsumr)r}r4r5r6rVrs```` @rYrrs AaDy  OOAq !A % Q41 !A#    6a!''!*q.(( ( Fa1 F FFrncjddzz }|dzk\sJjd|k\sJcxkr |ksJJtfdt|DS)z'Naive B-spline evaluation, another way.rr0c3FK|]}|t|zywrrrrr5r6r4r}s rYrz _naive_eval_2..s%=qqthq!Q**=!)rrr)r}r4r5r6rVrWs```` rYrrsr  acA !8O8 771:?? Q41 !    =E!H= ==rnct||dzz }||dzk\sJ|jd|k\sJd}t|D]F}tj||||zdzd|}|||t j |zz }H|S)Nr0rr?r;Fr)rDrrr rrM nan_to_num)r}r4r5r6rWsrrXs rYrrs A!A#A !8O8 771:?? A 1X% @G ! !!Aac!e*% @ C QqTBMM!$ $$% Hrnctjt|}tj||dk|dkDz|dk\|dkz|dk\|dkzgdddg}|j |S)z+ A linear B-spline function B(x | 0, 1, 2).rr;r0cyNr?r+r}s rYzB_012..rnc|Srr+rs rYrzB_012..sArnc d|z SNr@r+rs rYrzB_012..s Arn)rM atleast_1dr piecewiserS)r}rVresults rYr{r{sx '*+A \\!q1uQ/!VA.!VQ/1)+~FHF ::f rnctjt|}|dk|dkD|dkz|dkDg}|dk(r dddg}n|dk(r ddd g}ntd |tj|||}|S) z0A quadratic B-spline function B(x | 0, 1, 2, 3).r0r;rc||zdz Srr+rs rYrzB_0123..s1Q3r6rncd|dz dzz S)Ng?rr;r+rs rYrzB_0123..s41T6A+-rncd|z dzdz S)NrAr;r+rs rYrzB_0123..sBqD19q=rncyNr2r+rs rYrzB_0123..rrncy)Ngr+rs rYrzB_0123..rrncyrr+rs rYrzB_0123..rrnznever be here: der=)rMrr rKr)r}rcondsfuncspiecess rYrrs '*+A UQUq1u%q1u -E ax!-(* .se455 \\!UE *F Mrn#r=c(tjjd}tj|j||zdz}|j|}|j ||j |}}t j |||S)N{r0)rMrTrrrSr construct_fast)rWr6rVrr4r5s rYrarasp ))   $C  1Q3q5!"A 1 A ::a="**Q-qA  ! !!Q **rnc#K|j|j}}|jj}|d|dd|d|d<t ||||jj}|d|d|dzt ||||jj}|d|| dz dt |||yw) zIncrease knot multiplicity.rrNr0r9)r5r6r4copyr )rXr5r6t1s rYrrs 33qA B2Br"I VBrF "a  B!uBt!H "a  B2Br!tuI "a sBcNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) TestInteropcltjddtjzd}tj|}t ||}|j |j |jf|_|||c|_ |_ |_ tjddtjzd|_ tj|j |j |j f}tj||f|_t!|j |j|j|_y)NrrBrr)rMrirrrr4r5r6rbrkrrXxnewc_dstackrr b2)rUrkrrXrs rY setup_methodzTestInterop.setup_method s [[BruuHb ) VVBZ r2 &CCacc?#%r1 $&KK2bee8R0 UU133QSS= !))RH%!##tww,rnc  |j|j|j}}}tt ||||ddtt ||j ||ddtt j|Dcgc]}t ||c}||ddttd5t ||dddttd|jjdz}|jj|}|j||j f}tt jt ||||j|ddycc}w#1swYxYw)Nr\r]zCalling splev.. with BSplinematchr0r)rrXr rrrbrMrSrIrKr1rr5r" transposer4r6)rUrrXr r}r2rrbs rY test_splevzTestInterop.test_splevs.iia dA$e% 9dAEE*$e% 9 #>AE!QK#>?$e% 9 :-K L  $O 5BDDII& '$ . TT^^B ttR 5s#344**2.U H$?   sE5 > E::Fc|j|j}}t||}tj||\}}}t |d|dt |d|d|d|k(sJt||d\}}}}t |d|dt |d|d|d|k(sJt ||} t || dt |} t || |dy)Nrr\rzr0r;T) full_output)rkrrrrrr ) rUr}rrbr4r5r6tck_frrrXs rY test_splrepzTestInterop.test_splrep1sww1Ql,,q!$1aA.A.1v{{ 1$7q!Qa!%0a!%0Qx1}}1c]2E* SM1Q4e,rnc&|j|j}}tj||f}t t 5t ||dddt t 5tj ||dddt td5t |dd|dddddt td5tj |dd|dddddy#1swYxYw#1swYxxYw#1swYXxYw#1swYyxYw)Nm > k must holdr r=) rkrrMrrIrKrrrJ)rUr}ry2s rYtest_splrep_errorszTestInterop.test_splrep_errorsIsww1 UU1a4[ : &  1bM  : & LLB  9,= > ! 1Ra5!BQ%  ! 9,= > ' LL2A"1 & ' '     ! ! ' 's/ C#!C/C;=D#C,/C8;DDctjdtjjd}t |\}}t j|\}}t ||dt tjt|||dt tjt|||dt |dd\\}}}}}t ||dt tjt|||dy) NrrGr=rr\rzrT)rr) rMrQrRrrrrrSr) rUr}rXurbu1b_fu_frs rY test_splprepzTestInterop.test_splprepYs IIb + 3 3F ;qz1--"R 2E* 5A;/? 5C=115A&a1$? cAq!3U+ 5c?3QUCrnctjdjd}ttd5t |dddttd5t j |dddtjddd}ttd 5t |gdddttd 5t j |gdddgd }ttd 5t |gdddttd 5t j |gdddgd }gd }ttd 5t |gd|gdddy#1swY0xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYvxYw#1swYyxYw)N<rztoo many values to unpackr rr?r=)numr) >Ir$ >Kr%zError on input data)r0r=r;r>)rg333333?g?r0) rMrQrrIrKrrrirJ)rUr}rs rYtest_splprep_errorszTestInterop.test_splprep_errorshsy IIe  $ $Y / :-H I  AJ  :-H I  MM!   KK21 % 9,= >  QCL  9,= >  MM1#   E :-B C  QCL  :-B C  MM1#     :-B C % qc4^ $ % %/                % %sS E0E=" F F< F""F. F:0E:=F FF"F+.F7:Gc|j|j}}tjgdtjz}t t ||ddt t |j|j|jf|ddttd5t |dddd|jjdd d }tjt |j||jfd}|jd k(sJt ||z tj|d y#1swYxYw)N)rrrDrrr]zCalling sproot.. with BSpliner rw)mestr0r;r)r=r;r>rrz)rXr rMrTrrrr4r5r6rIrKrrSrr)rUrXr rootsc2rrrs rY test_sprootzTestInterop.test_sproots2-.ruu4q 5t$?QSS!##/TM:-L M 2B  ddnnQ1% ZZc2440r: ;xx9$$$U BMM"$5EB  s #EE c J|j|j}}ttdd|tdd|jddttdd||j ddddt td5tdd|ddd|jjddd}tjtdd|j||jf}|jd k(sJt|tdd|dd y#1swYxYw) Nrr0ryF)r^check_0dzCalling splint.. with BSpliner r;)r=r;)r^ check_shape)rXr rrrbrrIrKr5rrMrSr4r6r)rUrXr r*integrs rY test_splintzTestInterop.test_splints2q!Qq!QUU+%% Iq!Q Aq) G:-L M  1a  ddnnQ1%F1a"$$RTT):;<||v%%%q!Qe H  s DD"c |j|jfD]@}t|jt|jz }|jj }|dkDr9t j|t j|f|jddzf}dD]}t|}tj|jj ||jf}t|j|ddt|j|dd|j|dk(sJt|tsJt|t rJCyNrr0r0r;r=r\rzr;)rXr rDr4r5rrMrzerosrrrr6rr r r1rUrXctb_crWbdtck_ds rY test_splderzTestInterop.test_splders&&$''" 0AQSSCH$B##((*CAveeC2%#))AB-*?!@@A 0AY acchhj#qss%;<eAhU;eAhU;ttuQx'''!"g...!%/// 0 0rnc |j|jfD]@}t|jt|jz }|jj }|dkDr9t j|t j|f|jddzf}dD]}t|}tj|jj ||jf}t|j|ddt|j|dd|j|dk(sJt|tsJt|t rJCyr3)rXr rDr4r5rrMrr5rrrr6rr r r1r6s rYtest_splantiderzTestInterop.test_splantiders&&$''" 0AQSSCH$B##((*CAveeC2%#))AB-*?!@@A 0]((!##((*c133)?@eAhU;eAhU;ttuQx'''!"g...!%/// 0 0rnc|j|j|j}}}|jjdz}d|j||j|dzzz}t ||t ||j|j |jf}}tt||t||dt|tsJt|tsJtt|j j}|j j|dddz} t ||j| |jf} t ||} tt!j"t|| jddd| |dt| tsJt| tsJy)Nr;rr0r\rzrr)rXr rkr4rrr5r6rrr r r1rr"rrMrS) rUrXr rkrtnbntck_nr2rtck_n2bn2s rYrzTestInterop.test_insertsbFFDGGTWWr2 CCHHM !##a&133qs8# $2qM6"qssACCo#>Eb" b%(u 6"g&&&%'''5# $ TT^^BqrFTM *RTT2rtt,-Rn  5V#45??1aHBe -#w'''&%(((rnN)r(r)r*r rrrr r&r,r1r;r=rr+rnrYrrs= -H6-0' D%:C"H(0"0")rnrceZdZejddej zZejeZdZ dZ dZ dZ e jjdgdd Ze jjdgdd Zd Zd Ze jjdgd dZdZdZdZe jjdgddZdZdZdZdZdZe jj=ddZdZ dZ!dZ"dZ#d Z$d!Z%d"Z&d#Z'e jjdgd$d%Z(d&Z)d'Z*d(Z+d)Z,d*Z-y+), TestInterpr?r@cn|j|j|j|jfSr)rSrkrrs rY_get_xyzTestInterp._get_xys'zz$''"BJJtww$777rnctt5t|j|jddddy#1swYyxYw)NrDrZ)rIrJrrkrrUs rYtest_non_int_orderzTestInterp.test_non_int_orders3 9 % 8 tww3 7 8 8 8s #<Ac|j|\}}t||d}t|||ddt||dd}t|||ddy)NrrZryr]r9r6rrGrrrUrVrkrrXs rY test_order_0zTestInterp.test_order_0Vb!B r2 +"rE: r2 4"rE:rnc|j|\}}t||d}t|||ddt||dd}t|||ddy)Nr0rZryr]r9rLrMrNs rY test_linearzTestInterp.test_linearrPrnr6rc|gd}gd}ttd5t|||dddy#1swYyxYw)Nrr0r;r=r>r)rr0r;r=r>rrrz Shapes of xr rZrIrKrrUr6r}rs rYtest_incompatible_x_yz TestInterp.test_incompatible_x_ys7  $ :] ; * q!q ) * * *s2;cgd}gd}ttd5t|||dddgd}ttd5t|||dddgd}tj|j d}ttd5t|||dddy#1swYxYw#1swYfxYw#1swYyxYw) N)rr0r0r;r=r>rTzx to not have duplicatesr rZ)rr;r0r=r>rzExpect x to be a 1D strictly)r0r9)rIrKrrMrSrrVs rY test_broken_xzTestInterp.test_broken_x s   :-G H * q!q ) *  :-K L * q!q ) *  JJqM ! !' * :-K L * q!q ) * * * * * *  * *s#B.B:C.B7:CCc~|j|\}}dD]$}t|||}t|||dd&y)Nr;r=r>rrrryr]rM)rUrVrkrr6rXs rYtest_not_a_knotzTestInterp.test_not_a_knotsCb!B# ?A"2r1-A AbE2E > ?rnc|j|\}}t||dd}t|||ddtddD]&}t||d|||d |d (t||ddd }t|||ddtddD]&}t||d|||d |d (y) Nrrr@ryr]r0rrr9dy=rzr6rAr)rGrrr)rUrVrkrrXrs rY test_periodiczTestInterp.test_periodic!sb!B r2J ?"rE:q! IA AbeNAbfO% H I r2JR H"rE:q! IA AbeNAbfO% H Irnr[cVd}tjjd}tj|j |dz}|j |dz}|d|d<|j ||j |}}t |||d}t|||d y) Nrrrhrr9rrr@ryrz)rMrTrrrBrSrr)rUr6rVrWrr}rrXs rYtest_periodic_randomzTestInterp.test_periodic_random1s ii##D) GGC%%a(2- .   a 3 &u!zz!}bjjm1 q!q* =!ae,rnc|jjd}tjj d}|j |dztj z}tj|}d|d<dtj z|d<tjd|f}tj||d<tj||d<|j||j|}}t||ddd }t|D] }t||||dd|fd "t||d||dd y) Nrrr;r?r9r0rrr_ryrz)rkrrMrTrrBrrr5rrrSrrr)rUrVrWrr}rrXrs rYtest_periodic_axiszTestInterp.test_periodic_axis>s GGMM! ii##D)   a 1 $ruu , GGAJ!BEE " HHaV vvay!vvay!zz!}bjjm1 q!q*1 Eq :A AadGQq!tW5 9 :!A$1R56rnc0tjjd}d}d}tj|j |}|j |}|ddz |d<t t 5t|||ddddy#1swYyxYw) Nrrrr9r0rrr@)rMrTrrrBrIrKr)rUrr6rWr}rs rYtest_periodic_points_exceptionz)TestInterp.test_periodic_points_exceptionOsii##D)   GGC%%a( )   a uqy! : & > q!q* = > > >s 3B  BcPtjjd}d}d}tj|j |}|j |}tj |d|zz}t t5t||||ddddy#1swYyxYw)Nrr=rr;r) rMrTrrrBr5rIrKr)rUrr6rWr}rr4s rYtest_periodic_knots_exceptionz(TestInterp.test_periodic_knots_exceptionZsii##D)   GGC%%a( )   a  HHQQY  : & 7 q!Q: 6 7 7 7s BB%rct|j|j|d}t|j|jd|}t |j|}t |||jdt d|D]:}t |j||}t |||j| d <y) Nrr@T)rr6ryrzr0rrr)rrkrrrrr)rUr6rXrbr9rs rYtest_periodic_splevzTestInterp.test_periodic_spleves tww1j ITWWdgg415DGGS!QtwwZe4q! ?A!,C C477q!1 > ?rnc t|j|jdd}t|j|jd}t ||j||jdt j jd}d}t j|j|dz}|j|d z}|d |d <t||dd}t||d}t ||||dy) Nr=rr@reryrzrrhrr9r) rrkrrrrMrTrrrB)rUrXcubrrWr}rs rYtest_periodic_cubiczTestInterp.test_periodic_cubicrs tww1j I$''477J?$'' CLu=ii##D)  GGC%%a(2- .   a 3 &u! q!q* =!Q 3!c!f51rncTdt|j|jd}t|jt |j|jt j fd}t||j||jdy)Nr=rr@c,t|tS)Nrx)rrM)r}r5r6r4s rYrz6TestInterp.test_periodic_full_matrix..sK1ar$Brnryrz)rrkrr r!rM vectorizer)rUrXr5r5r6r4s @@@rYtest_periodic_full_matrixz$TestInterp.test_periodic_full_matrixsu  tww1j I DGGQ ' &twwA > \\B C$'' BtwwKerrrre*)rrrU)rUr}rlrs rYtest_deriv_speczTestInterp.test_deriv_specs%$A : & @ q!wi-> ? @: & 6 q!W 5 6: & 8 q!gY 7 8: & 1 q!R 0 1 "1 : & 5 q!aV 4 5 5 @ @ 6 6 8 8 1 1  5 5s:CC%C) C57DCC&)C25C>D c |jd}|dz}dgdg}}ttd5t||||fddddgdg}}ttd 5t||||fdddy#1swY:xYw#1swYyxYw) Nrr;)rrr0rzBad boundary conditions at 0.r re)irzBad boundary conditions at 6.)rQrIrKr)rUrVr}rrrs rYtest_deriv_order_too_largez%TestInterp.test_deriv_order_too_large s IIaL qDx&1 :-L M 5 q!aV 4 5x'1 :-L M 5 q!aV 4 5 5  5 5  5 5sA8B8BB cd}|j|\}}|d|zz}dgdg}}t|||||f}t|||ddtj||dd |dd d sJtj||d d |dd d sJd D]%}t||| }t|||dd'y)Nr=r1)r0y@)r0y@@reryr]rr0r r9rr0rZ)rGrrcmathr)rUrVr6rkrryrzrXs rYrzTestInterp.test_complexs b!B #b&[!zK=u r2q5%. A"rE:}}Qr!ua[%(1+uEEE}}Qr"vq\58A;FFF ?A"2rQ/A AbE2E > ?rnc|jd|j}|jd|j}dD]}t|||}||y)NrhrGrrZ)rQint32r)rUrVr}rr6rXs rY test_int_xyzTestInterp.test_int_xy(sT IIbI ) IIbI ) A"1a1-A aD rnct|jddd}|ddd}|ddd}dD]}t|||y)Nr9r0rrrrZ)rir)rUrVrkr}rr6s rYtest_sliced_inputzTestInterp.test_sliced_input2sI [[Q $ ssG ssG *A q!q ) *rnc|jd|j}|dz}|j|j|j fD]=}t j |dj |}ttt||?y)NrhrGr;r9) rQrRrOrPxpxatsetrIrKr)rUrVr}rzs rYtest_check_finitezTestInterp.test_check_finite<sp IIb I + qD&&"&&266'* @Aq" !!!$A *&8!Q ? @rn)r0r;r=rcvttd}|Dcgc]}|dz }}t|||ycc}w)Nrhr;rZ)listrr)rUr6r}ars rYtest_list_inputzTestInterp.test_list_inputEs9 rO aQT  1a1% s 6c|j|\}}|j||fd}dddgfg}dddgfg}t||d||f}t|||d d t||d d|j |d d|j d d t||d d|j |d d|j d d y)Nr0rr2r@rArBr=r@ryr]rrGr9)rGrrrrSrR)rUrVrkrryrzrXs rYtest_multiple_rhszTestInterp.test_multiple_rhsLsb!B XXr2hQX 'b"Xb"X r2UEN C"rE: beQK JJuQx{"**J 5E   bfaL JJuQx{"**J 5E rnctjjd}d\}}tj|j|}|j|dddf}t |||}|j j |dddfk(sJd|jdfg}d|jdfg}t |||||f }|j j ||zdz dddfk(sJy) Nrr=rrrrrr0rrrre)rMrTrrrr5r) rUrr6rWr}rrXd_ld_rs rY test_shapeszTestInterp.test_shapes]sii##D)1 GGCJJAJ& ' JJQ1aLJ ) q!Q 'ssyyQ1aL(((3::i()*3::i()* q!Qc ;ssyyQUQY1a0000rnc|j|\}}|j|}t||dd}t||ddgdgf}t|j|jdt||dd}t||ddgdgf}t|j|jdt||d d }t||d dd gf}t|j|jdt||dd }t||dd}t|j|jdt t 5t||dd ddd|j|j||j|fd}dddgfg}d ddgfg}t||d||f}t||dd}t|j|jdtjjd}d\} } tj|j| } |j| dddf} |j| |j| } } d|jdfg} d|jdfg}t| | | | |f}t| | | d}t|j|jdy#1swYyxYw)Nr=rLr@rxr\rz)rLrKrr;)NrKrrMtypor0rr?rJrrrrrrrrerK)rGrrrr5rIrKrrrMrTrrrSr5)rUrVrkrr5r ryrzrr6rWr}rrrs rYtest_string_aliaseszTestInterp.test_string_aliaseslsb!B VVBZ B!Y ? B!vh5I Jbdd/ B!(>@ B!*0F8(<>bdd/ B!5F G B!dXJ5G Hbdd/ B!\ B B!T :bdd/: & < r2F ; <XXrvvbz266":.QX 7b"Xb"X B!eU^ D B!(>@bdd/ii##D)1 GGCJJAJ& ' JJQ1aLJ )zz!}bjjm1288I&'(288I&'( 1a#s < 1a ;bdd/1 <  7 [[S,/ ?0 ?2 = ,;, T [[j) E* E>& ;5. 5?"*@ [[S,/&0&  " 130j 9?rnrEc|j|jk(sJ|j|j|zdzk(sJ|j}tj||ftj}t |D]R}||}|||k(r|}ntj ||dz }t j||||} | ||||z |dzf<Ttj||} | S)zAssemble an spline order k with knots t to interpolate y(x) using full matrices. Not-a-knot BC only. This routine is here for testing only (even though it's functional). r0rG) rrMr5rRrrr,evaluate_all_bsplslr) r}rr4r6rWArrleftbbr5s rYrrs 66QVV   66QVVaZ!^ ## # A !Qrzz*A 1X !t 1Q4<D??1d+a/D ' '1dD 9 !T!VDF]  ! AA HrncFttj|||f\}}}|j}|j|z dz }tj||ftj }t |D]R}||}|||k(r|} ntj||dz } tj|||| } | ||| |z | dzf<Ttj|j|} tj|j|} tj| | } | || ffS)z,Make the least-square spline, full matrices.r0rG)r|rMrSrr5rRrrr,rdotTrr)r}rr4r6rrWrrrrrr,Yr5s rYmake_lsq_full_matrixrs "**q!Qi(GAq! A  QA !Qrzz*A 1X !t 1Q4<D??1d+a/D ' '1dD 9 !T!VDF]  ! qssAA qssAA AA q!f9rnmethodnorm-eqqrc eZdZejj dZd\ZZejej eZ ej eZ e eje de ddeZedZedZdZed Zed Zd Zed Zd ZedZedZedZedZedZej<j?de e!dddZ"dZ#y)TestLSQr)rr=rr9rc|j|j|j|jf\}}}}t ||||\}}t |||||}t |j||jj|j|z dz fk(sJ|\} } tjj| |d\} } } } t |j| y)Nrr0r9)rcond) r}rr4r6rrrr5rrrMrlstsq) rUrr}rr4r6c0AYrXaarrrs rY test_lstsqzTestLSQ.test_lstsqsVVTVVTVVTVV3 1a%aAq1B Aq!Qv 6R ssyyQVVaZ!^----Biioob!2o6 Aq!R rncgt|j|j|j|jf|j \}}}}|j |}t|||||}t||||||} t|j| jdt|j| jd|j | j k(sJy)Nrwrryrz) r|rSr}rr4r6rtrrr5) rUrrVr}rr4r6rrXb_ws rY test_weightszTestLSQ.test_weights!sHc"**tvvtvvtvv&>?GG 1a LLO Aq!Qv 6aAqAf=SUU/SUU/sscee||rncTgt|j|j|j|jf|j \}}}}t jjdj|jd}|j|}t|||||d}t|||||d}t||||d} t|j|jd |j|j| j|jz dkrJy) Nrrrrrrrryrz)r|rSr}rr4r6rMrTrrrrrr5rabs) rUrVr}rr4r6rb_neb_qrb_no_ws rYtest_weights_samezTestLSQ.test_weights_same.sGc"**tvvtvvtvv&>?GG 1a II ! !$ ' / /QWWQZ / @ JJqMq!QQyAq!QQt< Aq!D9U366"&&DFF!23e;<<<$$5;7a"Q%''78::qss/ $7s 0D((D-cgt|j|j|jf|j|j \}}}}d}t jj||f}|j|}t||||d}t||||d} t|j| jdy)Nr=rrrrr\rz) r|rSr}r4r6rWrMrTrrr5) rUrVr}r4r6rWrrrb_neqs rYtest_multiple_rhs_3zTestLSQ.test_multiple_rhs_3TsGc"**tvvtvv&67GGG 1a II  1d)  , JJqMq!Q$71a9=e4rncgt|j|j|jf|j\}}}|j|j dz}t |||||}t ||j||||}t ||j||||} t||||d| |zzddy)N?@rr1r\r]) r|rSr}r4r6rrrrr) rUrrVr}r4r6r\rXrrs rYrzTestLSQ.test_complex]s=3rzzDFFDFF#34! HHUOQXXe_ahhuo6 eafn # WR[u=rnctjddd}|ddd}|ddd}t|d}t|||d|y)Nr9r0rr=r)rMrirr)rUrrkr}rr4s rYrzTestLSQ.test_sliced_inputsK[[Q $ ssG ssG AqM1a1V4rnc tjdjt}|dz}t |d}tj tj tj fD] }||d<ttt||||"y)N r;r=r9r) rMrQrrrrOrPrIrKr)rUrr}rr4rs rYtest_checkfinitezTestLSQ.test_checkfinitesq IIbM  ' qD AqM&&"&&266'* OAAbE *oq!Qv N Ornc|j|j|j}}}|jd|jd|jdt ||||y)NFrq)r}rr4r)r}rr4rsrrs rYrtzTestLSQ.test_read_onlysU&&$&&$&&a1      !qAf5rnr6r0c|j|j}}ttj|d|dd|}t ||||d}t ||||d}|dd|ddzdz }t ||||d y) Nrr9rrrrr0r@r\rz)r}rrrMrirr)rUr6r}rr4 spl_norm_eqspl_qrrks rYtest_qr_vs_norm_eqzTestLSQ.test_qr_vs_norm_eqsvvtvv1 BKK!aeQ/ 3%aA9E  AqAd;eafn # B%@rncdtj|jd}tj|jd}t |j|j|j dd}t |||j dd}|dd|ddzdz }t ||||d y) Nr;r=rrr0r9r@r\rz)rMrepeatr}rrr4r)rUr}rrspl_2rks rYtest_duplicateszTestLSQ.test_duplicatess IIdffa  IIdffa !DI1dff$?eafn #b 5959rnN)$r(r)r*rMrTrrrWr6rr}rrrir4parametrize_lsq_methodsrrrrrrrrrrrrrtrcrrrrrrr+rnrYrrsz ))   %C DAq 1 A 1 A  AaD!B%+Q/A ! !   =:: 0 05 M MM,55 > >55OO66 [[S$uQ{"34A5A:rnrc,eZdZdZdZedZdZy) PackedMatrixasA simplified CSR format for when non-zeros in each row are consecutive. Assuming that each row of an `(m, nc)` matrix 1) only has `nz` non-zeros, and 2) these non-zeros are consecutive, we only store an `(m, nz)` matrix of non-zeros and a 1D array of row offsets. This way, a row `i` of the original matrix A is ``A[i, offset[i]: offset[i] + nz]``. c||_||_||_|jdk(sJ|jdk(sJ|jd|jdk(sJy)Nr;r0r)rrncr"r)rUrrrs rY__init__zPackedMatrix.__init__sW vv{{{{awwqzV\\!_,,,rncL|jjd|jfSNr)rrrrIs rYrzPackedMatrix.shapesvv||A''rncjtj|j}|jjd}t |jdD]_}t |j |j|z |}|j|d|f|||j||j||zf<a|S)Nr0r)rMr5rrrminrr)rUoutnelemrnels rYtodensezPackedMatrix.todenseshhtzz" Qsyy|$ JAdgg A.6C:>&&DSD/C4;;q>$++a.3"666 7 J rnN)r(r)r*__doc__r propertyrrr+rnrYrrs%-((rnrc hddlm}|dtjd}|j}|j }|j }|j\}} |jd|k(sJ|j} |j} t||D]} || } t| |D]}|t| |k\r(|| |df| | df\}}}|| |df<td| D]*}t||| ||f| | |f\| ||f<| | |dz f<,d| | df<t| jdD]'}t||| ||f| | |f\| ||f<| | |f<)tt| jd}t| tj|tj |}|| fS) zjThis is a python counterpart of the `_qr_reduce` routine, declared in interpolate/src/__fitpack.h r)get_lapack_funcslartg preferred)rHilp64r0r?r9rG)scipy.linalg.lapackrrMrRrrrrrrr fprotarrrTint64)a_prstartrowrdlartgrrrrnzRrroirr5rrroffsR_ps rY _qr_reduce_pyr$s5 gRZZ{ KF A ZZF B GGEAr 771:?? A B 8Q F AYr2 FAC2JQq!tWa1g.GAq!AadG1b\ D%+Aq!AqD'1QT7%C"!Q$1ac6 DAaeH288A;' F%+Aq"QT(Bq!tH%E"1a4"QT( F FF* aggaj! "D q"((4rxx8" =C 7Nrnc8||z||zz}| |z||zz}||fS)zLGivens rotate [a, b]. [aa] = [ c s] @ [a] [bb] [-s c] [b] r+)r5rrrXrrs rYrrs1 A#!)B A!B r6Mrnc|j}|j\}}|j}|jd|jdk(sJtj|d|}||dz ||dz dfz ||dz df<t |dz ddD]P}t |||z }||d|df||dz||zdfzjd} ||| z ||dfz ||df<R|S)zBacksubsitution solve upper triangular banded `R @ c = y.` `R` is in the "packed" format: `R[i, :]` is `a[i, i:i+k+1]` rNr0.r;r9r)rrrrMrrr r) r#rr rrrr5rrsumms rYfpbackr( s A GGEAr B 771: ## # afART7Qr!tQwZ'AbdCiL 2a4R ."bdm!QsUD.!Aac!C%in$55:::BdTka1g-!S& . Hrnc0eZdZdZdZdZdZdZdZy) TestGivensQRc|d}tj|t}|dzdd|zz z}t||}||||fS)Nr=rGr0)rMrQrr)rUrWr6r}rr4s rY_get_xytzTestGivensQR._get_xyt7sF  IIau % qD1ac7N 1 !Qzrnc td}|j|\}}}}tj|||}tj|j \}}|j |z} |j\} } | |jd|z dz k(sJ|jdd|dz} tj| tj} |jj| |dz} t| | | }|dddf}tj | | | |t#tj$|j |z|j |z tj&|dt#tj$t)| |dddfz t)| |dddfztj&| dtj*|| }tj,|j.|j0||||tj2||\}}}t#||dddfdy) Nrhrr0rGr\rzgvIh%L=-a=)r,r rErrrrrindicesrMascontiguousarrayrdatarrr, qr_reducerminimumrrrr(rrrt)rUrWr}rr4r6a_csrqrqTyrrrrr y_c_fullc_bandedrs rY test_vs_fullzTestGivensQR.test_vs_full>s ]]1% 1a%%aA.uuU]]_%1ccAg 2QWWQZ!^a''''w!A#w'%%fBHH= JJ  q!A# & FB ' q$wZ1fb"-  199;?#$99;?457]]15EE S 3sR1X~#6#&sR1X~#689;s9KRW Y!S!!addAr1aaRTU!QAU;rncd}|j|\}}}}tj|||}|j\}}||jd|z dz k(sJ|jdd|dz} t j | t j} |jj||dz} t| | |} |dddf} t| | \} }tj| | || t| j| jdt!| j"| j"d| j$| j$k(sJt|| dy) Nrhrr0rGr\rzF) check_dtype)r,r rErr/rMr0rr1rrr$r,r2rrrrr)rUrWr}rr4r6r4rrrrr r7RRrs rYtest_py_vs_compiledz TestGivensQR.test_py_vs_compiledas, ]]1% 1a%%aA. 2QWWQZ!^a''''w!A#w'%%fBHH= JJ  q!A# & FB ' q$wZq"%B1fb2.acc. 188?uu}}BU+rnc6d}|j|\}}}}tjd|dzt}t j ||||\}}} |j d} tj|||} | |dddfzj} | jj| |dzf} | jdd|dzjtj}t|| dt!||| |j d|z dz k(sJy)Nrhr0rGrr\rz)r,rMrQrr, data_matrixrr rEtocsrr1rr/rrrr)rUrWr}rr4r6rrrrrr4a_wA_offset_s rYtest_data_matrixzTestGivensQR.test_data_matrix|s ]]1% 1a IIa1E * ,,Q1a8 62 GGAJ%%aA.qDz!((* XX  q!A#h '++g1g&--bhh72E*(QWWQZ!^a''''rnc d}|j|\}}}}tj||dzf}tj|||tj |\}}}t |||} tj||||t| |} tj||||||tj ||\} } } t| | dy)Nrhr;ryrz) r,rMrr,r@rtrr2r(r) rUrWr}rr4r6rrrr r5rrs rY test_fpbackzTestGivensQR.test_fpbacks ]]1% 1a EE!QT'N ,,Q1bll1oF 62 FB '1fb!, 1aL??1b!Q1bll1oqIAqAE*rnc d}|j|\}}}}tj|dzft}|D]N}t t j |||||dz|t t j ||||d|z|Py)Nrhr0rGr;)r,rMr5rrr,r)rUrWr}rr4r6 zero_arrayrs rYtest_evaluate_all_bsplz#TestGivensQR.test_evaluate_all_bspls ]]1% 1aXXq1uhe4  LD **1aq!a%@* N **1aq!A#>  L LrnN) r(r)r*r,r:r>rErGrJr+rnrYr*r*4s"!r;r=r?rr0r9rz)``x`` and ``y`` length must be at least 5r )rMrTrrrBrnormalrIrKrrrrQrrcr )rUrrWr}rx_duplexception_messages rYtest_invalid_inputz&TestSmoothingSpline.test_invalid_inputsii##D)  GGC%%a(1,q0 1 qD266!a%= 1a4 '#**Ra*@ @: & , !!QqrU + , : & , !!AB% + , : & 6 !!))Aq/1 5 6: & . !!DbD'1 - .1Iq : & - !&! , - IIaL GGAJG ]]:-> ? ( !!Q ' ( (+ , , , , 6 6 . .  - - ( (sHG:G!G(G4 H8 H GG%(G14G=H  Hctjtd5}|d}|d}|d}dddt|}t |dddy#1swY,xYw) ae Data is generated in the following way: >>> np.random.seed(1234) >>> n = 100 >>> x = np.sort(np.random.random_sample(n) * 4 - 2) >>> y = np.sin(x) + np.random.normal(scale=.5, size=n) >>> np.savetxt('x.csv', x) >>> np.savetxt('y.csv', y) We obtain the result of performing the GCV smoothing splines package (by Woltring, gcvspl) on the sample data points using its version for Octave (https://github.com/srkuberski/gcvspl). In order to use this implementation, one should clone the repository and open the folder in Octave. In Octave, we load up ``x`` and ``y`` (generated from Python code above): >>> x = csvread('x.csv'); >>> y = csvread('y.csv'); Then, in order to access the implementation, we compile gcvspl files in Octave: >>> mex gcvsplmex.c gcvspl.c >>> mex spldermex.c gcvspl.c The first function computes the vector of unknowns from the dataset (x, y) while the second one evaluates the spline in certain points with known vector of coefficients. >>> c = gcvsplmex( x, y, 2 ); >>> y0 = spldermex( x, c, 2, x, 0 ); If we want to compare the results of the gcvspl code, we can save ``y0`` in csv file: >>> csvwrite('y0.csv', y0); z gcvspl.npzr}ry_GCVSPLN-C6?F)r^r_r<)rMloadrRrr)rUr1r}rr[y_comprs rYtest_compare_with_GCVSPLz,TestSmoothingSpline.test_compare_with_GCVSPLsqRWWY|, - (S AS AJ'H  ( .'1-a0 4US ( (s AA#ctjjd}d}tj|j |dzdz }|dztj d|zz|dzz|j dd|z}|j||j|}}t||d}t||dd }|j|d |d d|z}t||||d y)z In case the regularization parameter is 0, the resulting spline is an interpolation spline with natural boundary conditions. rrr>r;r=r?r)lamrLrerr9r\rzN) rMrTrrrBrrVrSrrrir) rUrVrrWr}r spline_GCV spline_interpgrids rYtest_non_regularized_casez-TestSmoothingSpline.test_non_regularized_case s ii##D)  GGC%%a(1,q0 1 qD266!a%= 1a4 '#**Ra*@ @zz!}bjjm1*1aR8 *1aIF {{1Q42A. 4(%d+" $rnr;ctjjd}d}tj|j |dzdz }|dztj d|zz|dzz|j dd|z}t|j||f\}}t||}|jtdd D]}|j|}tj|t|j!d t|||} t#|||||z } t#| ||||z } | | kst%d | d d | d y)Nrrr>r;r=r?rrhrg>@zJSpline with weights should be closer to the points than the original one: z.4z < )rMrTrrrBrrVr|rSrchoicerrrrrrrrK) rUrVrrWr}rr9indrspl_worigweighteds rYtest_weighted_smoothing_splinez2TestSmoothingSpline.test_weighted_smoothing_spline sZii##D)  GGC%%a(1,q0 1 qD266!a%= 1a4 '#**Ra*@ @2::1v&1#Aq)::eCjr:2 3C A FF1c#h  # #C ()!Q2Es1S6{QsV+,D53=1S612Hh $CCG)3$,R="233 3rnN) r(r)r*rYr_rercr fail_slowrlr+rnrYrTrTs6 (>6Tp$* [[133rnrTc  |\ |\tz dz }|dzk\sJtz dz dzk\sJt fdt|D}tj|S)z-A naive 2D tensort product spline evaluation.r0c 3K|]9}tD])}||ft|zt |z+;ywr)rr,) rixiyr5r6nytxtyr}rs rYrzbspline2..B sU7E"I7&(Bi!Aq"b/)AaBO;7;7s?A)rDrrrMrS) rr4r5r6nxr#rrrsrtr}rs `` @@@@@rYbspline2rv: s DAq FB R1qB !A#II R1qB !A#II 77b 7 7C ::c?rncb|dk(r|||cxkr ||dzkrdSdS|||z||k(rd}n)|||z |||z||z z t||dz ||z}|||zdz||dzk(rd}||zS|||zdz|z |||zdz||dzz z t||dz |dz|z}||zS)Nrr0r2r?r,rs rYr,r,G sAvda(!AaC&(s1(c11v1~ !A$h1Q3!A$ '!AqsAq/ 91Qx1QqS6  7N!AhlQqs1uX!A#. /!AqsAaC2C C 7Nrnctz dz }|dzk\rt|k\sJtfdt|DS)Nr0c3FK|]}|t|zywrrxrs rYrzbspline..X s$6qta1am#6r)rDrr)r}r4r5r6rWs```` rYrWrWU sD A QA 1H3q6Q;' ' 6U1X6 66rnceZdZddZdZy) NdBSpline0ct|}|t|jksJ t|td|D|_td|D|_||_y#t$r |f|z}YGwxYw)aTensor product spline object. c[i1, i2, ..., id] * B(x1, i1) * B(x2, i2) * ... * B(xd, id) Parameters ---------- c : ndarray, shape (n1, n2, ..., nd, ...) b-spline coefficients t : tuple of 1D ndarrays knot vectors in directions 1, 2, ... d ``len(t[i]) == n[i] + k + 1`` k : int or length-d tuple of integers spline degrees. c3FK|]}tj|ywr)operatorindex)rkis rYrz&NdBSpline0.__init__..t s6bx~~b)6s!c3RK|]}tj|t!yw)rGN)rMrSr)rtis rYrz&NdBSpline0.__init__..u s?rrzz"E22?s%'N)rDrrJr1r6r4r5)rUr4r5r6r"s rYr zNdBSpline0.__init__\ sx1vs177|###  F 6A66?Q??  T A s A((A:9A:cbt|j}t||k(sJdg|z}t|D]}|j|||}}|j|}|||k(r|||<nt j ||dz ||<||||cxkr|||dzksJJ|||k\r||t||z krJt |}d}t|Dcgc]'}t|||j|z ||dz)} }tj| D]o} |j| t jt|Dcgc]0}t|||j|| ||j|2c}z} || z }qt j|Scc}wcc}w)Nnoner0r) rDr4rr6rMrr1 itertoolsproductr5prodr,rS) rUr}r"rrtdxdr6ritersidxterms rY__call__zNdBSpline0.__call__x s466{1v~~JtOt 4AVVAY!Bq ARU{!r2.2!ad8r/R!QZ/ // //Q4191B! !3 33 4 !H r=r;rr)r>r=)r9r;)r.)r0.)rrvrMrSrRrrrrDrOrrrTrrrrr)rUrVrrrr6r}rtargetrrr bspl2rrrrrs rYtest_2D_separablezTestNdBSpline.test_2D_separable s- /%%%, B467&1a!Q$!Q$1*%77   (*+1A+ 2   ZZ"** - 2 6 <$)" $ "bA&Ry3r7+---b  6 <5 JZZ^664 ZZ^r+RZZ8L-MNii##E* [[i[ (1 , ZZ^r||v%%%jjW%''6{CK1B/11qs +% 9M8 , sI#I.I).I.)I.c>d}gd}|Dcgc]\}}|dz|dzd|zzz}}}|j|\}}} |j||||fd} d} t|| d} | | } t||| | }| jdk(sJt | |j|gd zd | |jt j|dd | j j|dzk(sJt | ||j||j dddfdd|j| d}t||d}|| } | jdk(sJ|jd|j |z}t | |d ||jt j|dd |j j|dzk(sJt |||j||j ddddfddycc}}w)Nr;rr=rxrrrZr>r>ryrzr9rGFgvIh%,=rrrr;r;)r;r;) rrrrrrMr5rSrRrr)rUrVr"rr}rrrrr6c2_4rbspl2_4r val_singlec2_22bspl2_22 target2_22s rYtest_2D_separable_2z!TestNdBSpline.test_2D_separable_2 s /467&1a!Q$!Q$1*%77%%%, BxxRR(qx1 B*)Yr2q)"- ||t###*!12 @ r{  BHHRL"$5 8N$NNNN  6 ryrz)rrrrr)rUrVrr}rrr4rrr6rrrrrs rYtest_2D_separable_2_complexz)TestNdBSpline.test_2D_separable_2_complex% s /467&1a!Q$!Q$1*%77$*+q!bd(++%%%, B 6]xxRR(qx1 B*)Yr2q)"- ||t###*a0u >8+s C Cc tjjd}d}tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }|j |j |z dz |j |z dz f}t||f|| }d }t||t|||f||d tjgd gdf}t|||Dcgc]}t|||f||c}d ycc}w)Nrr=rrrrr>r0rZ)r2r2ryrzr0rr;皙?g?@) rMrTrrrrrrrrvr) rUrr6rsrtr5r9rrs rYtest_2D_randomzTestNdBSpline.test_2D_random8 sAii##E*  UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L KKbggaik27719Q;7K 8R!q) B b"Xq!45 BUU;"#$B@BC""r2h15C" $Cs7E c|j\}}}}gd}|Dcgc]\}}|dz|dzd|zzz}}}t||||f} | |jt|fk(sJt | ||dycc}}w)N)ffffff?r)rDg333333@)rrr=r;rZryrz)rrrrDr) rUrrkxkyrr}rrrs rY test_2D_mixedzTestNdBSpline.test_2D_mixedK s++-BB 1467&1a!Q$!Q$1*%77"bRH-Ry3r7+---b U ,8sBc|j|\}}}}gd}t||||f}dddddd d |zd fd d |zd fdd d|zfdd d|zfdd|zd |zfdi}|jD]y\} } |j|D cgc]\} } | | | c} } |j} ||| }t || d|j | |}t || d{dD]M}tt5|||dddtt5|j |dddOycc} } w#1swY;xYw#1swYlxYw)NrxrrZrc*d|dzz|dzd|zzzSNr=r;r+r}rs rYrz2TestNdBSpline.test_2D_derivative..[ sQTQTAaCZ!8rnr0r0c$d|dzzd|zdzzSrr+rs rYrz2TestNdBSpline.test_2D_derivative..\ sQTQqS1W!5rn)rrc$|dz|dzd|zzzSrr+rs rYrz2TestNdBSpline.test_2D_derivative..] sAA!!4rnr;r0cyr r+rs rYrz2TestNdBSpline.test_2D_derivative..^ rrnrcyr r+rs rYrz2TestNdBSpline.test_2D_derivative.._ rrnr=cyr r+rs rYrz2TestNdBSpline.test_2D_derivative..` rrncyr r+rs rYrz2TestNdBSpline.test_2D_derivative..a rrncyr r+rs rYrz2TestNdBSpline.test_2D_derivative..b rrnrGrryrz))r9rr9rr0) rritemsrSrRrrrIrK)rUrVrrrrrr test_casesr expected_fnr}r expected_valsdirect via_methodbad_nus rYtest_2D_derivativez TestNdBSpline.test_2D_derivativeT s++r+2BB 1"bRH- 8 5 4 rT1I~ rT1I~ "I~ "I~ rT1R4L.   *//1 COB JJ/12tq!Q"2"**'M 2"%F FM >.))"-b1J J E B C# )Fz* %bV$ %z* )  ( ) )  )3 % % ) )sE  E$EE E c tjjd}d\}}tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }|j |j |z d z |j |z d z f}tjgd gd f}t||f|||f }t||f|||f } t|||D cgc] } | |  c} d ycc} w)Nr)r;r=rrrr=rr>r0rrrZryrz) rMrTrrrrrrrr|r rUrrrrsrtr5rrbspl2_0rVs rYtest_2D_mixed_randomz"TestNdBSpline.test_2D_mixed_randomz s.ii##E*B UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L KKbgglQ." q0@AK B UU;"#$2r(A"b2b"XqRH5b /122 @2s*E ctjd}tjddz}t||dzd}t||dzd|zzd}|jdddf|jdddfz}t |j |j f||j |j f}|dzdddf|dzd|zzdddfz}t||f|}tj||D cgc] \} } | | f } } } || } tj| jrJt| || dt| j|j|dycc} } w) Nrrrr=rZr;ryrz)rMrQrr5rr4r6rrrranyrrr) rUr}rspl_xspl_yrrvaluesrgirrXrbxis rYtest_tx_neq_tyzTestNdBSpline.test_tx_neq_ty sM IIaL IIaL3 "1ada0"1adQqSjA6 WWQW a 0 0%''577+R%''5771CDQ$4AqD1Q3Ja#88%q!ff5!*!2!21a!8 9Aq!f 9 92h88C=$$&&&SW51 FLL16F :s<E6c|jd}|dz}t||d}|dzd|zz}t||d}|dzd|zzdz}t||d}|j|j|jf} |jddddf|jddddfz|jddddfz} | | dfS)Nrr=rZr;r0r) rUrVr}rr9rry_2rrrs rY make_3d_casezTestNdBSpline.make_3d_case s IIaL qD A+dQqSj"1cQ/dQqSj1n"1cQ/eeUWWegg &eeAtTM"ggdAtm$%ggdD!m$%2qyrnctjjd}|jddz\}}}|dz|dzd|zzz|dzd|zzdzz}|j \}}}t ||d} t |||D cgc]} | } } | | } | jd k(sJt| |d ycc} w) Nrr=rFrrr=r;r0rZrFryrz) rMrTrrrrziprr) rUrr}rrrt3c3r6bspl3rrrs rYtest_3D_separablezTestNdBSpline.test_3D_separable sii##E*++7++a/1aAA!$1qs Q7%%' B"bA&Q1 &Aa & &r||u$$$U3's B=c ^|j|\}}}t||d}tjj d}|j ddz\}}} t ||| D cgc]} | } } |j| } dd d d d d } | jD]z\}}| Dcgc]\}}}||||}}}}|j||j}t|| ||dt|j|| |d|ycc} wcc}}}w)Nrxr=rZrrrrcHd|dzz|dzd|zzz|dzd|zzdzzS)Nr=r;r0r+r}rrs rYrz2TestNdBSpline.test_3D_derivative.. s4q1a4x1a4!A#:'>!Q$1*q.'QrncBd|z|dzd|zzz|dzd|zzdzzSNrr=r;r0r+rs rYrz2TestNdBSpline.test_3D_derivative.. s0q1u1qs ';q!taczA~'NrncBd|zd|dzzdzz|dzd|zzdzzSrr+rs rYrz2TestNdBSpline.test_3D_derivative.. s0q1u!Q$ ';q!taczA~'Nrnc*d|zd|dzzdzzdzS)Nrr=r;r+rs rYrz2TestNdBSpline.test_3D_derivative.. s q1u!Q$ ';a'?rncyrr+rs rYrz2TestNdBSpline.test_3D_derivative.. rrn))r0rr)r;rr)r;r0r)r;r0r=)r;r0r>rGrryrz) rrrMrTrrrrSrrRrr)rUrVrrr6rrr}rrrxi_nprrrrxi_yi_zi_rs rYtest_3D_derivativez TestNdBSpline.test_3D_derivative s2%%%, B"bA&ii##E*++7++a/1a1aL)q)) ZZ RNN?*   *//1 QOB MRSSMCc[c37SMSJJ}BJJJGM E",m% H 0E,,R04m% P  Q*Ts , D#1D( c jtjjd}d}tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }|j |j |z dz |j |z dz |j |z dz f}t|||f|| }t|||f|| }d } t|| || d tjgd gdgdf} t|| | D cgc] } ||  c} d ycc} w)Nrr=rrrrr>r0rZ)r2r2r0ryrzrrg?rgffffff?) rMrTrrrrrrr|rr) rUrr6rsrttzr5r9spl_0rrVs rYtest_3D_randomzTestNdBSpline.test_3D_random sii##E*  UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L KKbggaik27719Q; ! DK ER a1-BB<a0 Br7 UU;""#$ Bb!9%)!9F!9sF0 c tjjd}d}tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }|j |j |z dz |j |z dz |j |z dz f|j |j |z dz |j |z dz |j |z dz fd zz}t|||f|| }t|||f|j| }t|||f|j| } tjgd gd gd f} t|| || d | | zzdy)Nrr=rrrrr>r0r1rZrrrryrz) rMrTrrrrrrrrrr) rUrr6rsrtrr5r9rrrs rYtest_3D_random_complexz$TestNdBSpline.test_3D_random_complex sii##E*  UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L [[rwwqy{BGGAIaK1E[ F [[rwwqy{BGGAIaK1E[ Fr IJR a1-BB<15BB<15 UU;""#$ Br Rr ]2 @rn cls_extrapNT call_extrapcR|j\}}}t||d|}gdgdgd} }}ttj||| f\}}} t ||| D cgc]} | } } |dz|dzd|zzz| dzd| zzdzz} || |} t | | d ycc} w Nr=)r6r)r9r)r8r r)r9g@r;r0rryrzrrr|rMrSrr)rUrrrrr6rr}rrrrrrs rYtest_extrapolate_3D_separablez+TestNdBSpline.test_extrapolate_3D_separable s%%' B"bA:>a1bjj1a),1aQ1 &Aa & &AA!$1qs Q7r{3U3 's B$r)FT)TNc\|j\}}}|\}}t||d|}gdgdgd} } }ttj|| | f\}} } t || | D cgc]} | } } |dz| dzd| zzz| dzd| zzdzz} || |}t || d ycc} wrr)rUrrrr6rrrr}rrrrrrs rYtest_extrapolate_3D_separable_2z-TestNdBSpline.test_extrapolate_3D_separable_2 s%%' B"( K"bA:>a1bjj1a),1aQ1 &Aa & &AA!$1qs Q7r{3U3 's% B)c|j\}}}t||d}gdgdgd}}}ttj|||f\}}}t |||Dcgc]}|} }|dz|dzd|zzz|dzd|zzdzz} || d } tj | d sJtj | d sJt| dd | dd d ycc}w)Nr=rZ)rr0r)r8rr)r9rrr;r0Frrr9ryrz)rrr|rMrSrrr) rUrrr6rr}rrrrrrs rY#test_extrapolate_false_3D_separablez1TestNdBSpline.test_extrapolate_false_3D_separable s%%' B"bA&nna1bjj1a),1aQ1 &Aa & &AA!$1qs Q7ru-xxq """xxr ###q fQrl? 's Cc |j\}}}t||d}tjddtjdddtjg}tjdddtjdd d g}tjd dddtjd d g}t |||Dcgc]}|} }|dz|dzd|zzz|dzd|zzdzz} tj |tj |ztj |z} tj| | <|| } tj | | jsJt| | d ycc}w)Nr=rZrr0r;rr8rrr9rryrz) rrrMrSrOrrrr) rUrrr6rr}rrrrrrrs rY test_x_nan_3DzTestNdBSpline.test_x_nan_3D- sD%%' B"bA& JJArvvq!Q7 8 JJCBFFAsC8 9 JJCArvvsC8 9Q1 &Aa & &AA!$1qs Q7xx{RXXa[(288A;6vvt rxxt %))+++U3's> E<c tjjd}d\}}tj|j ddd}tj |df|z||df|zf}tj|j ddd}tj |df|z||df|zf}|dddj jrJ|dddj jrJ|j |jdz|z d z |jdz|z d z f }|j}|j jrJtjgd gd f}t|ddd|dddf|||f }t|ddd|dddf|||f } t|||D cgc] } | |  c} dycc} w)Nrr=r=rr>rr9r;r0rrrrZryrz)rMrTrrrrflags c_contiguousrrrrr|rrs rYtest_non_c_contiguousz#TestNdBSpline.test_non_c_contiguous? sii##E*B WWS[[QQR[8 9 UUBqE8B;RVIbL0 1 WWS[[QQR[8 9 UUBqE8B;RVIbL0 1cc7==----cc7==---- KKbggqj2o1277A:?Q3FGK H CC77'''' UU;"#$2cc7BssG,aB8<b1gr#A#w/r2h?b /122 @2s:G c|j\}}}t||d}tdD]}d||j_d|j_t||d}|d|dk(sJy)Nr=rZFr4)rrrr writeable)rUrrr6rrbspl3_s rY test_readonlyzTestNdBSpline.test_readonlyX sy%%' B"bA&q *A$)BqEKK ! *"2rQ'Y6)#4444rncr|j\}}}tjgdgdg}t|||j |||}tj|||||g}|j d|j dk(sJt |j|jdtt5tjgd||gdzdddttd5tjd d gg||gdzdddy#1swYBxYw#1swYyxYw) Nr4)r>rrrgؗҜ ? 9  # #aVHb1#a% 8 9 9 : : 9 9sD!9D-!D*-D6c tjjd}d}tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }tjddddtj|j ddzddddf }|j |j |z dz |j |z dz |j |z dz f}t|||f|| }d }td ||y) Nrr=rrrrr>r0rZcLtjgdgdgdf}||y)Nrrr)rMr)rr9rs rYrwz1TestNdBSpline.test_concurrency..worker_fn} s&{&&'(B Grnrh) rMrTrrrrrrr() rUrr6rsrtrr5r9rws rYrxzTestNdBSpline.test_concurrencys s+ii##E*  UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L UU1aArwws{{{':;a?Aq!K L KKbggaik27719Q; ! DK ER a1-  Is3rn)r(r)r*rrMrrrrrrrrrrrrrrrrcrrrrr r rrrrxr+rnrYrr sB,!  " &*9X)$V>&$&,$)L@ G(!& 4Q0G&@& [[\D$<8 [[]T4L9 4:9 4 [[X |'DE4F4 @ 4$@2 5 94rnrc<eZdZdZdZej jdgddZdZ dZ dZ d Z ej jd e je jgd Zd Zej jd gddZdZy) TestMakeNDctjd}tjddz}|dddfdz|dzd|zzdddfz}tj||Dcgc] \}}||f }}}t ||f|d}t |||j dt||dzd}t||dzd|zzd} |jdddf| jdddfz} t | |jd d d d l m } | ||f|d } t | |||dycc}}w)Nrrr=r;r0rZr\rzr^rr])rlinearrry) rMrQrrr)rravelrr5scipy.interpolater) rUr}rrrrXrrrrrRGIrs rYtest_2D_separable_simplez#TestMakeND.test_2D_separable_simple s1 IIaL IIaL3 1d7Q!Q$1*dAg!66!*!2!21a!8 9Aq!f 9 9Aq66Q/R&,,.u=#1ada0"1adQqSjA6 WWQW a 0 0DFFQ7 E1a&&2Bb6:s&D?ctjd}tjd}tj||Dcgc] \}}||f }}}|dddfdz|dzd|zzdddfz}tj||||f}t ||f|dt j}||} tj||||fjt} | jdk(sJt| jddd| d|jd } t ||f| dt j}||} | jd k(sJt| jdddd| jd dycc}}w) Nrr=r;r6solver)$r>r>ryrzr)r(r;r;) rMrQrrr r)sslspsolverrrrr) rUr}rrrXrrvalues4rrrvalues22s rYtest_2D_separable_trailing_dimsz*TestMakeND.test_2D_separable_trailing_dims si IIaL IIaL!*!2!21a!8 9Aq!f 9 91d7Q!Q$1*dAg!66))VVVV<=Aq67a DbFFFF;<CCEJ||w&&&q!Q/U ,??<0Aq68qEb||z)))q!Q2|45 B':sF r6)rr)r=r0)r0r=rc~tjd}tjddz}tj||Dcgc] \}}||f }}}|dzdddf|dzd|zzdddfz}t ||f||t j }t|||jdycc}}w) Nrrrr=r;r&r\rz) rMrQrrr)r)r*rr!) rUr6r}rrrXrrrs rYrzTestMakeND.test_2D_mixed s IIaL IIaL3 !*!2!21a!8 9Aq!f 9 9Q$4AqD1Q3Ja#88Aq66Qs{{CR&,,.u= :sB9c*tjd}tjddz}tj|d<|dddfdz|dzd|zzdddfz}tt5t ||f|ddddy#1swYyxYw)Nrrr9r=r;r0rZ)rMrQrOrIrKr))rUr}rrs rY test_2D_nanszTestMakeND.test_2D_nans s IIaL IIaL3 "1d7Q!Q$1*dAg!66 : & - A! , - - -s /B  Bc tjgd}tjgd}tjgdgdgdgdgdgdg}|||fS)N)rr@rArBrr)r0r;r0r;r0r0)r0r;r=r;r0r0)r0r;r;r;r0r0)rMrT)rUr}rrs rY_get_sample_2d_datazTestMakeND._get_sample_2d_data sR HH. / HH. / HH""""""   !Qwrnc|j\}}}t||f|d}t||f|d}tjgdgdgj }t ||||dy) Nr0rZr rr0gffffff@g333333@rffffff @333333?r=r0r5r6rBrbr2r=ryrzr2r)rrMrTrrrUr}rrrrrs rYtest_2D_vs_RGI_linearz TestMakeND.test_2D_vs_RGI_linear sq**,1aAq61*%q!faA XX66899:  R#b'6rnc|j\}}}t||f|dtj}t ||f|d}t j gdgdgj}t||||dy) Nr=r& cubic_legacyrr4r7ryrz r2r)r)r*rrMrTrrr9s rYtest_2D_vs_RGI_cubiczTestMakeND.test_2D_vs_RGI_cubic sw**,1aAq61#++>%q!faG XX66899:  R#b'6rnr'c|j\}}}t||f|d|d}t||f|d}tjgdgdgj }t ||||dd y) Nr=ư>)r6r'r_r<rr4r7ryrr]r8)rUr'r}rrrrrs rYtest_2D_vs_RGI_cubic_iterativez)TestMakeND.test_2D_vs_RGI_cubic_iterative sy **,1aAq61&tD%q!faG XX66899:  R#b'DArnc|j\}}}t||f|dtj}t ||f|d}t j gdgdgj}t||||dy) Nrr&quintic_legacyrr4r7ryrzr=r9s rYtest_2D_vs_RGI_quinticz!TestMakeND.test_2D_vs_RGI_quintic sx**,1aAq61#++>%q!fa8HI XX66899:  R#b'6rnzk, meth))r0r )r=r<)rrCctjjd}tj|j d}tj|j d}tj|j d}|j d}t |||f||t j}t|||f||} tjj d d d } t|| | | d y)Ni@rrrrr0r&rgffffff?rrErryrz) rMrTrcumsumrr)r)r*rr) rUr6methrndmr}rrrrrrs rYtest_3D_random_vs_RGIz TestMakeND.test_3D_random_vs_RGI syy$$V, IIdlll* + IIdlll* + IIdlll* +9-Aq!9f#++F%q!QiE YY  3Sw  ?R#b'6rnc:|j\}}}ddi}ttd5t||f|fddi|dddttd5t||ft j ||ffddi|dddy#1swYJxYw#1swYyxYw)Nmaxiterr0r'r r6r=)r2rIrKr)rMr )rUr}rr solver_argss rYtest_solver_err_not_convergedz(TestMakeND.test_solver_err_not_converged s**,1a !n :X 6 7 A 6Q 6+ 6 7:X 6 G A 1a& 1 FQ F+ F G G 7 7 G GsB'BBBN)r(r)r*r$r-rcrrrr0r2r:r>r)gmresgcrotmkrArDrIrMr+rnrYrr s7(B4 [[S"JK>L>- 77 [[X 3;;'?@ BA B7 [[N 7 7GrnrcHeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) TestFpcheccd}tjdjdd}tjd}tjt d5t j|||dddtjt d5t j|||dddy#1swYExYw#1swYyxYw)Nr0rr;rz 1D sequencer )rMrQrrcr rK_bfpcheckrUr6r4r}s rY test_1D_x_tzTestFpchec.test_1D_x_t# s  IIbM ! !!Q ' IIbM ]]:] ; JJq!Q  ]]:] ; JJq!Q       sB4C4B=C cd}d|dzzdz }|dz}tj|}tj|}tj|||dk(sJt j t d5tj|||dddd|dzzdz}||z dz }tj|}tj|}tj|||dk(sJt j t d5tj|||dddy#1swYxYw#1swYyxYw)Nr=r;r0rFrhz Need k+1*r ) rMrQdfitpackfpchecrcr rKrSrT)rUr6rWrr4r}s rYtest_condition_1zTestFpchec.test_condition_1. s  AY] F IIaL IIaLq!Q'2--- ]]:[ 9 JJq!Q   qsGaK EAI IIaL IIaLq!Q'2--- ]]:[ 9 JJq!Q       s4D)D5)D25D>cd}dg|dzzdgzdg|dzzz}gd}tj|||dk(sJtj|||J|j }|d|d<tj|||dk(sJt j td 5tj|||ddd|j }|d|d<tj|||dk(sJt j td 5tj|||dddy#1swYyxYw#1swYyxYw) Nr=rr0r;r)r0r;r=r>rr9rz Last k knots*r zFirst k knots*rXrYrSrTrrcr rK)rUr6r4r}tts rYtest_condition_2zTestFpchec.test_condition_2C s:  C1IOqc1Q3i ' q!Q'1,,,zz!Q"*** VVXA2q"a(B... ]]:_ = ! JJq"a  !VVX21q"a(B... ]]:-= > ! JJq"a  ! !  ! !  ! !s"D2D>2D;>Ecd}dg|dzzddgzdg|dzzz}gd}tj|||dk(sJtj|||Jdg|dzzddgzdg|dzzz}tj|||dk(sJt j t d 5tj|||dddy#1swYyxYw) Nr=rr0r;rr0r;r=rr>rzInternal knots*r )rXrYrSrTrcr rKrUs rYtest_condition_3zTestFpchec.test_condition_3Y s  C1IA !ac * "q!Q'1,,,zz!Q"*** C1IA !ac *q!Q'2--- ]]:-> ? JJq!Q    s #CC c2d}dg|dzzdg|dzzz}gd}tj|||dk(sJtj|||J|j }|d|d<tj|||dk(sJtj|||J|j }|ddz |d<tj|||dk(sJt j td5tj|||ddd|j }|d dz|d <tj|||dk(sJt j td5tj|||dddy#1swY|xYw#1swYyxYw) Nr=rr0rr`r?zOut of bounds*r r9r\)rUr6r4r}rks rYtest_condition_4zTestFpchec.test_condition_4f s  C1IQqS ! "q!Q'1,,,zz!Q"*** VVX!1r1a(A---zz!Q"*** VVX!q1r1a(B... ]]:-= > ! JJr1a  !VVX22r1a(B... ]]:-= > ! JJr1a  ! !  ! !  ! !s.F F F  Fcd}gd}gd}tj|||dk(sJtjtd5t j |||dddgd}tj|||dk(sJtjtd5t j |||dddy#1swYexYw#1swYyxYw)Nr0)rrr0r;r;)rrrrwSchoenberg-Whitney*r )rrrrXrYrcr rKrSrTrUs rYtest_condition_5_x1xmz TestFpchec.test_condition_5_x1xm s   q!Q'2--- ]]:-B C JJq!Q   q!Q'2--- ]]:-B C JJq!Q       sB>C >C Ccd}gd}ddg}tj|||dk(sJtj|||Jy)Nr0)rrr0r0rg333333?r)rXrYrSrTrUs rYtest_condition_5_k1zTestFpchec.test_condition_5_k1 sI   #Jq!Q'1,,,zz!Q"***rncd}dg|dzzdgzdg|dzzz}dgdz}tj|||dk(sJtjtd5t j |||ddddg|dzzdgzdg|dzzz}dgdz}tj|||dk(sJtjtd5t j |||dddy#1swY~xYw#1swYyxYw) Nr=rr0r;rrwrfr rgrUs rYtest_condition_5_1zTestFpchec.test_condition_5_1 s   C1IOqc1Q3i ' CEq!Q'2--- ]]:-B C JJq!Q  C1IOqc1Q3i ' CEq!Q'2--- ]]:-B C JJq!Q       sC, C8,C58Dcd}dg|dzzddgzdg|dzzz}dgdzdgz}tj|||dk(sJtjtd 5t j |||ddddgdzddgz}tj|||dk(sJt j |||Jy#1swYJxYw) Nr=rr0r;rrr>rwrfr rgrUs rYtest_condition_5_2zTestFpchec.test_condition_5_2 s  C1IA !ac * E!GqcMq!Q'2--- ]]:-B C JJq!Q  E!Gq!f q!Q'1,,,zz!Q"***   s B==Ccd}gd}gd}tj|||dk(sJtjtd5t j |||dddy#1swYyxYw)Nr0) rrr;r=r>rrrr)r0r0r0@rprprrwrfr rgrUs rYtest_condition_5_3zTestFpchec.test_condition_5_3 s`  ' )q!Q'2--- ]]:-B C JJq!Q    s A##A,N) r(r)r*rVrZr^rbrdrhrjrlrnrqr+rnrYrQrQ s4   *!,  !>  +   + rnrQctj|||| }tt|dz Dcgc]}|||||dzj  }}||dd}tt|D]'}||dz }||xx|z cc<||dzxx|zcc<)|dxx|dz cc<t t |t |d||fScc}w)z)Split the knot interval into "runs". r0r9r;r\rz)rMrrrDrr) r}r4r6 residualsrprfpartscarriescarrys rY_splitrw s AaG $B6;CGAI6F Gi1b1g&**, GF G1R!G 3w<  Qq U qs u   2J)B-JCKYe< 2:Hs#Cc\t||||\}}d}d}tt|D]$}||dz||z dkDs|||kDs|}||}&|dk(r td||||dzzdzdz} || } t j || } tj |d| | || df} | S)z Insert a new knot given reduals.ig}Ô%Ir0z5Internal error, please report it to SciPy developers.r;N)rwrrDrKrMrr) r}r4r6rsrtrpidx_max fpart_maxr idx_newknotnew_knotidx_tt_news rY _add_knotr s1a+JFBGI 3v; " ac7RU?Q 6!9y#8Gq I" $PQQg;GAI.2q8K~H OOAx (E EE!FU)Xqy0 1E LrnceZdZdZej j dgddZdZdZ dZ dZ d Z d Z d Zej j d gd ej j dgddZdZdZy)TestGenerateKnotsctjdt}|dzdd|zz z}d}tjdg|dzzdg|dzzz}t ||||}|||z d z}d d lm}|j||||}t||||} t|| d t ||||} | ||z d z} |j|||| } t|||| } t| | d y)NrrGr=r2r0r?r)r6r4r;r_fitpack_repror\rz) rMrQrrTrr"radd_knotrr)rUr}rr6r4r9rs_frnew_tnew_t_pyspl2 residuals2new_t2 new_t2_pys rYtest_split_add_knotz%TestGenerateKnots.test_split_add_knot s IIau % qD2q1u:   HHbT1Q3Z2$!*, -aa1-VaZ!O ; Q1i0Q1i0xe4q!qE21gkA% a:6a:6  6rnr6rc0|jd|j}|j||jzdz }t t |||dd}t |||dd}|j||j}t||dy)NrrGrr6rr9r\rz) rQrRrrrr$rrSr)rUr6rVr}rr4r]s rYtest_s0zTestGenerateKnots.test_s0 s IIarzzI * FF1RUU719  1Q/ 0 4 AqA #A & ZZ"**Z -2E*rncd}|j||j}|dz}tt||dd}|j t t j|d}t|d|dy) NrhrGr=rrr9r\rz)rQrRrr$rSrrMrrUrVrWr}rknotsrls rY test_s0_1zTestGenerateKnots.test_s0_1 sh  IIarzzI * qD^AqA34::k"**Q-;<b 8%8rncd}|j|}|dz}tt||dd}|jt t j|d}t |d|dy)Nrr=rrr9r\rz)rQrr$rSrrMrrs rY test_s0_n20zTestGenerateKnots.test_s0_n20 s_  IIaL qD^AqA34::k"**Q-;<b 8%8rnc tjd}|dz}tt5t t ||ddddddy#1swYyxYw)Nrhr=rr6rnest)rMrQrIrKrr$rUr}rs rY test_s0_nestzTestGenerateKnots.test_s0_nest& sJ IIbM qD : & : 1QR8 9 : : :s A  Ac|jd|j}|j|tjzdz }d}t t |||d}gdgdgdgd gd g}|Dcgc]}|j||j!}}t|t|k(sJt||D]\}} t|| d t|||d\}} } t|d |j|d ycc}w)aY To generate the `wanted` list below apply the following diff and rerun the test. The stdout will contain successive iterations of the `t` array. $ git diff scipy/interpolate/fitpack/fpcurf.f diff --git a/scipy/interpolate/fitpack/fpcurf.f b/scipy/interpolate/fitpack/fpcurf.f index 1afb1900f1..d817e51ad8 100644 --- a/scipy/interpolate/fitpack/fpcurf.f +++ b/scipy/interpolate/fitpack/fpcurf.f @@ -216,6 +216,9 @@ c t(j+k) <= x(i) <= t(j+k+1) and store it in fpint(j),j=1,2,...nrint. do 190 l=1,nplus c add a new knot. call fpknot(x,m,t,n,fpint,nrdata,nrint,nest,1) + print*, l, nest, ': ', t + print*, "n, nmax = ", n, nmax + c if n=nmax we locate the knots as for interpolation. if(n.eq.nmax) go to 10 c test whether we cannot further increase the number of knots. rrGr=rr)r?r?r?r?rrrr) r?r?r?r?rBrrrr r?r?r?r?r@rBrrrr) r?r?r?r?r@rBrrrrr) r?r?r?r?r@rArBrbrrrrr\rzr9N) rQrRrrMrrr$rSrDrrr) rUrVr}rr6rwantedwantr4r]rs rY test_s_switchzTestGenerateKnots.test_s_switch- s. IIarzzI * FF1RUU719  ^AqA6726:>A   BHH"**T*4HH5zS[(((' /EAr Ar . /Ad+1ab 2::a=u=Is2$Dc`ttd}t||dd}t|y)Nrrr0)rr6)rrr$next)rUr}gens rYrz!TestGenerateKnots.test_list_inputY s& qNQSA. S rnc |jd|j}|j||jzdz }d}t t ||d|d}t |d|jgd|jd tt5t t ||dd dddy#1swYyxYw) NrrGrr=rhrr9rr\rzr>)r6r) rQrRrrrr$rrSrIrK)rUrVr}rrrs rY test_nestzTestGenerateKnots.test_nest_ s IIarzzI * FF1RUU719  ^AqA<= "I JJ?rzzJ R : & 4 12 3 4 4 4s B99Cc tjd}tj|tjzdz }t t 5t t||tjddddt t 5t t||tjd dddy#1swYMxYw#1swYyxYw)NrrFr) rMrQrrrIrKrr$rrs rYrzTestGenerateKnots.test_weightsp s IIaL FF1RUU719  : & 8 1 " 6 7 8: & 6 1 4 5 6 6 8 8 6 6s+C,CCCnpts)rarwrr)r{Gz?rctjjd}dtj|j |z}tj |tj zdz tj|dz dz z}d}t||||d}tt||||d }t||d y) Nrrhrrr;r=rrr9r\rz) rMrTrrrrrexprrr$r) rUrrrHr}rr6r4r]s rYtest_vs_splrepz TestGenerateKnots.test_vs_splrepz syy$$U+ rwwt|||./ / FF1RUU72: !A# !2 2  1a1 "1 % .Aa0 1" 5AE*rnc*d}tj|}|dz}tt||dd}t j t 5}t||dd}dddtdk(sJt|ddy#1swY,xYw)Nr=Jz5rr0r9r) rMrQrr$rcwarnsRuntimeWarningrrDr)rUrWr}rrrrbs rYtest_s_too_smallz"TestGenerateKnots.test_s_too_small s  IIaL qD^AqA78 \\. ) -QAe,C -1v{{b 3q6* - -s B  Bctgdgdgdd}tjtd5t |dddy#1swYyxYw)N)r?r2r@rA)rBrbrr)r?r?r?r?r0)rrzweights are zeror )r$rcr rKr)rUrs rYtest_zero_weightsz#TestGenerateKnots.test_zero_weights s=]M]aP ]]:-? @  I   s AAN)r(r)r*rrcrrrrrrrrrrrrrr+rnrYrr s70 [[S/2+3+99:*>X 4"6 [[V]3 [[S.1+24+" +rnrcZ|jd}|||z dz ||z }|d|zz dz }||dz||z dz }tj|d}|dddxxdz cc<|dddxxdzcc<t|tj||z dz |||}tj |dz |jdft }td|jddD]"} || ddf|| dzddfz || dzddf<$|||z |ztj|z z}|S)zStraitforward way to compute the discontinuity matrix. For testing ONLY. This routine returns a dense matrix, while `_fitpack_repro.disc` returns a packed one. rr0r;NrrrG) rrMrr rCemptyrrr factorial) r4r6rWdeltanrintrtiirmatrrs rY disc_naiver sD  A a!eaiL1Q4 E !GaKE 1Q3qs1uB ))B C!HH1II(266!a%!)$a(3A 88U1Waggaj) 7D 1aggaj! $,!Q$!AaCF)+QT1W , U5[1 t~~a0 00D KrnceZdZdZddZdZy)F_densezm The r.h.s. of ``f(p) = s``, an analog of _fitpack_repro.F Uses full matrices, so is for tests only. Nc||_||_||_||_|t j |t n||_||_|jjdk(sJt|t j|jd|z dz |||}||jdddfz|_ ddlm} t!| j#||j%|_|jdk(sJ||jz} tj(| t j*|j&jdf|_||_y)NrGr0rrr)r}rr4r6rMrtrrrr"r rCra_denser"rrdiscrb_denserr5rr) rUr}rr4r6rrrrrrs rYr zF_dense.__init__ s12au-&vv{{aU'!RVVAGGAJNQ$67 TUVW40 ;#SXXa^4<<> vv{{ Z%%BHHT\\%7%7%:;;<rnctj|j|j|z f}ddlm}m}||d\}}|j|jz}|jd}||d|d|f|d|} t|j| |j|j} tj|jdz| |j |j"z dzz} | |_| |j&z S)Nr)rreconomic)r{r0rr;)rMvstackrr scipy.linalgrrrrrr r4r6rrrr}rr9r) rUrabrrr5rqyrr5r9fps rYrzF_dense.__call__ s YY dllQ&67 8 +":&1 SS477] WWQZ !CRC"H+r#2w 'dffaT5E5EF VVDFFAITVVtvv!5 99 :DFF{rn)NT)r(r)r*rr rr+rnrYrr s.rnrceZdZdZefdZdZdZejjdgddZ dZ ejjdgdd Z ejjd Zejjdgd d Zd ZejjdddgejjdgddZejjdddgdZejjdddgejjdgddZejjdgdejjdgddZejjdgdejjdgddZy)_TestMakeSplrepBaseNc|jdk(r<|jddtjzd}|j |}d}|||fS|jddd}|j |dzdz dz}d}|||fS) Nrrr;rhgq@H&?rrFgQ @)rArirMrr)rUrVr}rrs rY _get_xyktz_TestMakeSplrepBase._get_xykt s <<: % Aqw+AAAa7N Aq"%A$ #Q&AAa7Nrnc tjddd}tjddd}tt5t |||j dddtt5t ddd|j dddtt5tj |jdddf}t ||d|j dddtj d}tt5t ||d z|d|j dddtj d }tt5t ||d z|d|j dddtj |jddf}tt5t ||d z|d|j dddtt5t |ddd |d zd|j dddtt5t ||d zd d|j dddtt5t ||d zd |j dddtt5t ||d zd dd|j dddtt5t ||d zdd|j dddtt5t tjdtjdd|j dddy#1swYxYw#1swYxYw#1swYxxYw#1swY;xYw#1swYxYw#1swYxYw#1swYxYw#1swY\xYw#1swY5xYw#1swY xYw#1swYxYw#1swYyxYw)NrrhrFrrer0r;rrrAr=)rrrAr9rDr6rrA)r6rrrA)rrrAr ) rMrirIrKr%rArrrrJrQrUr}rrs rYtest_input_errorsz%_TestMakeSplrepBase.test_input_errors s, KK2r " KK2r " : & 4 1dll 3 4: & ; 1T\\ : ;: & ;Aq)*A 1T\\ : ; GGBK : & C 1a41T\\ B CWWR[L : & C 1a41T\\ B C GGQWWQZO $ : & C 1a41T\\ B C: & D $B$Adll C D9 % E 1a43#t|| D E: & = 1a42t|| < =: & K 1a411T\\ J K: & E 1a412t|| D E: & Q  ! biilc4<< P Q Qa 4 4 ; ; ; ;  C C  C C  C C D D E E = = K K E E Q QsL8.M=MMM,,M9!#NNN N-:N:/AO8MMMM),M69NNN N*-N7:OOc xtt|||||jd}t|||||j}t||||||j}t |j |j dt |j |j d|j|jk(sJy)Nrr9)r4r6rrAr\rz)rr$rAr%rr4r5r6)rUr}rr6rr4spl_autospl_ts rY_test_with_knotsz$_TestMakeSplrepBase._test_with_knots2s 1Q E Fr Jq!qAt||DAqAaF EGG%8 EGG%8zzUWW$$$rnr6r0r;r=r>rrcT|j\}}}|j||||yr)rr)rUr6r}rrs rYtest_with_knotsz#_TestMakeSplrepBase.test_with_knots<s(.."1a aAq)rncVt||||j}t||||j}tt ||||jd}t |j |j dt |j|dt |j|dy)Nr@r9r\rz)r%rArrr$rr5r4)rUr}rr6r9spl_ir4s rY_test_default_sz#_TestMakeSplrepBase._test_default_sBs|!Q!T\\:"1a1dllC 14<<@ A" EuwwU3qu-/rncR|j\}}}|j|||yr)rr)rUr6r}rrs rYtest_default_sz"_TestMakeSplrepBase.test_default_sKs&.."1a Q1%rnc d}|jdk(r>tjddtjzd}tj|}ntj d}|dz}t j5t jdtt||d||jdk(}dddt j5t jdtt||d||j }dddtjdttj|jdg|j d zzf|d d y#1swYxYw#1swYnxYw) NgKH9rrr;rr=r7r6rrrr0r.rz)rArMrirrrQwarningscatch_warnings simplefilterrrr%rr4rr5r6)rUrr}rrbr9s rYrz$_TestMakeSplrepBase.test_s_too_smallPsB  <<: % Aqw+Aq A " A1A  $ $ & K  ! ! Aadllj.HJC  K $ $ & D  ! ! aa1dllCC  D s1v&ceeaS#%%']23AU , K K D Ds>8E(5E4(E14E=r4c$d}|jdk(r>tjddtjz|}tj|}ntj |}|dz}t ||||j}t |||d|j}|jjd k(sJ|jjd k(sJt ||d d |zz z|d|j}|jjd k(sJy) Nrhrrr;r=r@h㈵>rr0) rArMrirrrQr%r5r")rUr6rWr}rr9rrs rY test_shapez_TestMakeSplrepBase.test_shapems  <<: % Aqw*Aq A ! A1A!Q!T\\:AqAt||DuuzzQww||q   Aq1ac7{a4Nww||q   rncd}ddtjz}}tj|||dz}tj|}t t 5t ||dddddy#1swYyxYw)Nrhrr;r0:0yE>nonsenser)rMrrirrIrKr%rUNrrXr}rs rYtest_error_on_invalid_bc_typez1_TestMakeSplrepBase.test_error_on_invalid_bc_typesf !BEE'1 KK1a!e $ FF1I : & : 1j 9 : : :s A//A8rArrc tjddtjzd}tj|}tjddt |}t ||||d|dk(}t |||d|| }t|j|dttj|jdg|jdzzf|dd y) Nrr;rhr0rrrrr6rrrrr6rArz) rMrirrrDrr%rr4rr5r6)rUrAr6r}rrrbr9s rY%test_make_splrep_with_unequal_weightsz9_TestMakeSplrepBase.test_make_splrep_with_unequal_weightss KK1RUU7B ' FF1I KK1c!f %QQ!t'Z2GI!Q!tq'Bs1v&ceeaS#%%']23AT +rnc d}tjddtjzd}tj|}tj||fj \}}|j jdusJ|j jdusJtjddt|}tj||fj \}} |j jdusJt|||dd |d k( } t|||d d| } || | t|||d d| } || | t|||d d| } || | t|||d d| } || | y) Nct|j|dttj|jdg|j dzzf|ddy)Nrr0rrz)rr4rMrr5r6)r9rbs rYcheckzO_TestMakeSplrepBase.test_make_splrep_with_non_c_contiguous_input..checksG CEE3q6 * BEE#%%!ceeAg"67F /rnrr;rhFr0rr=rrrr) rMrirrrrrrrDrr%) rUrArr}rrrrw1rrbr9s rY,test_make_splrep_with_non_c_contiguous_inputz@_TestMakeSplrepBase.test_make_splrep_with_non_c_contiguous_inputs\ / KK1RUU7B ' FF1Iq!tBxx$$---xx$$--- KK1c!f %ad Axx$$---QQ!t'Z2GI"a17C c3!R17C c3!Q"7C c3"bB$!")+ c3rnc |tjddd}tjdtjz|z}|d|d}}d}d}tj|g|dzz|g|dzzz}t |||||||d| } t | tsJ| jjdd|dzzk(sJ| j|k(sJt| jd|dztj|g|dzzt| j|dz dtj|g|dzzy) Nrr0rhr;r9r=r)xbxer6rr4rrA) rMrirrrTr%r r r4rr6rrS) rUrAr6r}rrrrr4r9s rY%test_make_splrep_impl_no_optimizationz9_TestMakeSplrepBase.test_make_splrep_impl_no_optimizations/ KK1b ! FF1ruu9q= !1quB   HHbTQU^rda!en4 5!Q2"qtW>#w'''uu{{1~a!e,,,uuzzdqs RZZA%?@!fg B41q5>(BCrnrW)r3rrFr)rhrrr0rc ltjjd}tjdtj|j ddtj z|dz dtj zf}tj|tj|z}t||||jdk(\}}}t||||j} |dk(r|d k(r|jdk(s8t| j|d t| j|d| d z d yy) Nrrr;rrrrrrFr0r\rz)rMrTrrrrrrrrrAr%rr4r5 rUrWrrr}rr4r5r6r9s rY(test_make_splrep_matches_splrep_periodicz<_TestMakeSplrepBase.test_make_splrep_matches_splrep_periodicsii##C( EE!RWWS[[AIAE[BCQYN O FF1Iq !A)CE1a!Q!T\\:RAFt||z'A CEE15 1 CEE1Wqb1f:E :(Brnc  tjjd}tjdtj|j ddtj z|dz dtj zf}tj|tj|z}t||||jdk(\}}}t||||j|} t| j|d| dz d y) Nrrr;rrr)rrAr4r0r\rz)rMrTrrrrrrrrrAr%rr5rs rY"test_make_splrep_with_splrep_knotsz6_TestMakeSplrepBase.test_make_splrep_with_splrep_knotssii##C( EE!RWWS[[AIAE[BCQYN O FF1Iq !A)CE1a!Q!T\\Q?q1"q&z6rn)r(r)r*rArMrrrrcrrrrr thread_unsaferrrrrrrrr+rnrYrr sG 5Qn% [[S"45*6* 0 [[S"45&6& [[,,8 [[S),!-!(: [[YT(:; [[S/2 +3< + [[YT(:;+<+\ [[YT(:; [[S/2D3<D, [[S"34 [[S"56 ;75 ; [[S"34 [[S"567757rnrceZdZejj dgddZejj dgddZdZdZ dZ dZ d Z d Z d Zd Zd Zy)TestMakeSplrepr6rcddlm}|j\}}}tjdg|dzzddgzdg|dzzz}|||dddf|||}t |||||}dD]} t || || d y) NrFr0rDrBrr0rhrr\rz) scipy.interpolate._fitpack_reprorrrMrTrr) rUr6rr}rrr4ff_drs rYtest_fitpack_FzTestMakeSplrep.test_fitpack_Fs 7.."1a HHaS!A#Y#s+qc1Q3i7 8 a1d7Q1 %aAq!$ 6A AaD#a&u 5 6rncddlm}|j\}}}tjdg|dzzddgzdg|dzzz}tj |j dt}|||dddf||||}t||||||} t|||||} d D]B} t|| | | d tj| | | | d sBJy) Nrrr0rDrBrrGrrr\rz) r rrrMrTrQrrrrr) rUr6rr}rrr4rfwfw_dr rs rYtest_fitpack_F_with_weightsz*TestMakeSplrep.test_fitpack_F_with_weights s 7.."1a HHaS!A#Y#s+qc1Q3i7 8 IIaggaj . q!AtG*aA +q!Q1*aAq!$ @A BqE47 7{{3q647? ?? @rnc ddlmcm}tjj d}tj ddddtj|jddzddddf }t|d}}t|j||j}t||}|jd|d|zz dz k(sJt||d y) Nrrrrrr=r;r\rz)r  interpolaterrMrTrrrrrDrrrrrr)rUrrr4rWr6DD_denses rYtest_disc_matrixzTestMakeSplrep.test_disc_matrixs 76ii##E* EE!Q1bggckkqk&9:1*CEE>rncbd}|j\}}}tt||||d}t||||}t|||||}t |j |j dt |j |j d|j|jk(sJy)Nr=rr9)r4r6rr\rz)rrr$r%rr4r5r6)rUr6r}rrr4rrs rYrzTestMakeSplrep.test_with_knots8s .."1a 1Q/ 0 4q!qA.AqAa0 EGG%8 EGG%8zzUWW$$$rncd}|j||j}|dz}d}t|||d}|jjdd|dzzk(sJy)NrhrGr=r0rrr;)rQrRr%r4r)rUrVrWr}rr6r9s rYtest_no_internal_knotsz%TestMakeSplrep.test_no_internal_knotsEs]  IIarzzI * qD !Q!q)uu{{1~AaC(((rncd}|j||j}|dz}t||d}t||d}t |j |j dy)NrhrGr=rZr\rz)rQrRr%rrr5)rUrVrWr}rr9rs rYrzTestMakeSplrep.test_default_sNsV  IIarzzI * qD!Q!$"1a1-uwwU3rncd}tj|}|dz}tjt5}t ||dd}t ||dd}t|j|dttj|jdg|jdzzf|dddddtd k(sJy#1swYxYw) Nrr=rrrr0r.rzr;)rMrQrcrrrr%rr4rrr5r6rD)rUrWr}rrrbr9s rYrzTestMakeSplrep.test_s_too_smallWs  IIaL qD \\. ) 0QAe,Caa51C CEE3q6 * BEE#%%!ceeAg"67F 0  0 1v{{  0 0s A7CCcttjgdtj}tjgdtj}tjdg|jdztj}t t 5t |||dddddy#1swYyxYw) N)g4@gR9c@g ףp=e@gףp= Oi@gQm@gQo@g)\(p@gfffffq@gq= ףq@g(\Yr@g)\r@g(\r@gQs@gQ3s@g33333Gs@皙as@rghs@rG)gJ@gH@gLH@gffffffG@gE@g)\(D@gC@gB@g333333@@g<@g3333339@g5@g2@g,@g%@g@g @r?g5{2?rr;r)rr6r)rMrSrRrrIrKr%rs rYtest_issue_22704zTestMakeSplrep.test_issue_22704es JJ%-/JJ 8 JJ&(ZZ 1 JJy1771:-RZZ @ : & . 1Q" - . . .s B..B7c^d\}}|j||j}|dz}t|||}t|||d}|jjdk(sJ|jjdk(sJt||dd|zz z|d}|jjdk(sJy)Nrhr=rGr=rZrrr0)rQrRr%r5r") rUrVrWr6r}rr9rrs rYrzTestMakeSplrep.test_shapess1 IIarzzI * qD!Q!$AqA.uuzzQww||q   Aq1ac7{a48ww||q   rncd\}}|j||j}|dz}t||dd}t||dd}|jjdk(sJ|jjdk(sJ|j j d||zdzk(sJ|j j dd|dzzk(sJy)Nr!rGr=rrr0r;)rQrRr%r5r"r4r)rUrVrWr6r}rrrs rYtest_s0_vs_notzTestMakeSplrep.test_s0_vs_nots1 IIarzzI * qDAqA+AqA+ww||q   ww||q   ww}}Q1q519,,,ww}}Q1A;...rnN)r(r)r*rcrrr rrrrrrrrrr#r+rnrYrrs} [[S"45 66 6 [[S"45@6@ 0 ? %)4  .! /rnrc|eZdZdZej j dgddZej j dgddZdZ dZ ej j dd d gd Z ej j d gd ej j ddgdZ y)TestMakeSplrepPeriodicrr6rc|jddd}|jd}t|||d|j}|jj dd|dzzk(sJy)NrrhrFrr0rr;)rirr%rAr4r)rUr6rVr}rr9s rYrz-TestMakeSplrepPeriodic.test_no_internal_knotss[ KK2r " GGEN!Q!q$,,?uu{{1~AaC(((rncd}tjddtjz|}tj|tj|z}t |||d|j }t |||d|j }|jjdk(sJ|jjdk(sJ|jjd|d|zzk(sJy)Nrhrr;rr0) rMrirrrr%rAr5r"r4r)rUr6rWr}rrrs rYr#z%TestMakeSplrepPeriodic.test_s0_vs_nots  KK1RUU7A & FF1Iq !AqADLLAAqADLLAww||q   ww||q   ww}}Q1q1u9,,,rncTd}dd|jz}}|j|||dz|j}|j|}t ||d|j }t t|||dd |j||j|z}t ||d |j }t t|||dd d |j|z|j|dzz}t ||d|j }t t|||dd y)Nrhrr;r0rGrrrr\r]rr@rr=) rrirRrr%rArrr)rUrVrrrXr}rr9s rY test_periodic_with_periodic_dataz7TestMakeSplrepPeriodic.test_periodic_with_periodic_datas !BEE'1 KK1a!e2::K 6 FF1I!Q$ =a qt$? FF1Iq !!Q%>a qt$? bffQiK"&&)A+ %!Q$ =a qt$?rnc d}ddtjz}}tj|||dz}tj|}t t 5t ||d|jdddy#1swYyxYw)Nrhrr;r0rr)rMrrirrIrKr%rArs rY$test_periodic_with_non_periodic_dataz;TestMakeSplrepPeriodic.test_periodic_with_non_periodic_datasj !BEE'1 KK1a!e $ FF1I : & < 1dll ; < < TestMakeSplrepPeriodic.test_make_splrep_periodic_m_eq_2_k_eq_1s HHc3Z  HHc3Z $%EBHHc3Z t!Q!q*BQQ!a0 q5 CEE3q6 *ceeaS#%%']23A rnk_fp))r0-C6)r;r/)r=gAȘr\c @d}tjdd|}tjdtjz|z}|\}}t ||||d}d} t ||dddf|d||} t j5t jdtt| d d| |z ftj||z ff|d z} ddd| j} tjd d g} | | }t|d|dy#1swYGxYw) Nrhrr0r;rrrr7r@MbP?r?r2)rMrirrrr"rrrrr#rPr9rTr)rUr.rrWr}rr6rrbfp0splinerbsx_checky_checks rYtest_fperiodic_basic_fitz/TestMakeSplrepPeriodic.test_fperiodic_basic_fits  KK1a  FF1ruu9q= !2QQ!+1a4j#a&A;  $ $ & T  ! !(N ;&#C!G rvvrAv6F'GUSA T ZZ((C:&W+ GAJ/ T Ts ADDN) r(r)r*rArcrrrr#r)r+r-r7r+rnrYr%r%sG [[S"45)6) [[S"45 -6 -@"< [[S1e*-  .   [[V%QR [[S4&)0*S0rnr%ceZdZddefdZej jdgddZej jdgddZ dZ d Z d Z y ) TestMakeSplpreprhr=c|j||j|jz|z }|j||j |g}|||fS)NrG)rQrRrrr)rUrr6rVr}rs rY_get_xykzTestMakeSplprep._get_xyksK IIarzzI *RUU 2Q 6 VVAYq "!Qwrnrrrr1rc 8d\}}tj|tjz|z }tj|tj|g}ddddd}t ||\\}}}} t ||\} } t| | d t| j|d t|||k(sJtj|j} t| j| d t| | t|||d | d y) Nr!rrrrhr<rr\rzr0r)rMrQrrrrr&rr4rDrSrr5r ) rUrrr6r}r num_knotsr4r5u_r9rrs rYtest_simple_vs_splprepz&TestMakeSplprep.test_simple_vs_splpreps1 IIaL255 1 $ VVAYq "%u=  Q Aq2a1%Q 2E* qu-1v1%%%ZZ]__r. A01aa03% ArncT|j\}}}t|tsJtj|ddk(sJt ||\}}t tj ||\}}t|||t|d|ddt|dt|dk(sJt|d|dd|d|dk(sJtjt||tj|k(sJt||\}}t||dt|j|ddt|jj|dd|j|dk(sJ||jtj|k(sJttj ||\}}t||dt|j|ddt|jj|dd|j|dk(sJ||jtj|k(sJtt 5ttj |j|dddy#1swYyxYw)Nrr;r>rzr\r0)r;r rrMrrrSrrDrr&r4r5rr6rIrK) rUrrrrbrtck_au_ar9s rYtest_array_not_listz#TestMakeSplprep.test_array_not_lists(--/1a!T"""xx{1~"""aQRZZ]a0 s3Q'Aau53q6{c%(m+++Aau51vq!!!xxa &"((1+555a1%Q3U+uQxe4q6uua   1v||rxx{***bjjmq1Q3U+uQxe4q6uua   1v||rxx{*** : & / AA . / / /s *+JJ'c|jd|\}}}t|\}}t|||j|dy)NrhrrVr\rzr;r&rrrUrVr}rr6r9rs rYtest_default_s_is_zeroz&TestMakeSplprep.test_default_s_is_zero:s?--"-,1aaQA %8rnc|jd|\}}}t|d\}}t|d\}}t||dt|||j|dt|||j|d|j|jk(sJ|j j |j j k(sJy)NrhrGrr>r\rzr)r;r&rrrr5r) rUrVr}rr6ru_ispl_nu_ns rYtest_s_zero_vs_near_zeroz(TestMakeSplprep.test_s_zero_vs_near_zero@s--"-,1a!!q) s!!u- sSu-c BHHQKe<c BHHQKd;zzUZZ'''ww}} ---rnctjdt}tt5t |dddtt5t |ddddtt5t |ddddt |gd\}}t |gd\}}||jdk(sJt|||gd y#1swYxYw#1swYxYw#1swYpxYw) NrrGrr>rrr0rr\rz) rMrQrrIrKrr&rr)rUr}rbr@r9rs rYrzTestMakeSplprep.test_1DMs IIau % : &  AJ : & ! a  !: & # c " #1#&RqcT*Q1v||v%%%A%0   ! ! # #s# CC$3C0C!$C-0C9N r(r)r*rMr;rcrrrArErJrOrr+rnrYr9r9slqR  [[S"67A8A: [[S"67/8/B9 .1rnr9ceZdZddefdZej jdgddZej jdgddZ dZ d Z d Z y ) TestMakeSplprepPeriodicrhr=c|jdd|jz||j}|j||j |g}|||fS)Nrr;rG)rirrRrr)rUrWr6rVr}rs rYr;z TestMakeSplprepPeriodic._get_xykbsI KK1RUU7ARZZK 8 VVAYq "!Qwrnrrr\rr@c d}tjddtjz|}tj|tj|g}ddddd}t ||d\\}}}} t ||d \} } t| | d t| j||k(sJt| | t|||d | ddy)Nrhrr;rrrVr0rrrr\rzrr@r]) rMrirrrrr&rrDr4r ) rUrrWr}rr?r4r5r6r@r9rs rYrAz.TestMakeSplprepPeriodic.test_simple_vs_splprepgs  KK1RUU7A & VVAYq ""Bb9  QA. Aq2a1j9Q 2E*355zYq\))) A 81aa 8 ;" 0rnc|j\}}}t|tsJtj|ddk(sJt ||d\}}t tj ||d\}}t|||t|d|ddt|dt|dk(sJt|d|dD]\}} t|| d|d|dk(sJtjt||tj|k(sJt||d\} }t||d| j|dk(sJ| |jtj|k(sJttj ||d\} }t||d| j|dk(sJ| |jtj|k(sJtt5ttj |j|ddddy#1swYyxYw) Nrr;r0rrzr\rr)r;r rrMrrrSrrDrrr&r6rIrKr) rUrrrrbrrCrDrrr9s rYrEz+TestMakeSplprepPeriodic.test_array_not_lists--/1a!T"""xx{1~"""aQ'QRZZ]aQ7 s3Q'Aau53q6{c%(m+++#a&%(+ 0FB B / 01vq!!!xxa &"((1+555a1j9Q3U+uua   1v||rxx{***bjjmq*EQ3U+uua   1v||rxx{*** : & C AAz B C C Cs ,H77Ic|jd|\}}}t|d\}}t|||j|dy)NrhrWrVrrer\rzrHrIs rYrJz.TestMakeSplprepPeriodic.test_default_s_is_zerosB--"-,1aa4QA %8rnc<|jd|\}}}t|dd\}}t|dd\}}t||d|j|} t||| dt||| d d |j|jk(sJy) NrhrZrrrrr\rzrr@r])r;r&rrr) rUrVr}rr6rrLrMrNy_arrs rYrOz0TestMakeSplprepPeriodic.test_s_zero_vs_near_zeros--"-,1a!!q*= s!!ujA sSu- c E6c E4@zzUZZ'''rnc$tjddtjzd}tj|}t t 5t |ddddt t 5t|dddddt t 5t|dddddt|gd d\}}||jd k(sJt|||gd y#1swYxYw#1swYzxYw#1swY`xYw) Nrr;rr0rrrrr\rQrz) rMrirrrIrKrr&rr)rUr}r9rs rYrzTestMakeSplprepPeriodic.test_1Ds KK1RUU7A & FF1I : &  A1  : & 5 a 4 5: & 7 c: 6 7qcUJ?Q1v||v%%%A%0   5 5 7 7s$ C.2C:D.C7:DDNrRr+rnrYrTrT_smqR  [[S"780900 [[S"78C9C<9 (1rnrTc,eZdZdZdZddZdZdZy) BatchSplinec tj||d}|jdd|_|j d|jdDcgc] }|||fi|c}|_||_ycc}wNr9)rMmoveaxisr _batch_shaper_splines_axis)rUr}rrr3kwargsyis rYr zBatchSpline.__init__se KK4 $GGCRL;<99RQS;UVR200V  Ws A.c|jDcgc] }|| }}tj||j|jz}|jr!tj |d|j S|Scc}wra)rdrMrrcrrbre)rUr}r3rs rYrzBatchSpline.__call__sc%)]] 36VAY 3 3 JJq$++agg5 612r{{1b$**-?a? 4sA<Nc|jDcgc]}|j|||}}tj||jScc}wr)rdrrMrrc)rUrrXrr3rs rYrzBatchSpline.integratesE?C}} MVV  aK 0 M Mzz!T..// NsA ctj|}|jDcgc]}|j|c}|_|Scc}wr)rdeepcopyrdrrUrr#r3s rYrzBatchSpline.derivatives<mmD!)rrhr9rr0)r4r6r)rr3r)rMrTrrirbrrLr_rtestingassert_allcloserrr)rU make_splinerfrrqrrWrdomainr}rr6r4r#refs rY test_batchzTestBatch.test_batchsii##$45 Aq!  KK # # # KK 5)2t 4 / )A11q5!QrUHA$66AdQ!n,F!Q4T4V4!QHT+HH CKK 1j 1 ""3q63q62%%a(#*<*>!$cnnQ&7S ""3q63q62 ""=3== =3==&;QRrnN) r(r)r*rcrrrrrryr+rnrYrprps [[2 b !  $ % . Hi0 1 Hd+ ,    [[\+;< [[VZ0S1= Srnrprrrr)}rLrrrrr|rrrnumpyrMscipy._lib._array_apirrrrrrr r scipy._external.array_api_extra _externalarray_api_extrarrcr rIr"r r rrrrrrrrrrrrrrrrrrscipy.sparse.linalgsparser)scipy.interpolate._bsplinesrrrr r!r r"r#r$r%r&scipy.interpolate._fitpack_implr _fitpack_implrscipy._lib._utilr'scipy._lib._testutilsr(scipy.interpolate._ndbspliner)r*rXr+rSr,rrrr.rrrrrrr{rrarrrErrrrrrr$rr(r*rRrTrvr,rWr|rrrQrwrrrrrrr%r9rTr_rpr+rnrYrsK    .-*  !!EEBGG//&952-&;;//KK117t )t )t )nG]+,W$W$-W$t6,$ G> $b+$_)_)D%&`?`?'`?F 88!++11(Y*Z  (nLnLb* ()J3J3*J3^  7 :":"z9j4j4j4ZWGWGta a L(.>"ll#l^2,,^Q7Q7h;R/(R/ R/j;\00\0 \0~< h1h1!h1V< b1b1!b1J8$S$Srn