-iifWddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z mZmZmZmZddlmcmZddlmZddlZddlmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*ddl+m,Z-ddl.m/cm,Z0ddl1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8m9Z9ddlm:Z:m;Z;mcm?Z@ddlAmBZBdd lCmDZDdd lEmFZFdd lmGZHdd lmIZJdd lmKZKejLjMZMeeGddZNeeGddZOdZPdZQdZRdZSdZTe fdZUdSdZVdde fdZWdZXGddZYeeGdd ZZd!Z[dTd"Z\ejL]d#d$d%gZ^eeGd&d'Z_Gd(d)Z`dUd+Zad,Zbd-ZcGd.d/Zdd0Zeee)Gd1d2Zfd3Zgd4Zhd5ZiGd6d7Zjee(Gd8d9ZkGd:d;ZlGd<d=Zmd>Znd?Zoee:Gd@dAZpdBZqGdCdDZrGdEdFZsee;GdGdHesZtee;GdIdJesZuee<GdKdLZvee<GdMdNZwGdOdPZxGdQdRZydS)VN)xp_assert_equalxp_assert_closexp_default_dtype concat_1dmake_xp_test_casexp_ravel)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) _dfitpack) _bsplines)_dierckxcVeZdZdZdZdZdZdZdZe dd e d d e d d dZ dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZdZdZd Zd!Zd"Zd#Zd$Z d%Z!e"j#$d&e%d'd(d)Z&d*Z'd+Z(e"j#$d,gd-e"j#$d.e%d/d0Z)d1Z*d2Z+d3Z,e"j#$d4gd5d6Z-e"j#$d4gd5d7Z.d8Z/d9Z0d:Z1e"j#2e3j4d;kdZ6d?S)@ TestBSplinec @tttftfit ddgdgdt jd5tttfit dt jgdgddddn #1swxYwYtttfit dt jgdgdtttfit ddgdgdtttfit dgdggdgdtttfit gd dgdtttfit gd ddgd tttfit gd gd dtttfit gd gd dtttfit gdgdd d\}}| ||zdz|j }| t j |}t|||}t||jt||j||jksJdS)N??rtckignore)invalidrr.rr.r9r9)r=r0@@@)r0r0r0cubic@)r=rr.r.r9r;)r0r.r. r;dtype) assert_raises TypeError ValueErrorr dictnperrstatenaninfarangefloat64asarrayrandomrr2r3r4)selfxpnr4r2r3bs k/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/scipy/interpolate/tests/test_bsplines.py test_ctorzTestBSpline.test_ctor4s:y*-w 1 1!SbTQ/// 1 1 1 [ * * * S S *g R RBF tq1Q1Q1Q R R R S S S S S S S S S S S S S S Sj'NNTQKB41-M-M-MNNNj'JJTQGtq-I-I-IJJJj'MMTaS1#J2$!-L-L-LMMM j'KKTIII!-J-J-JKKKj' < <RH::: < < < i O O111\\\WMMM O O Oi K K111\\\SIII K K K j' B B,,, a@@@ B B B1 IIac!e2:I . . JJry''** + + Aq!  1313ACxxxxxxs6B  BBc@t}|j}t|j|dddt|j|ddd|j|dksJt jt5d|_ddddS#1swxYwYdS)NrV瞯<atolrtolr.r9foo) _make_random_splinetckrr2r3r4pytestr AttributeError)rSrVr`s rWtest_tckzTestBSpline.test_tckWs  ! !eSV%e<<<<SV%e<<<<sc!f}}}}]> * *  AE                  s>BBBcHt|gd|ddgd}|ddd}||dk|d|j d }t |||dS) N)rr0r9r?r<rr1r9 r0rEr@)r rQlinspacewhererPr)rSrTrVxxexpecteds rWtest_call_namespacezTestBSpline.test_call_namespacecs bjj,, B70C0Cq I I I [[Ar " "88BGRZZ"*Z%E%EsKK"x(((((cFtjddd}tddgdgd}t||tj|dztgdddgd}t||tj|d kdd dS) Nrr.rer?r1)rffffff?r.r;r<rmr@)rKrfr r ones_likerg)rSrhrVs rW test_degree_0zTestBSpline.test_degree_0ls [Ar " " q!f * * *"r|B//#5666 lllq!f 2 2 2"rxT 3<<=====rkc f|gd}|gd}d}t|||}|ddd|j}t |||dt || z|dt |dz | zz|d t |d z | zzd t tj|||f\}}} t||| |f} t |||| d dS) Nr:)r0r9r;r.r0r?2rErrTr9+=r\) rQr rfrPrB_012maprKr) rSrTr2r3r4rVxx_npt_npc_np splev_results rW test_degree_1zTestBSpline.test_degree_1us: JJ ' ' JJ{{{ # #  Aq!   KKS"BJK 7 7 AaDD aDqR 1Q4acb(9(9(9#9 9AaDqsrARARARr?r@)r7r.gr>reT extrapolatersrt)rQr reshaper rfr)rSrTr4r2r3bpbsplrhs rWtest_bernsteinzTestBSpline.test_bernsteins  JJsAaCyA3!9, - - JJ''' ( ( 2::a))2::q!f+=+= > >q!Q [[b" % %24000RT222 @ @ @ @ @ @rkz dask.arrayz"_naive_eval is not dask-compatiblereasonz jax.numpyztoo slow; XXX a slow-if marker?torchz OOB on CIc t}|j\    dz d}||} fd|D}t ||d fd|D}t ||ddS)Nrrr.rqc 8g|]}t|Srr) _naive_eval.0rwr3r4r2rTs rW z4TestBSpline.test_rndm_naive_eval..s,CCC1 Aq!Q2666CCCrkrsrtc 8g|]}t|Sr) _naive_eval_2rs rWrz4TestBSpline.test_rndm_naive_eval..s,FFFaq!Qb999FFFrk)r_r`rfstackr) rSrTrVrhy_by_ny_n2r3r4r2s ` @@@rWtest_rndm_naive_evalz TestBSpline.test_rndm_naive_evals 2 & & &%1a [[1q!Aw + +aeehhCCCCCCCCCCDDSu----xxFFFFFFF2FFFGGT......rkc t}|j\}}}tj|||| dz d}t ||t ||||fddS)Nr.rqrsrt)r_r`rKrfrrrSrVr2r3r4rhs rWtest_rndm_splevzTestBSpline.test_rndm_splevsn  ! !%1a [1q!Aw + +"uR!Q33%@@@@@@rkctjd}tj|d}|d}t ||}t |}|j|j}}tj|||| dz d}t||t||ddS)Nr.Prsrt) rKrR RandomStatesortrr r2r4rfrr) rSrngrwyr`rVr2r4rhs rWtest_rndm_splrepzTestBSpline.test_rndm_splrepsi##D)) GCJJrNN # # JJrNNQll SMsAC1 [1q!Aw + +"uR~~E::::::rkc@t|}||j|_||j|j|j|j dz d|j}t||||dS)Nrrr.drE)r_rnr3rfr2r4rPr)rSrTrVrhs rWtest_rndm_unityzTestBSpline.test_rndm_unitys 2 & & &ll13 [[QS13tAv;2:[ F F"r||B//00000rkctjd}d\}}tj||}||ddf}t |j||f\}}t |||}|||| dz } }|| |z ||dzz} || jdksJdS) Nrr;sizer.r;r<)r;r<rrr)rKrRrrrvrQr shape) rSrTrrUr4r2r3rVtmtprhs rWtest_vectorizationzTestBSpline.test_vectorizationsi##D))1 GCJJqMM " " JJQ1IJ & &2:1v&&1 Aq!  1q!AwB 27bjjI)>)>??? ?quu{o------rkc tjd}d\}}tj|||zdz}||}tj|||dzf}t |||t |||}}|d|dz } tj|d| z |d| zd} t|| || dt|| t| |||fdt|| t| |||fddS) Nr)!r;r.r7rrqrsrt) rKrRrrr_r rfrr) rSrrUr4r2r3c_padrVb_paddtrhs rW test_len_czTestBSpline.test_len_csOi##D))1 GCJJqs1u%% & & JJqMMaAaC()1a##WQq%9%95 rUQqT\ [1AbEBJ 3 3"uuRyyu5555"uR!Q33%@@@@"uR!UA77eDDDDDDrkc t}|j\}}}|||| dz }}dD]3}t|||g|||dz|dz g|dd4dS)Nr.TF绽|=& .>Hz>r[r_r`r) rSnum_parallel_threadsrVr2_r4rrextraps rWtest_endpointszTestBSpline.test_endpointss  ! !%1a1q!AwB# W WF AAr2h//ArEz2:6??dQU W W W W W W Wrkc t}|j\}}}t|||dz| dz dz |||dz| dz dzddS)Nr.rrrtr)rSrrVr2rr4s rWtest_continuityzTestBSpline.test_continuitys  ! !%1a!AaC1H+-..!AaC1H+2E0F0F      rkct|}|j\}}}|d|dz }||||z || dz |zd}|||k||| dz kz}t|||d|||dt t j|||f\} } } |t| | | |fd } t||d| dS) Nrrr7rr.rqTrF)ext)r_r`rfrrvrKrQr) rSrTrVr2r3r4rrhmaskxx_npryrzr{s rW test_extrapzTestBSpline.test_extraps% 2 & & &%1a rUQqT\ [[1AqbdGbL" 5 5!r b1aRT7l+ "T(555"T(666 8 8 8  RAJ77tTzz%dAA"F"F"FGG "$///>>>>>rkct}|j\}}}|ddz |ddzg}||}tjtj|rJdS)Nrr.r7)r_r`rKallisnan)rSrVr2rr4rhyys rWtest_default_extrapzTestBSpline.test_default_extrapse  ! !%1adQh" " QrUU6"(2,,'''''''rkc tjd}tj|d}|d}t |j||f\}}d}t |||d}|jd|dzz }|d |dz }||||z |||zd } ||| ||z ||||z zz} t tj| ||f\} } } t||j krd nd }t|| |t| | | |f| |gd} ||| ||z ||||z zz} t|| d|| dt||j krdnd dS)Nrr<r;periodicrrr.r7rq-q=gH׊>rt)r7r?r.TrsgƠ>) rKrRrrrvrQr rrfrrPrr)rSrTrr2r3r4rVrUrrhxyxy_npryrzr\s rWtest_periodic_extrapz TestBSpline.test_periodic_extrapsi##D)) GCJJqMM " " JJqMM2:1v&&1  Aq! 4 4 4 GAJ!a%  rUQqT\ [[1AaD2Ir 2 2 qTR!A$Y1Q4!A$;/ / RAJ77tT(,, ::uu AbEE2::eED$?;;<<4    ZZ ( ( qTR!A$Y1Q4!A$;/ / Abj ) ) ) Abd # # #*2.."*<<$      rkct}|j\}}}tj|||f}t j|||| d}t ||||dddS)Nrrsr[)r_r`r from_splinerKrfr)rSrVr2r3r4pprhs rW test_ppolyzTestBSpline.test_ppolysz  ! !%1a  1ay ) ) [1q!uc * *"rr"vvE>>>>>>rkct}|j\}}}tj|d|dd}tj||f}t d|dzD]4}t ||||f|}t||||d5t|||dztj|ddS) Nrr7rqr.dernursrt) r_r`rKrfrrangerr zeros_like)rSrVr2r3r4rhryds rWtest_derivative_rndmz TestBSpline.test_derivative_rndm#s  ! !%1a [1qub ) ) U2q5\AaC== ; ;CrAq!9#...B B" E : : : : : "1 r}R'8'8uEEEEEErkc8d}gd}tjd}tjdd|dddf}t |||}tjgd}t |||dkdz |||dkdztj|d |d rJtjd d g}t ||dz d ||dzd tjd dg}tj||dz d ||dzd rJtj||dz d||dzdrJdS)Nr9) r7r7rr.r.r;r<rrrrrrrr)r.r;r<rrrg2H@gη@r;r<r.r)rKrRrrr rQrallclose) rSr4r2rr3rVrwx0x1s rWtest_derivative_jumpsz!TestBSpline.test_derivative_jumps0s  2 2 2i##D)) E!Q 1 q!+ , Aq!   J||| $ $!AF)e+,,!AF)e+,, . . .;qq{{AAgJJ77777ZA  "u*+++"u*+++ - - - ZA  ;qqe222AAb5jQ4G4G4GHHHHH;qqUq11111QY13E3E3EFFFFFFFrkc 0|ddd}tj|gd}t t j||j|jf\}}}|t||||j f}t|||dt||j krdnd}t|||t||j |tj|gd }|d d d |j }t||||dk||zd|z d zddS)Nr7r<rrr.r9r;)r2rsrtrrErr.r.r9rr9rer.r>)rfr basis_elementrQrvrKr2r3rr4rrrPB_0123rg) rSrTrhrVrryrzr{r\s rWtest_basis_element_quadraticz(TestBSpline.test_basis_element_quadraticJs [[Q # #  !BJJ|||$<$< = = = RacN;;tTzz%dAC/@"A"ABB "|%8888&r**bj88UUd"rzz&**BJzGGdSSSS  !BJJ|||$<$< = = = [[Ar[ 4 4"aBB 33% A A A A A Arkc t}|j\}}}tj|||| dz d}t ||t ||||ddS)Nr.rrsrt)r_r`rKrfr_sum_basis_elementsrs rWtest_basis_element_rndmz#TestBSpline.test_basis_element_rndmZsm  ! !%1a [1q!Aw + +"22q!Q??eLLLLLLrkct}|j\}}}|dz}t|||}t||jj|}t||jj|}t j|||| dz d}t||j||dt||j||ddS)Ny?@r.rrsrt) r_r`r r3realimagrKrfr) rSrVr2r3r4ccb_reb_imrhs rW test_cmplxzTestBSpline.test_cmplx`s  ! !%1a (^ Ar1  q!#(A&&q!#(A&& [1q!Aw + +" DDHH59999" DDHH5999999rkctj|gd}|||jsJdS)Nr)r rrQrrM)rSrTrVs rWtest_nanzTestBSpline.test_nanmsJ  !"**\\\":": ; ;xx"& """""""rkcHtd|}|j\}}}t|||}||||| dz d}t d|D]<}|}t |||||dd=dS)Nr)r4rTr.rrr[)r_r`r rfr derivativer) rSrTrVr2r3r4b0rhjs rWtest_derivative_methodz"TestBSpline.test_derivative_methodrs ! + + +%1a Q1   [[1q!Aw + +q! F FA A BBr1IIqquu5u E E E E E F Frkc6t|}|j\}}}||||| dz d}t||||dd||||fd}|||fd}t|||}t||||dddS)Nrrr.rrsr[axisr9)r_r`rfrantiderivativerrr )rSrTrVr2r3r4rhs rWtest_antiderivative_methodz&TestBSpline.test_antiderivative_method{s) 2 & & &%1a [[1q!Aw + +7((**5577;;"E 7 7 7 7 HHaAYQH ' ' HHaV!H $ $ Aq!  7((**5577;;"E 7 7 7 7 7 7rkc D tj|gd}tj|ddddsJtj|ddddsJtj|ddddsJtj|ddddsJtj|ddddsJtj|ddddsJtj|ddd dsJtj|ddd d dsJtj|ddd ddsJtj|ddd ddsJtj|ddd t jdd|jdsJd |_ | }||d|dz |j }tj|dd|sJtj|ddd|zsJtj|dd|sJtj|ddd|zsJt|dd||d|dz t|dd||d|dz |dz|dz t|dd||d|dz |dz|dz t|dd||d|dz |dz|dz d|zzt|dd||d|dz t|dd||d|dz t|dd||d|dz d|zz dS)Nr8rr.rrsabs_tolr7r=TrFrr9rEiii?r;g+@rir<) r rrQmathisclose integrate_implrr`rrrPr)rSrTrVi period_ints rW test_integralzTestBSpline.test_integralsj  !"**YYY"7"7 8 8|AKK1--sEBBBBBB|AKK1--xGGGGGG|AKK1--tUCCCCCC|AKK1--sEBBBBBB|AKK1--xGGGGGG|AKK1--tUCCCCCC|AKKA..UCCCCCC|AKKA4K@@#uUUUUUU|AKKA5KAA3PUVVVVVV|AKK25KAA4QVWWWWWW|AKK25KAA!LB66GGG G G G#    ZZ!qqtt 2:Z>> |AKK1--z:::::|AKK1--rJ?????|AKKB//<<<<<|AKKB//Z@@@@@ C-- 11S66AAcFF?33 5 5 5 C++ 11Q44!!A$$;1#5##>?? A A A Hf55 11Q44!!A$$;1#5##>?? A A A C00 11Q44!!A$$;1#5##>Z#OPP R R R  Ar**BJJqqttaadd{,C,CDDD B,,bjj1!.E.EFFF Ar** 11Q44!!A$$;Z#?@@ B B B B Brkcgd}t||}d|_tj|}dD]=\}}t ||||||>dS)Nr:r))r)rr)r  )r rr rrr)rSrwrVprrs rWtest_integrate_ppolyz TestBSpline.test_integrate_ppolys OO q! $ $"  a 5 1 1FB AKKB//KKB// 1 1 1 1 1 1rkctjgd}dD]C}|dd|}t|tjsJ|jdksJDdS)Nr8rrr.r)r rr isinstancerKndarrayndim)rSrVrress rWtest_integrate_0D_alwaysz$TestBSpline.test_integrate_0D_alwaysso  !))) , ,( ! !K++a +<.BN)__name__ __module__ __qualname__rkrWBr s Drkr%)rr.r9r9)r r __class__rr)rSr%rVs rWtest_subclassingzTestBSpline.test_subclassings         OOLLL ) ){a||~~'1,,,,!!+q000000rkrr r<c  d\}}|dd||zdz}gd}|dz}|||t|}tjd}|||} t|| ||} | jj ||f|d|z||dzdzksJ|d }| |j |d||j z||dzdzksJ| j dz | j fD]+} tttfit|| || ,t|| ||t|| ||d t|| ||t|| ||d fD]} | j| jksJdS) Nrrr.rrrr<rrrr)r2r3r4rr9)rfrtuplerKrRrrQr r3rrrGr%rJrrr) rSrrTrUr4r2shpos_axisrr3rVaxb1s rW test_axiszTestBSpline.test_axiss?1 KK1a!eai ( ( YY!8 (A 2YYi##D)) JJszzrz** + + Aq!$ ' ' 'syR\Ob(m;b!oMMMMM ZZ " "quu{b(mbh6HQJKKHHHHHF7Q;' : :B )W : : 1QR888 : : : :1a...99;;1a...99!<<1a...==??1a...==a@@B % %B7af$$$$$  % %rkc d}|gd}|gdgdg}t|||d}t||dddf|}t||dddf|}t|d ||d |d gdS) Nr9)rr.r9r;r<rr)r7r9rr7)r9rr.r7rrr.rB)rQr rr)rSrTr4r2r3splspl0spl1s rW test_neg_axiszTestBSpline.test_neg_axiss  JJ,,, - - JJ 6 7 7aAB'''q!AqqqD'1%%q!AqqqD'1%%C"((DDIIttCyy+A"B"BCCCCCrkcd}d}d}dD]}||||tdddD]}|||dd}tdd D]}|||d d S) 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). cdtjd}tj||dzdz }||dzdz }|dkr |d|d<t ||||}tjt|j|z dz }t|j|||}tj ||j| } t| |j z|d t|| d d S) zY To avoid repetition of code the following function is provided. r(rrr7rr4bc_typer.rsrtN)rKrRrr random_sampler eyelenr2r design_matrixtoarrayrr3) rUr4r:rrwrrr3 des_matr_def des_matr_csrs rWrun_design_matrix_testszHTestBSpline.test_design_matrix_bc_types..run_design_matrix_testss))''--C))!,,r1B677A!!!$$r)B.A*$$u!%aaAAADs46{{Q*++A07461a0033L"01512444;GII  L46115 A A A A L,U C C C C C CrkrDr;clampednaturalrr9 not-a-knotrrrN)r)rSrBrUr4bcs rWtest_design_matrix_bc_typesz'TestBSpline.test_design_matrix_bc_typess D D D*  ( . .B # #Aq" - - - -q!Q 8 8A # #Aq, 7 7 7 7 q! 6 6A # #Aq* 5 5 5 5 6 6rkr)FTrdegreerc tjd}|d|dzz}tj|tj|}}|}tjtj|dz |dz |tj||d|dzztj|dz|dz|f}tjt||z dz } t|| ||} t| |tj |||| tj|dz |dz |dz|dzg}|sKtjt"5tj ||||ddddS#1swxYwYdSt| |tj |||| dS)z;Test that design_matrix(x) is equivalent to BSpline(..)(x).rrer.r9r N)rKrRrr;aminamaxrrfr<r=r rr>r?arrayrar rI) rSrrIrrwxminxmaxr4r2r3bsplines rW'test_design_matrix_same_as_BSpline_callz3TestBSpline.test_design_matrix_same_as_BSpline_call%s#i##D))   bFQJ/ 0 0WQZZd  E"+dQhq&99+dD!vz*:;;+dQhq&99: ; F3q66A:> " "!Q;// GAJJ-aA{CCKKMM   HdRi4#:tbyA B B z** < <%aA{;;; < < < < < < < < < < < < < < < < < <  %aA{;;CCEE     sF%%F),F)ctjd}d}d}tj||dzdz }||dzdz }t |||}t ddD]]}|d|}|d|} tj||j | } t| |j z| d ^dS) Nrrer;r8rr4r.r<rsrt) rKrRrrr;r rr r>r2r?rr3) rSrrUr4rwrrrxcycrAs rWtest_design_matrix_x_shapesz'TestBSpline.test_design_matrix_x_shapesAsi##D))   GC%%a((2-2 3 3   a 2 % *!!Q!,,,q! C CA2A2B2A2B"01512444;GII  L4612E B B B B B  C Crkcgd}tjd|d}t|gdgddS)N)r0r0r0r>r?r@r@r@r>r;)g?gmਪ?gK}\UU?r=rsrt)r r>r?r)rSr2des_matrs rWtest_design_matrix_t_shapesz'TestBSpline.test_design_matrix_t_shapesRs_ , , ,(Q22::<<;;;<" $ $ $ $ $ $rkc:tjd}d}d}tj||dzdz }||dzdz }t |||}t t5tj ||j ddd|dddn #1swxYwYd}gd }gd }t t5tj |||ddddS#1swxYwYdS) Nrrer;r8rrSr7r9)r=r0r>r?r@@r~) rKrRrrr;r rGrIr r>r2)rSrrUr4rwrrr2s rWtest_design_matrix_assertsz&TestBSpline.test_design_matrix_assertsZsi##D))   GC%%a((2-2 3 3   a 2 % *!!Q!,,,: & & 6 6  !!TVDDbD\1 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6  $ $ $    : & & + +  !!Q * * * + + + + + + + + + + + + + + + + + +s$%CC C,DDDr:)rErDrrFctjd}tj|d}|d}|dkr |d|d<t |||}t j||}tjddd}t||||d t|||}t|j |j d dS) Nrrrrr7r:r.rZrt) rKrRrrrr from_power_basisrfrr r3) rSr:rrwrcbrrhbspl_news rWtest_from_power_basisz!TestBSpline.test_from_power_basislsi##D)) GCJJrNN # # JJrNN j aDAbE Aw / / /'G<<< [Ar " "2Ru5555%aG<<< 777777rkc tjd}tj|d}|d|ddzz}|dkr |d|d<t |||}t j||}t||j|}t||j |}t|j |j d|j zzd dS) Nrrr/rrr7r^rZrt) rKrRrrrr r_r rrrr3) rSr:rrwrr`r bspl_new_real bspl_new_imags rWtest_from_power_basis_complexz)TestBSpline.test_from_power_basis_complex|si##D)) GCJJrNN # # JJrNNSZZ^^b0 0 j aDAbE Aw / / /'G<<<*1afgFFF *1afgFFF  "}2F FUSSSSSSrkctjgd}tjgd}tjt ||dd}t |jgdddS) 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:)r.r.r.r.r.rEr^)r0r.r.r.r.r.r.rZrtN)rKrMr r_rrr3)rSrwrrs rWtest_from_power_basis_exmpz&TestBSpline.test_from_power_basis_exmpsy H___ % % H___ % %' Aq)(L(L(L09;;; 7 7 7eDDDDDDrkctjddg}tjdg}|d|dtjddd}|dt ||d}t ||tj|dzdS)Nrr.r?Fwriterer1)rKrMsetflagsrfr rrn)rSr2r3rhrVs rWtest_read_onlyzTestBSpline.test_read_onlys HaV   HcUOO     [Ar " " %    a1 " " ""r|B//#566666rkcTt}fd}td||dS)Nrrc|j\}}}|||| dz d}||dS)Nr.i')r`rf)rrVr2r4rhrTs rW worker_fnz/TestBSpline.test_concurrency..worker_fnsDeGAq!QqT1aRT7E22B AbEEEEErkre)r_r&)rSrTrVrps ` rWtest_concurrencyzTestBSpline.test_concurrencysF 2 & & &      Iq11111rkcygwinz'threading.get_native_id not implemented)rr c Ht}tjddd}||}tj}tjt |d|dd|jj |jj }|j|dd<tjt |d|dd|j j |j j }|j |dd<||_||_ t|||dS) Nrr.rer2z.datzw+)moderFrr3) r_rKrf threading get_native_idmemmapstrjoinr2rFrr3r)rStmpdirrVrhritidt_mmc_mms rW test_memmapzTestBSpline.test_memmaps  ! ! [Ar " "1R55%''yV[[S7788t sy ;;;#QQQyV[[S7788t sy ;;;#QQQ"x(((((rkN)7r!r"r#rXrcrjror|rskip_xp_backendsrrrrrrrrrrrrrrrrrrrrrrrr'ramark parametrizerr/r5rHrQrVrYr\rbrfrhrmrqxfailsysplatformrbr~r$rkrWr,r,1s=!!!F   )))>>>DDD @ @ @l+OPPPk*KLLLgk222 / /32MLQP /AAA ; ; ;111 . . .EEE&WWW??? (((   6??? F F FGGG4AAA MMM : : :### FFF 7 7 7+B+B+BZ 1 1 1!!!111 [VUU2q\\22%%32%:DDD(6(6(6T [],E,E,EFF [XuuQxx0010GF4CCC"$$$+++$ [Y)C)C)CDD 8 8DD 8 [Y)C)C)CDD T TDD TEEE& 7 7 7 2 2 2 [  8 ))  )))rkr,cPeZdZejdgddZejdgddZdZejdgd d Z ejd d d gdZ dZ dZ d S) TestInsertxval)r=r0rBr<@@c X|d|j}||dz}t||d}|j|j|jf}tt||}| |}t|j | |j dt|j | |j d|j dz d||dkr|n |dd} || d |dd|ddzzf} t|| || d||dz} t|| d} t|||| fd d} | |} t| j |j dt| j || |j | |j fd d||dkr|n |dd} || d |dd|ddzzf} t| | | | ddS) NrrEr;rSrZrtr.r7rr)rOrPsinr _t_cr4r r insert_knotrr2rQr3concatcosr)rSrrTrwrr2r`spl_1fspl_1rhy1spl_y1spl_yyspl_yy1s rW test_insertzTestInsert.test_inserts IIarzI * * FF1IIqL A+++vsvsu%&s++,%%FH!5!5EBBBBFHYsufQhY,?!@!@uMMMM!B%--QQQssV YYC1223B301 2 2Br7777VVAYY\#ArQ///#ArxxBax'@'@AFFF$$T** 577777 I HHcood++-v/A/A$/G/G/IJQRH S S    !B%--QQQssV YYC1223B301 2 2r GGBKKe<<<<<!DDDEa((FH!5!5EBBBBFHYsufQhY,?!@!@uMMMM!B%--QQQssV YYC1223B301 2 2Br777777rkctjd}d\}}|tj|||zdz}|||ddf}t |||}||||dz|| dz }||} ||||dz|| dz d } t|| | | d dS) N90rCr.rr;r9)lowhighrrrrrZrt) rKrR default_rngrQruniformr rr) rSrTrrUr4r2r3r2xvrrhs rWtest_insert_randomzTestInsert.test_insert_randoms+i##E**1 JJrws{{!A{6677 8 8 JJs{{Aq {22 3 3aA ZZ !A#Qr!tW == > >## ZZ !A#Qr!tW2 FF G GBr777777rkr) r皙?r>r@@@@gffffff@rc >|d|j}||dz}t||d\}}}t |j||f\}}t |||d}||} t||j d\} } }t| j || d t| j d| d z || d| d z d tjd d dd} || } t| | |t#| | | |fd dS)NrrEr;rSrrTperrZrtr.rrr)r)rOrPrrrvrQr rrr`rr2r3rKrRrrr)rSrrTrwrr2r3r4r2rtfcfrrhs rWtest_insert_periodiczTestInsert.test_insert_periodics~ IIarzI * * FF1IIqLA###1a2:1v&&1aA:666##2swD111 BBe<<<<!A 2er!te9(=(=EJJJJ %%d++333KK ZZ  b 2::eEBA;.G.G#H#HuUUUUUUrkrNrc|d|jdztjz}||||}}t ||d|zzd}||_t ||d}||_t ||d}||_d} || } || } || } t| j | j d t| j | j d t| j | j d| j zzd dS) NrrEr9r/r;rS @rZrt) rOrPrKpirrr rrrr2r3) rSrrTrwy_rey_imr2spl_respl_imrrspl_1respl_1ims rW test_complexzTestInsert.test_complex0s5 IIarzI * *Q . 6VVAYYq d D2d7Na888%#Atq111(#Atq111( ##$$R(($$R((77777777R \!9FFFFFFrkcd}d}tjdg|dzzgdzdg|dzzz}tjt||z dz }t |||d}t t 5t||||fd dddn #1swxYwYt t 5||ddddS#1swxYwYdS) Nrr;rr.r9r;r<rrrrTr) rKrMonesr=r rGrIrr)rSrr4r2r3r2s rW+test_insert_periodic_too_few_internal_knotsz6TestInsert.test_insert_periodic_too_few_internal_knotsGsv  HaS!A#Y-QqS 9 : : GCFFQJN # #aA:666 : & & , , 21ayd + + + + , , , , , , , , , , , , , , ,: & &  OOB                     s$=BB#&B#>C!!C%(C%cZd}tjdg|dzzgdzdg|dzzz}tjt||z dz }t |||}t t 5|ddddn #1swxYwYt t 5|ddddn #1swxYwYt t 5|ddddddS#1swxYwYdS) Nr;rr.rrr7rr)rKrMrr=r rGrIr)rSr4r2r3r2s rWtest_insert_no_extrapz TestInsert.test_insert_no_extrap[s  HaS!A#Y-QqS 9 : : GCFFQJN # #aA : & &  OOB                  : & &   OOA                  : & & $ $ OOAO # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s69BB"B:CC #C ;D  D$'D$) r!r"r#rarrrrrrrrrr$rkrWrrs/ [V%A%A%ABB!=!=CB!=H [CCC 8 8 8 8 8 8 [T$8$8$8VVV" []T:,>??GG@?G,   ( $ $ $ $ $rkrc dd}dD]e}t|}tt|D]6\}}|||td|dzD]}||||dd7fdS) Nrrsc 4|j\}}}tj|}tj|ddz d|dd|ddzz|ddzf}t t ||||f||||||d|d|jdS) Nrrrr.r7zder = z k = )r\r]err_msg)r`rKuniquerrrr4) rVrrr\r]r2r3r4rws rW check_splevz,test_knots_multiplicity..check_splevos%1a IaLL E!A$s(C1222A2/2s: ;a!QC00!!As)).Gs.G.G!#.G.G I I I I I Irkr.r9r;r<rrSr.r)rrsrs)r_ enumerate_make_multiplesr)rr4rVrr.rs rWtest_knots_multiplicityrksIIII66 ! $ $ $q1122 6 6EAr KA   Q!}} 6 6 B3u5555 6 666rkc|dkr#|||cxkr||dzkrnndndS|||z||krd}n6|||z |||z||z z t||dz ||z}|||zdz||dzkrd}nE|||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]) rr.r0r=_naive_Brwr4rr2c1c2s rWrrs  Avvda((((!AaC&(((((ssc11v1~~ !A$h1Q3!A$ '(1ac1a*@*@ @1Qx1QqS6 !AhlQqs1uX!A#. /(1ac1Q32J2J J Grkc4krn|dz cxkrdzksnJkrjdz ksJtfdtddzDS)z= Naive B-spline evaluation. Useful only for testing! r.rc3ZK|]%}|z t|z zV&dSNr)rrr3rr4r2rws rW z_naive_eval..sBFF1q1vAqsA...FFFFFFrk) searchsortedrsumr)rwr2r3r4rTrs```` @rWrrs AaDyy  OOAq ! !A % Q41    !A#       66a!'!*q.(((( FFFFFFFFa1 FFF F FFrkcjddzz }|dzksJjd|ksJcxkr |ksnJtfdt|DS)z'Naive B-spline evaluation, another way.rr.c3NK|]}|t|zV dSrrrrr3r4r2rws rWrz _naive_eval_2..s:==qqthq!Q***======rk)rrr)rwr2r3r4rTrUs```` rWrrs  acA !8888 71:???? Q41    !       =======E!HH=== = ==rkc.t||dzz }||dzksJ|jd|ksJd}t|D]O}tj||||zdzd|}|||t j|zz }P|S)Nr.rr=r9Fr)r=rrr rrK nan_to_num)rwr2r3r4rUsrrVs rWrrs A!A#A !8888 71:???? A 1XX%% @G !!Aac!eG*% @ @ @ C C QqTBM!$$ $$ Hrkctj|}tj||dk|dkz|dk|dkz|dk|dkzgdddg}||S)z+ A linear B-spline function B(x | 0, 1, 2).rr9r.cdSNr=r$rws rWzB_012..sbrkc|Srr$rs rWrzB_012..sArkc d|z SNr>r$rs rWrzB_012..s Ark)rK atleast_1d piecewiserQ)rwrTresults rWrurus aA \!q1uQ/!VA.!VQ/1)L++~~FHHF ::f  rkctj|}|dk|dk|dkz|dkg}|dkr dddg}n!|dkr ddd g}ntd |tj|||}|S) z0A quadratic B-spline function B(x | 0, 1, 2, 3).r.r9rc||zdz Srr$rs rWrzB_0123..s1Q3r6rkcd|dz dzz S)Ng?r r9r$rs rWrzB_0123..s41T6A+-rkcd|z dzdz S)Nr?r9r$rs rWrzB_0123..sBqD19q=rkcdSNr0r$rs rWrzB_0123..2rkcdS)Ngr$rs rWrzB_0123..s3rkcdSrr$rs rWrzB_0123..rrkznever be here: der=)rKrrIr)rwrcondsfuncspiecess rWrrs aA UQUq1u%q1u -E axx!!--((* 4s44555 \!UE * *F Mrk#r;cDtjd}tj|||zdz}||}||||}}t j|||S)N{r.)rKrRrrrQr construct_fast)rUr4rTrr2r3s rWr_r_s{ )   $ $C  1Q3q5!!""A 1 A ::a=="**Q--qA  !!Q * **rkc#K|j|j}}|j}|d|dd<|d|d<t |||V|j}|d|d|dz<t |||V|j}|d|| dz d<t |||VdS) zIncrease knot multiplicity.rrNr.r7)r3r4r2copyr )rVr3r4t1s rWrrs 3qA B2Br"uI VBrF "a   B!uBt!tH "a   B2Br!tuuI "a  rkcPeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) TestInteropc tjddtjzd}tj|}t ||}|j|j|jf|_|||c|_ |_ |_ tjddtjzd|_ tj |j|j|jf}tj||f|_t!|j|j|j|_dS)Nrr@rr)rKrfrrr r2r3r4r`rhrrVxnewc_dstackrr b2)rSrhrrVrs rW setup_methodzTestInterop.setup_methods [BruHb ) ) VBZZ r2 & &Cac?#%r1 $&K2be8R00 U13QS= !)RH%%!#tw,,rkc|j|j|jc}}tt ||ddtt |j|ddtt jfd|D|ddttd5t ||dddn #1swxYwYttd|j j dz}|j |}|j||jf}tt jt |||||dddS)NrZr[c0g|]}t|Sr$)r)rrwrVs rWrz*TestInterop.test_splev..s!#>#>#>AE!QKK#>#>#>rkzCalling splev.. with BSplinematchr.r)rrVrrrr`rKrQrGrIr*rr3r transposer2r4)rSrrr+rr`rVs @rW test_splevzTestInterop.test_splevsi a dA$e% 9 9 9 9dAE**$e% 9 9 9 9 #>#>#>#>#>#>#>??$e% 9 9 9 9 :-K L L L   $OOO               5BDI&& ' '$ . T^^B  tR 5s#3#3444**2..U H H H H H HsCC"%C"c&|j|j}}t||}tj||\}}}t |d|dt |d|d|d|ksJt||d\}}}}t |d|dt |d|d|d|ksJt ||} t || dt |} t || |ddS)NrrZrtr.r9T) full_output)rhrrrrrr ) rSrwrr`r2r3r4tck_frrrVs rW test_splrepzTestInterop.test_splreps)w1Qll,q!$$1aA....A....1v{{{{ 1$777q!Qa!%0000a!%0000Qx1}}}}1c]]2E**** SM11Q44e,,,,,,rkc|j|j}}tj||f}t t 5t ||dddn #1swxYwYt t 5tj||dddn #1swxYwYt td5t |dd|dddddn #1swxYwYt td5tj|dd|ddddddS#1swxYwYdS)Nm > k must holdrr;) rhrrKrrGrIrrrH)rSrwry2s rWtest_splrep_errorszTestInterop.test_splrep_errors/s*w1 U1a4[ : & &   1bMMM                : & &  LB                  9,= > > > ! ! 1RaR5!BQB%  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! 9,= > > > ' ' L2A2"1" & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'sGAAA3BBB6!C##C'*C'&D77D;>D;cftjdtjd}t |\}}t j|\}}t ||dt tjt|||dt tjt|||dt |dd\\}}}}}t ||dt tjt|||ddS) Nr rEr;rrZrtrT)rr ) rKrOrPrrrrrQr) rSrwrVur`u1b_fu_frs rW test_splprepzTestInterop.test_splprep?s Ib + + + 3 3F ; ;qzz1-""R 2E**** 5A;;//???? 5C==1115AAAA&a1$??? cAq!3U++++ 5c??33QUCCCCCCrkctjdd}ttd5t |dddn #1swxYwYttd5t j|dddn #1swxYwYtjddd}ttd 5t |gdddn #1swxYwYttd 5t j|gdddn #1swxYwYgd }ttd 5t |gdddn #1swxYwYttd 5t j|gdddn #1swxYwYgd }gd }ttd 5t |gd|gddddS#1swxYwYdS)N<rztoo many values to unpackrrr8r;)numr) >Ir >KrzInvalid inputs)r.r;r9r<)rg333333?g?r.) rKrOrrGrIrrrfrH)rSrwrs rWtest_splprep_errorszTestInterop.test_splprep_errorsNs Ie   $ $Y / / :-H I I I   AJJJ                :-H I I I   M!                   K21 % % % 9,= > > >   QCLLL                9,= > > >   M1#                   E D D :-= > > >   QCLLL                :-= > > >   M1#                   LL    :-= > > > % % qc4^ $ $ % % % % % % % % % % % % % % % % % %s}AA!A;BB #B C11C58C5D44D8;D8E66E:=E:F99F=F="G==HHc|j|j}}tjgdtjz}t t ||ddt t |j|j|j f|ddttd5t |ddddn #1swxYwY|j dd d }tj t |j||j fd}|jd ksJt ||z tj|d dS)N)rr rBrrr[zCalling sproot.. with BSplinerrq)mestr.r9r)r;r9r<rrt)rVrrKrMrrrr2r3r4rGrIr rQrr)rSrVrrootsc2rrrs rW test_sprootzTestInterop.test_sprootksn2---..ru4q 5t$????QS!#//TMMMM:-L M M M  2B                   dnnQ1%% Zc240r::: ; ;x9$$$$U BM"$5$5EBBBBBBsB88B<?B<c n|j|j}}ttdd|tdd|jddttdd||ddddt td5tdd|dddn #1swxYwY|j ddd}tj tdd|j ||j f}|jd ksJt|tdd|dd dS) Nrr.rsF)r\check_0dzCalling splint.. with BSplinerr9r;r9)r\ check_shape)rVrrrr`rrGrIr3r rKrQr2r4r)rSrVrr#integrs rW test_splintzTestInterop.test_splint|s2q!Qq!QU++%% I I I Iq!Q Aq)) G G G G:-L M M M   1a                  dnnQ1%%F1a"$RT):;;<<|v%%%%q!Qe H H H H H HsB//B36B3c |j|jfD]D}t|jt|jz }|j}|dkr7t j|t j|f|j ddzf}dD]}t|}tj |j||j f}t|j|ddt|j|dd|j |dksJt|tsJt|t sJFdSNrr.r.r9r;rZrtr9)rVrr=r2r3rrKrzerosrrrr4rrr r*rSrVctb_crUbdtck_ds rW test_splderzTestInterop.test_splders?&$'" 0 0AQSCHH$B#((**CAvveC2%#)ABB-*?!@!@@A 0 0AYY achhjj#qs%;<<eAhU;;;;eAhU;;;;tuQx''''!"g.....!%////// 0 0 0rkc |j|jfD]D}t|jt|jz }|j}|dkr7t j|t j|f|j ddzf}dD]}t|}tj |j||j f}t|j|ddt|j|dd|j |dksJt|tsJt|t sJFdSr-)rVrr=r2r3rrKrr/rrrr4rrr r*r0s rWtest_splantiderzTestInterop.test_splantiders@&$'" 0 0AQSCHH$B#((**CAvveC2%#)ABB-*?!@!@@A 0 0]](!#((**c13)?@@eAhU;;;;eAhU;;;;tuQx''''!"g.....!%////// 0 0 0rkc|j|j|j}}}|jjdz}d|j||j|dzzz}t ||t ||j|j|jf}}tt||t||dt|tsJt|tsJtt|jj}|j|dddz} t ||j| |jf} t ||} tt!jt|| ddd| |dt| tsJt| tsJdS)Nr9rr.rZrtrr)rVrrhr2rrr3r4rrrr r*rrr rKrQ) rSrVrrhrtnbntck_nr+rtck_n2bn2s rWrzTestInterop.test_insertsFDGTWr2 CHM !#a&13qs8# $2qMM6"qsACo#>#>Eb" b%((u 6 6 6 6"g&&&&&%'''''5## $ $ T^^BqrrFTM * *RT2rt,--Rnn  5V#4#455??1aHHBe - - - -#w'''''&%(((((((rkN)r!r"r#rr rrrrr%r+r5r7rr$rkrWrrs - - -HHH6---0''' D D D%%%:CCC"HHH(000"000")))))rkrceZdZejddejzZejeZdZ dZ dZ dZ e jdgdd Ze jdgdd Zd Zd Ze jdgd dZdZdZdZe jdgddZdZdZdZdZdZe jddZdZ dZ!dZ"dZ#d Z$d!Z%d"Z&d#Z'e jdgd$d%Z(d&Z)d'Z*d(Z+d)Z,d*Z-d+S), TestInterpr=r>cj||j||jfSr)rQrhr)rSrTs rW_get_xyzTestInterp._get_xys)zz$'""BJJtw$7$777rkctt5t|j|jdddddS#1swxYwYdS)NrBrS)rGrHr rhrrSs rWtest_non_int_orderzTestInterp.test_non_int_orders 9 % % 8 8 tw3 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s?AAc||\}}t||d}t|||ddt||dd}t|||dddS)NrrSrsr[r7r4rrAr rrSrTrhrrVs rW test_order_0zTestInterp.test_order_0b!!B r2 + + +"rE:::: r2 4 4 4"rE::::::rkc||\}}t||d}t|||ddt||dd}t|||dddS)Nr.rSrsr[r7rFrGrHs rW test_linearzTestInterp.test_linearrJrkr4rcgd}gd}ttd5t|||ddddS#1swxYwYdS)Nrr.r9r;r<r)rr.r9r;r<rrrz Shapes of xrrSrGrIr rSr4rwrs rWtest_incompatible_x_yz TestInterp.test_incompatible_x_ys    $ $ $ :] ; ; ; * * q!q ) ) ) ) * * * * * * * * * * * * * * * * * *s?AAcgd}gd}ttd5t|||dddn #1swxYwYgd}ttd5t|||dddn #1swxYwYgd}tj|d}ttd5t|||ddddS#1swxYwYdS) N)rr.r.r9r;r<rNzx to not have duplicatesrrS)rr9r.r;r<rzExpect x to be a 1D strictly)r.r7)rGrIr rKrQrrPs rW test_broken_xzTestInterp.test_broken_xs       :-G H H H * * q!q ) ) ) ) * * * * * * * * * * * * * * *    :-K L L L * * q!q ) ) ) ) * * * * * * * * * * * * * * *    JqMM ! !' * * :-K L L L * * q!q ) ) ) ) * * * * * * * * * * * * * * * * * *s3>AA#BB BC..C25C2c||\}}dD]/}t|||}t|||dd0dS)Nr9r;r<rrrrsr[rG)rSrTrhrr4rVs rWtest_not_a_knotzTestInterp.test_not_a_knotscb!!B# ? ?A"2r1--A AAbEE2E > > > > > ? ?rkc||\}}t||dd}t|||ddtddD]6}t||d|||d |d 7t||ddd }t|||ddtddD]6}t||d|||d |d 7dS) Nrrr9rsr[r.rrr7dy=rtr4r:r)rAr rr)rSrTrhrrVrs rW test_periodiczTestInterp.test_periodics5b!!B r2J ? ? ?"rE::::q! I IA AAbeNNNAAbfOOO% H H H H H r2JR H H H"rE::::q! I IA AAbeNNNAAbfOOO% H H H H H I IrkrUcd}tjd}tj||dz}||dz}|d|d<||||}}t |||d}t|||d dS) Nrrrerr7rrr9rsrt)rKrRrrr;rQr r)rSr4rTrUrrwrrVs rWtest_periodic_randomzTestInterp.test_periodic_randoms i##D)) GC%%a((2- . .   a 3 &u!zz!}}bjjmm1 q!q* = = =!ae,,,,,,rkc|jjd}tjd}||dztjz}tj|}d|d<dtjz|d<tjd|f}tj ||d<tj ||d<| || |}}t||ddd }t|D]-}t||||dd|fd .t||d||dd dS) Nrrr9r=r7r.rrrYrsrt)rhrrKrRrr;rrr/rrrQr rr)rSrTrUrrwrrVrs rWtest_periodic_axiszTestInterp.test_periodic_axis$sK GM! i##D))   a 1 $ru , GAJJ!BE " HaV  vayy!vayy!zz!}}bjjmm1 q!q*1 E E Eq : :A AAadGGQqqq!tW5 9 9 9 9 9!A$1R5666666rkc`tjd}d}d}tj||}||}|ddz |d<t t 5t|||dddddS#1swxYwYdS) Nrrrr7r.rrr9)rKrRrrr;rGrIr )rSrr4rUrwrs rWtest_periodic_points_exceptionz)TestInterp.test_periodic_points_exception5si##D))   GC%%a(( ) )   a uqy! : & & > > q!q* = = = = > > > > > > > > > > > > > > > > > >sB##B'*B'cxtjd}d}d}tj||}||}tj|d|zz}t t5t||||dddddS#1swxYwYdS)Nrr;rr9r) rKrRrrr;r/rGrIr )rSrr4rUrwrr2s rWtest_periodic_knots_exceptionz(TestInterp.test_periodic_knots_exception@si##D))   GC%%a(( ) )   a  HQQY   : & & 7 7 q!Q: 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7sB//B36B3rc|t|j|j|d}t|j|jd|}t |j|}t |||jdt d|D];}t |j||}t |||j| d  > > > > ? ?rkcBt|j|jdd}t|j|jd}t ||j||jdt jd}d}t j| |dz}| |d z}|d |d <t||dd}t||d}t ||||ddS) Nr;rr9r^rsrtrrerr7r) r rhrrrrKrRrrr;)rSrVcubrrUrwrs rWtest_periodic_cubiczTestInterp.test_periodic_cubicXs tw1j I I I$'47J???$' CCLLu====i##D))  GC%%a((2- . .   a 3 &u! q!q* = = =!Q 333!cc!ff5111111rkc:dt|j|jd}t|jt |j|jt jfd}t||j||jddS)Nr;rr9c6t|tS)Nrr)rrK)rwr3r4r2s rWrz6TestInterp.test_periodic_full_matrix..osK1ar$B$B$Brkrsrt)r rhrrrrK vectorizer)rSrVr.r3r4r2s @@@rWtest_periodic_full_matrixz$TestInterp.test_periodic_full_matrixhs  tw1j I I I DGQ ' ' &twA > > \BBBBBB C C$' BBtwKKe<<<<<.#?#?#?IRC#?#?#?rkr7cg|]\}}|Sr$r$rxs rWrz2TestInterp.test_quintic_derivs..rzrk)rOrPrr rrrQ) rSrTr4rUrwrrsrtrVs rWtest_quintic_derivszTestInterp.test_quintic_derivssZ1 IIarzI * * FF1IIF#'" q!q5%. A A A!ae%8888!!AaD!**aa!ajj!9:: #?#?#?#?#?rz RR T T T!!AbE1++qq2{{!;<< #?#?#?#?#?rz RR T T T T Trkunstablercd}t|j|}ddg}t|j|j|||df}t ||j|jdddS)Nr;rq)r9r@r^rsr[)rrhr rr)rSr4r2rsrVs rWtest_cubic_deriv_unstablez$TestInterp.test_cubic_deriv_unstablesm  DGQ  '" twAt} M M M$' DG%eDDDDDDrkc d}||\}}t|||dz|dz|dd|ddzdz ||dz|dz}t||||dgdgf}t |||dd t j||ddd d sJt j||ddd d sJdS) Nr9r.rr7r>rrr^rsr[r=r)rArrr rr r )rSrTr4rhrr2rVs rWtest_knots_not_data_sitesz$TestInterp.test_knots_not_data_sitess' b!!B b! r!u$ABB"SbS'!R'! r"v%   r2q!)/6('; = = = "rE::::|AAbeQKKe<<<<<<|AAbfaLL#u========rkcd}|ddg}|ddg}t|||dgdgf}|ddd|j}|dz}t |||d d dS) Nr;r=r0r.r=rqr^rrErsr[)rQr rfrPr)rSrTr4rwrrVrhrs rWtest_minimum_points_and_derivz(TestInterp.test_minimum_points_and_derivs  JJBx  JJBx  q!Q'WI0F G G G [[R2:[ 6 6 U"rE::::::rkcgdx}}tt5t||dgdfdddn #1swxYwYtt5t||ddddn #1swxYwYtt5t||dgdddn #1swxYwYtt5t||ddddn #1swxYwYd\}}tt5t||||fddddS#1swxYwYdS)N)r0r9r;r<rrrr^*)rrrO)rSrwrlrs rWtest_deriv_speczTestInterp.test_deriv_specs%$$$A : & & @ @ q!wi-> ? ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @: & & 6 6 q!W 5 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6: & & 8 8 q!gY 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8: & & 1 1 q!R 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 "1 : & & 5 5 q!aV 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5sW=AAA;;A?A?B::B>B>C88C<?C<D>>EEc\|d}|dz}dgdg}}ttd5t||||fdddn #1swxYwYdgdg}}ttd 5t||||fddddS#1swxYwYdS) Nrr9)rrr.rzBad boundary conditions at 0.rr^)irzBad boundary conditions at 6.)rOrGrIr )rSrTrwrrrs rWtest_deriv_order_too_largez%TestInterp.test_deriv_order_too_largesI IIaLL qDx&1 :-L M M M 5 5 q!aV 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5x'1 :-L M M M 5 5 q!aV 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s#AAA?B!!B%(B%cd}||\}}|d|zz}dgdg}}t|||||f}t|||ddtj||dd |dd d sJtj||d d |dd d sJd D]0}t||| }t|||dd1dS)Nr;r/)r.y@)r.y@@r^rsr[rr.rr7rr.rS)rAr rcmathr )rSrTr4rhrrsrtrVs rWrzTestInterp.test_complexs( b!!B #b&[!zK=u r2q5%. A A A"rE::::}QQr!ua[[%(1+uEEEEEE}QQr"vq\\58A;FFFFFF ? ?A"2rQ///A AAbEE2E > > > > > ? ?rkc|d|j}|d|j}dD]}t|||}|| dS)NrerErrS)rOint32r )rSrTrwrr4rVs rW test_int_xyzTestInterp.test_int_xysn IIbI ) ) IIbI ) )  A"1a1---A AaDDDD  rkc|ddd}|ddd}|ddd}dD]}t|||dS)Nr7r.rrrrS)rfr )rSrTrhrwrr4s rWtest_sliced_inputzTestInterp.test_sliced_inputsf [[Q $ $ sssG sssG * *A q!q ) ) ) ) ) * *rkc|d|j}|dz}|j|j|j fD]F}t j|d|}ttt||GdS)NrerEr9r7) rOrPrMrNxpxatsetrGrIr )rSrTrwrzs rWtest_check_finitezTestInterp.test_check_finite"s IIb I + + qD&"&26'* @ @Aq" !!!$$A *&8!Q ? ? ? ? @ @rk)r.r9r;rczttd}d|D}t|||dS)Nrecg|]}|dzS)r9r$)ras rWrz.TestInterp.test_list_input../s   aQT   rkrS)listrr rPs rWtest_list_inputzTestInterp.test_list_input+sC rOO  1   1a1%%%%%%rkc ||\}}|||fd}dddgfg}dddgfg}t||d||f}t|||d d t||d d||d d|j d d t||d d||d d|j d d dS)Nr.rr0r>r?r@r;r9rsr[rrEr7)rArr rrQrP)rSrTrhrrsrtrVs rWtest_multiple_rhszTestInterp.test_multiple_rhs2s%b!!B XXr2hQX ' 'b"Xb"X r2UEN C C C"rE:::: AbeQKK JJuQx{"*J 5 5E      AbfaLL JJuQx{"*J 5 5E      rkctjd}d\}}tj||}||dddf}t |||}|jj|dddfksJd|dfg}d|dfg}t |||||f }|jj||zdz dddfksJdS) Nrr;rrrrrr.rrrr^)rKrRrrr r3r) rSrr4rUrwrrVd_ld_rs rW test_shapeszTestInterp.test_shapesCsi##D))1 GCJJAJ&& ' ' JJQ1aLJ ) ) q!Q ' 'syQ1aL((((3::i(()*3::i(()* q!Qc ; ; ;syQUQY1a0000000rkc.||\}}||}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 dddn #1swxYwY|||||fd}dddgfg}d ddgfg}t||d||f}t||dd}t|j|jdtj d}d\} } tj | | } | | dddf} | | | | } } d|dfg} d|dfg}t| | | | |f}t| | | d}t|j|jddS)Nr;rEr9rrrZrt)rErDrr9)NrDrrFtypor.rr=rCrrrrrrrr^rD)rArr rr3rGrIrrrKrRrrrQr/)rSrTrhrr.rrsrtrr4rUrwrrrs rWtest_string_aliaseszTestInterp.test_string_aliasesRsb!!B VVBZZ B!Y ? ? ? B!vh5I J J Jbd//// B!(>@@@ B!*0F8(<>>>bd//// B!5F G G G B!dXJ5G H H Hbd//// B!\ B B B B!T : : :bd////: & & < < r2F ; ; ; ; < < < < < < < < < < < < < < <XXrvvbzz266"::.QX 7 7b"Xb"X B!eU^ D D D B!(>@@@bd////i##D))1 GCJJAJ&& ' ' JJQ1aLJ ) )zz!}}bjjmm1288I&&'(288I&&'( 1a#s < < < 1a ; ; ;bd//////sE55E9<E9ctjd}d\}}tj||}||}t ||}t ||||}||}t|j|||f\} } } t| | || } t| j |dddS)Nr)r;rrrsr[) rKrRrrrmake_interp_full_matrrQrvr rr3) rSrTrr4rUrxy_npryrrwrr2rVs rWtest_full_matrixzTestInterp.test_full_matrixsi##D))1wszzqz))**zzqz!!4## "4tQ 7 7 ZZ^^bj4t"4551a q!Q * *Re%888888rkc Vtjd}d}tdddD]}t |dz dz }tj|d|f}td|dzD]}|d| |dfxxtj|d||z fz cc<||dd| fxxtj|d||z fz cc<|||f}||d|| df<|||f}||| dd|f<tj||f} tt|| dz dD]F\}} | d krtj|| | |d| f<)tj|| | || df<G||} tt| ||| |tj || d dS) z Random elements in diagonal matrix with blocks in the left lower and right upper corners checking the implementation of Woodbury algorithm. rr; r9r.Nr7r)offsetrsrt) rKrRrrintdiagflatr/rdiagonalrrlinalgsolve) rSrrUr4rrrurlldrrVs rW test_woodburyzTestInterp.test_woodburysX i##D)) q"a ? ?A!a%1%%F CJJ1v..//A1fqj)) B B#A2#qrr' bk#**aQZ*@*@AAA !""crc' bk#**aQZ*@*@AAA VV,--B#%Agvgwxx VV,--B#%Avghh !Q  A!%! R"@"@AA 8 81q55!{1Q777Aa!eHH!{1Q777AaeHH 1 A /2r1a@@IOOAq11 ? ? ? ? ?# ? ?rkN).r!r"r#rKrfrrhrrrArDrIrLrarrrQrSrVrZr\r^r`rbrdrgrkrorur|rrrrrrrrrrrrrrrrr$rkrWr?r?s  RBE " "B B888888;;;;;; [S,,,//**0/*  [S,,,// * *0/ *??? III  [S"4"4"455 - -65 -777" > > > 7 7 7 [S,,,// ? ?0/ ?222 ===   ,;;;, T T T [j)) E E*) E>>>& ; ; ;555. 5 5 5???"***@@@ [S,,,//&&0/&    " 1 1 1303030j 9 9 9?????rkr?c|j|jksJ|j|j|zdzksJ|j}tj||ftj}t |D]W}||}|||kr|}ntj||dz }t j||||} | ||||z |dzf<Xtj ||} | 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). r.rE) rrKr/rPrrr*evaluate_all_bsplslr) rwrr2r4rUArrleftbbr3s rWrrs 6QV     6QVaZ!^ # # # # A !Qrz***A 1XX ! !t 1Q4<<DD?1d++a/D '1dD 9 9 !T!VDF]  AA Hrkcttj|||f\}}}|j}|j|z dz }tj||ftj}t |D]W}||}|||kr|} ntj||dz } tj |||| } | ||| |z | dzf<Xtj |j |} tj |j |} tj | | } | || ffS)z,Make the least-square spline, full matrices.r.rE)rvrKrQrr/rPrrr*rdotTrr)rwrr2r4rrUrrrrrr%Yr3s rWmake_lsq_full_matrixrs"*q!Qi((GAq! A  QA !Qrz***A 1XX ! !t 1Q4<<DD?1d++a/D '1dD 9 9 !T!VDF]  qsAA qsAA AA q!f9rkmethodnorm-eqqrc neZdZejdZd\ZZej eeZ eeZ e ej e de ddeZedZedZdZed Zed Zd Zed Zd ZedZedZedZedZedZejde e!dddZ"dZ#dS)TestLSQr)rr;rr7rc~|j|j|j|jf\}}}}t ||||\}}t |||||}t |j||jj|j |z dz fksJ|\} } tj | |d\} } } } t |j| dS)Nrr.r7)rcond) rwrr2r4rrrr3rrrKrlstsq) rSrrwrr2r4c0AYrVaarrrs rW test_lstsqzTestLSQ.test_lstsqsVTVTVTV3 1a%aAq11B Aq!Qv 6 6 6R   syQVaZ!^-----Bioob!2o66 Aq!R     rkcgt|j|j|j|jf|jR\}}}}||}t|||||}t||||||} t|j| jdt|j | j d|j| jksJdS)Nrwrrsrt) rvrQrwrr2r4rnrrr3) rSrrTrwrr2r4rrVb_ws rW test_weightszTestLSQ.test_weightssHc"*tvtvtv&>??GGG 1a LLOO Aq!Qv 6 6 6aAqAf===SU////SU////sce||||||rkcBgt|j|j|j|jf|jR\}}}}t jd |j d}||}t|||||d}t|||||d}t||||d} t|j |j d ||| j |j z dkrJdS) Nrrrrrrrrsrt)rvrQrwrr2r4rKrRrrrrrr3rabs) rSrTrwrr2r4rb_neb_qrb_no_ws rWtest_weights_samezTestLSQ.test_weights_samesGc"*tvtvtv&>??GGG 1a I ! !$ ' ' / /QWQZ / @ @ JJqMMq!QQyAAAq!QQt<<< Aq!D999U333366"&&DF!233e;<<<<<<.4sH$$$a111a4!Qv>>>$$$rkc*g|]}|jSr$)r3)rrrs rWrz/TestLSQ.test_multiple_rhs_2..6s777a"Q%'777rkrZrt)rvrQrwr2r4rUrKrRrrrrrrr3) rSrrTrUnrhsrrVcoefsrr4r2rwrs ` @@@@@rWtest_multiple_rhs_2zTestLSQ.test_multiple_rhs_2+s&Gc"*tvtv&677GGGG 1ai##D)) JJQIJ & & JJqMM Aq!Qv 6 6 6$$$$$$$$T{{$$$77775;;77788:qs//////rkcngt|j|j|jf|j|jR\}}}}d}t j||f}||}t||||d}t||||d} t|j | j ddS)Nr;rrrrrZrt) rvrQrwr2r4rUrKrRrrr3) rSrTrwr2r4rUrrrb_neqs rWtest_multiple_rhs_3zTestLSQ.test_multiple_rhs_3:sGc"*tvtv&677GGGG 1a I  1d)  , , JJqMMq!Q$7771a9===e444444rkcgt|j|j|jf|jR\}}}||jdz}t |||||}t ||||||}t ||||||} t||||d| |zzdddS)N?@rr/rZr[) rvrQrwr2r4rrrrr) rSrrTrwr2r4rUrVrrs rWrzTestLSQ.test_complexCs=3rzDFDF#344rZrt)rKrOfloat32rrrfloatr)rSrrwrr2spl_f32spl_f64x2s rW test_f32_xyzTestLSQ.test_f32_xyms Ib + + + Ib + + + AOOO!!QQv>>>! HHUOOQXXe__ahhuoo6   eafn # WWR[[u======rkctjddd}|ddd}|ddd}t|d}t|||d|dS)Nr7r.rr;r)rKrfrr)rSrrhrwrr2s rWrzTestLSQ.test_sliced_inputzsa[Q $ $ sssG sssG AqMM1a1V444444rkc tjdt}|dz}t |d}tjtjtj fD]&}||d<ttt||||'dS)N r9r;r7r) rKrOrrrrMrNrGrIr)rSrrwrr2rs rWtest_checkfinitezTestLSQ.test_checkfinites IbMM  ' ' qD AqMM&"&26'* O OAAbE *oq!Qv N N N N N O Orkc|j|j|j}}}|d|d|dt ||||dS)NFrj)rwrr2r)rwrr2rlrrs rWrmzTestLSQ.test_read_onlysp&$&$&a1      !qAf555555rkr4r.cJ|j|j}}ttj|d|dd|}t ||||d}t ||||d}|dd|ddzdz }t ||||d dS) Nrr7rrrrr.r>rZrt)rwrrrKrfrr)rSr4rwrr2 spl_norm_eqspl_qrrhs rWtest_qr_vs_norm_eqzTestLSQ.test_qr_vs_norm_eqsvtv1 BK!aeQ// 3 3%aA9EEE  AqAd;;;eafn # B%@@@@@@rkc^tj|jd}tj|jd}t |j|j|jdd}t |||jdd}|dd|ddzdz }t ||||d dS) Nr9r;rrr.r7r>rZrt)rKrepeatrwrrr2r)rSrwrrspl_2rhs rWtest_duplicateszTestLSQ.test_duplicatess Idfa  Idfa !DIII1df$???eafn #b 55995999999rkN)$r!r"r#rKrRrrrUr4rrwrrrfr2parametrize_lsq_methodsrrrrrrrrrrrrrmrarrrrrrr$rkrWrrs' )   % %C DAq 1 A 1 A  AaD!B%++Q//A ! ! !    = = =::: 0 0 0555 M M MMMM,555 > > >555OOO666 [S$$uuQ{{"3"344AA54A:::::rkrc4eZdZdZdZedZdZdS) 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||_||_||_|jdksJ|jdksJ|jd|jdksJdS)Nr9r.r)rrncrr)rSrrr s rW__init__zPackedMatrix.__init__s\ v{{{{{awqzV\!_,,,,,,rkc4|jjd|jfSNr)rrr rCs rWrzPackedMatrix.shapesv|A''rkc>tj|j}|jjd}t |jdD]V}t |j|j|z |}|j|d|f|||j||j||zf<W|S)Nr.r)rKr/rrrminr r)rSoutnelemrnels rWtodensezPackedMatrix.todenseshtz"" Qsy|$$ J JAdg A.66C:>&DSD/C4;q>$+a.3"666 7 7 rkN)r!r"r#__doc__r propertyrrr$rkrWrrsW---((X(rkrr.c |ddlm}|j}|j}|j}|j\}}|jd|ksJ|} |} t||D]} || } t| |D]} | t| |krn|| | df| | df\}}}|| | df<td|D]4}t||| | |f| | |f\| | |f<| | |dz f<5d| | df<t| jdD]1}t||| | |f| | |f\| | |f<| | |f<2،tt| jd}t| tj |tj|}|| fS)zjThis is a python counterpart of the `_qr_reduce` routine, declared in interpolate/src/__fitpack.h r)dlartgr.r=r7rE)scipy.linalg.lapackrrrr rrrrfprotarrrKrMint64)a_prstartrowrrrr rnzRrroirr3rrroffsR_ps rW _qr_reduce_pyr!s+***** A ZF B GEAr 71:???? A B 8Q  FF AYr2 F FAC2JJfQq!tWa1g..GAq!AadG1b\\ D D%+Aq!AqD'1QT7%C%C"!Q$1ac6AaeH28A;'' F F%+Aq"QT(Bq!tH%E%E"1a4"QT(( F agaj!! " "D q"(4rx888" = =C 7Nrkc8||z||zz}| |z||zz}||fS)zLGivens rotate [a, b]. [aa] = [ c s] @ [a] [bb] [-s c] [b] r$)r3rrrVrrs rWrrs1 A#!)B A!B r6Mrkc|j}|j\}}|j}|jd|jdksJtj|d|}||dz ||dz dfz ||dz df<t |dz ddD]d}t |||z }||d|df||dz||zdfzd} ||| z ||dfz ||df<e|S)zBacksubsitution solve upper triangular banded `R @ c = y.` `R` is in the "packed" format: `R[i, :]` is `a[i, i:i+k+1]` rNr..r9r7r)rrr rKrrrr) r rrrrr r3rrsumms rWfpbackr%s A GEAr B 71: # # # # afART7Qr!tQwZ'AbdCiL 2a4R .."bdmm!QsUD.!Aac!C%in$55:::BBdTka1g-!S& Hrkc2eZdZdZdZdZdZdZdZdS) TestGivensQRcd}tj|t}|dzdd|zz z}t||}||||fS)Nr;rEr.)rKrOrr)rSrUr4rwrr2s rW_get_xytzTestGivensQR._get_xytsM  Iau % % % qD1ac7N 1  !Qzrkc d}||\}}}}tj|||}tj|\}}|j|z} |j\} } | |jd|z dz ksJ|jdd|dz} tj | tj } |j | |dz} t| | | }|dddf}tj| | | |t#tj||z||z tj|dt#tjt)| |dddfz t)| |dddfztj| dtj|| }tj|j|j||||tj||\}}}t#||dddfddS) Nrerr.rErZrtgvIh%L=-a=)r)r r>rrrrrindicesrKascontiguousarrayrdatarrr* qr_reducerminimumrrrr%rr rn)rSrUrwrr2r4a_csrqrqTyrr rrry_c_fullc_bandedrs rW test_vs_fullzTestGivensQR.test_vs_full#s; ]]1%% 1a%aA..uU]]__%%1cAg 2QWQZ!^a'''''ww!A#w'%fBH=== J  q!A# & & FB ' ' qqq$wZ1fb"---  199;;?#$99;;?4457]15E5EE S S S S 3sR1X~#6#6#&sR1X~#6#6889;s9K9KRW Y Y Y Y!S!!!adAr1aaRTUU!QAU;;;;;;rkcd}||\}}}}tj|||}|j\}}||jd|z dz ksJ|jdd|dz} t j| t j} |j ||dz} t| | |} |dddf} t| | \} }tj | | || t| j| jdt!| j| jd| j| jksJt|| ddS) Nrerr.rErZrtF) check_dtype)r)r r>rr,rKr-rr.rrr!r*r/rrrrr )rSrUrwrr2r4r1rr rrrr4RRrs rWtest_py_vs_compiledz TestGivensQR.test_py_vs_compiledFsS ]]1%% 1a%aA.. 2QWQZ!^a'''''ww!A#w'%fBH=== J  q!A# & & FB ' ' qqq$wZq"%%B1fb2...ac.... 18????u}}}}BU++++++rkcNd}||\}}}}tjd|dzt}t j||||\}}} |jd} tj|||} | |dddfz } | j | |dzf} | j dd|dz tj}t|| dt!||| |jd|z dz ksJdS)Nrer.rErrZrt)r)rKrOrr* data_matrixrr r>tocsrr.rr,rrrr)rSrUrwrr2r4rrrr rr1a_wA_offset_s rWtest_data_matrixzTestGivensQR.test_data_matrixas% ]]1%% 1a Ia1E * * * ,Q1a88 62 GAJ%aA..qDz!((** X  q!A#h ' '+gg1g&--bh772E****(((QWQZ!^a'''''''rkc d}||\}}}}tj||dzf}tj|||tj|\}}}t |||} tj||||t| |} tj||||||tj||\} } } t| | ddS)Nrer9rsrt) r)rKrr*r=rnrr/r%r) rSrUrwrr2r4rrr rr3rrs rW test_fpbackzTestGivensQR.test_fpbackrs ]]1%% 1a E!QT'N ,Q1bl1ooFF 62 FB ' '1fb!,,, 1aLL?1b!Q1bl1ooqIIAqAE******rkc *d}||\}}}}tj|dzft}|D]T}t t j|||||dz|t t j||||d|z|UdS)Nrer.rEr9)r)rKr/rrr*r)rSrUrwrr2r4 zero_arrayrs rWtest_evaluate_all_bsplz#TestGivensQR.test_evaluate_all_bspls ]]1%% 1aXq1uhe444  L LD *1aq!a%@@* N N N *1aq!A#>>  L L L L L LrkN) r!r"r#r)r7r;rBrDrGr$rkrWr'r'st!<!<! ? ? ? ( ( !!Q ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (slCC C"DD D &%EEE6FF #F G77G;>G;I%%I),I)ctjtd5}|d}|d}|d}dddn #1swxYwYt|||}t ||ddddS) 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.npzrwry_GCVSPLN-C6?F)r\r]r9)rKloadrOrr)rSr.rwrrXy_comprs rWtest_compare_with_GCVSPLz,TestSmoothingSpline.test_compare_with_GCVSPLsRWY|,, - - (S AS AJ'H  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( .'1--a00 4USSSSSSsAA A cJtjd}d}tj||dzdz }|dztjd|zz|dzz|dd|z}||||}}t||d}t||dd }| |d |d d|z}t||||d dS)z In case the regularization parameter is 0, the resulting spline is an interpolation spline with natural boundary conditions. rrr<r9r;r=r )lamrEr^rr7rZrtN) rKrRrrr;rrSrQrr rfr) rSrTrrUrwr spline_GCV spline_interpgrids rWtest_non_regularized_casez-TestSmoothingSpline.test_non_regularized_cases" i##D))  GC%%a((1,q0 1 1 qD26!a%== 1a4 '#**Ra*@*@ @zz!}}bjjmm1*1aR888 *1aIFFF {{1Q42A.. 4((% d++" $ $ $ $ $ $rkr9cFtjd}d}tj||dzdz }|dztjd|zz|dzz|dd|z}t|j||f\}}t||}| tdd D]}| |}tj|t|d t|||} t#|||||z } t#| ||||z } | | krt%d | d d | d dS)Nrrr<r9r;r=r rerg>@zJSpline with weights should be closer to the points than the original one: z.4z < )rKrRrrr;rrSrvrQrchoicerrrrrrrrI) rSrTrrUrwrr2indrspl_worigweighteds rWtest_weighted_smoothing_splinez2TestSmoothingSpline.test_weighted_smoothing_spline si##D))  GC%%a((1,q0 1 1 qD26!a%== 1a4 '#**Ra*@*@ @2:1v&&1#Aq))::eCjjr:22 3 3C A F1c#hh   # #C ( ( ()!Q22Ess1S6{{QsV+,,D553==1S6122Hh "2CG!O"2"2$,!1"2"2333 3 3rkN) r!r"r#rVr\rbrar fail_slowrir$rkrWrQrQsq (((>6T6T6Tp$$$* [133333rkrQc ( |\ |\tz dz }|dzksJtz dz dzksJt fdt|D}tj|S)z-A naive 2D tensort product spline evaluation.r.c 3K|]F}tD]4}||ft|zt |zV5GdSr)rr%) rixiyr3r4nytxtyrwrs rWrzbspline2..' s{77E"II77&(Bi!Aq"b//)AaBOO;7777777rk)r=rrrKrQ) rr2r3r4nxrrorprqrwrs `` @@@@@rWbspline2rs s DAq FB R1qB !A#IIII R1qB !A#IIII 7777777777b 777 7 7C :c??rkc|dkr#|||cxkr||dzkrnndndS|||z||krd}n6|||z |||z||z z t||dz ||z}|||zdz||dzkrd}nE|||zdz|z |||zdz||dzz z t||dz |dz|z}||zS)Nrr.r0r=r%rs rWr%r%, sAvvda((((!AaC&(((((ssc11v1~~ !A$h1Q3!A$ '!AqsAq// 91Qx1QqS6 !AhlQqs1uX!A#. /!AqsAaC2C2C C 7Nrkctz dz }|dzkrt|ksJtfdt|DS)Nr.c3NK|]}|t|zV dSrrurs rWrzbspline..= s866qta1amm#666666rk)r=rr)rwr2r3r4rUs```` rWrPrP: sj A QA 1HH3q66Q;;;; 6666666U1XX666 6 66rkceZdZddZdZdS) NdBSpline0r;c,t|}|t|jksJ t|n#t$r |f|z}YnwxYwtd|D|_td|D|_||_dS)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. c3>K|]}tj|VdSr)operatorindex)rkis rWrz&NdBSpline0.__init__..Y s,66bx~b))666666rkc3LK|]}tj|tV dS)rEN)rKrQr)rtis rWrz&NdBSpline0.__init__..Z s1??rrz"E222??????rkN)r=rrHr*r4r2r3)rSr2r3r4rs rWr zNdBSpline0.__init__A s1vvs17||####  FFFF   T AAA 66A66666??Q?????s;A Ac tj}t|ksJdg|z t|D]}j||}}j|}|||kr| |<nt j||dz |<| ||cxkr| |dzksnJ ||kr |t||z ksJt  d} fdt|D}tj|D]D j t j fdt|Dz} || z }Et j |S)Nnoner.rcjg|]/}t|j|z |dz0Sr.)rr4)rrrrSs rWrz'NdBSpline0.__call__..u s:HHHqqtdfQi'122HHHrkc xg|]6}t|j||j|7Sr$)r%r4r2)rridxrSrws rWrz'NdBSpline0.__call__..w sQ*@*@*@./+,AaD$&)SVTVAY*O*O*@*@*@rk) r=r2rr4rKrr* itertoolsproductr3prodrQ) rSrwrrtdxdr4riterstermrrs `` @@rW__call__zNdBSpline0.__call__] s46{{1vv~~~~JtOt 4 4AVAY!Bq ARU{{!r2..2!ad8r////R!QZ//////Q41991B! !3!3!3!3!3 !HH HHHHHE$KKHHH$e,  C6#;*@*@*@*@*@*@38;;*@*@*@"A"AAD dNFFz&!!!rkNr;)r!r"r#r rr$rkrWryry@ s78"""""rkryc|eZdZdZefdZefdZdZdZdZ dZ dZ d Z d Z d Zefd Zd ZdZdZdZejdddgejdddgdZejdddgdZdZdZdZdZdZdZdS) TestNdBSplinectjd}d\}}d}tj|||zdz}|||f}||}||}t |||}t|f||} |d} || } t| | dddf|| d| | dddfj | j d |j dfksJdS) NrrCrr.rrrsrtr) rKrRrrrrQr rrr) rSrTrrUr4n_trr2r3rVnbxis rWtest_1DzTestNdBSpline.test_1D s@i##E**1 GCKKQUQYK// 0 0 KKaYK ' ' JJqMM JJqMM Aq!   tQ " " [[b[ ! ! ZZ^^ 2aaag;"E + + + +r"QQQW+$!agaj(AAAAAAArkc|d}|dz}t||d}|dzd|zz}t||d}|j|jf}|jdddf|jdddfz}||dfS)Nrr;rSr9rOr r2r3 rSrTrwrr2y_1rt2rs rW make_2d_casezTestNdBSpline.make_2d_case s IIaLL qD A+++dQqSj"1cQ///eUW  U111d7^egdAAAg. .2qyrkcT|d}|dz}t||d}|d|jdz}|dzd|zz}t||d}|j|jf}|jdddf|jdddfz}|||j|jfS)Nrr;rSrrEr r9)rOr rPr2r3r4rs rW make_2d_mixedzTestNdBSpline.make_2d_mixed s IIaLL qD A+++ IIarzI * *S 0dQqSj"1cQ///eUW  U111d7^egdAAAg. .2sueg%%rkc gd}||\ d|D} fd|D}tj|tj}t |||||jddt d }||jt|fksJt |||||jd tj|}tj|d <||}t| |||gd tj d}| ddz}||}||}|jdksJ||dj}|d|d} } t t!||| d z| d zd| zzzd dS)Nr rB)rBr.)rr rrc6g|]\}}|dz|dzd|zzzSr(r$rrwrs rWrz3TestNdBSpline.test_2D_separable.. 1777&1a!Q$!Q$1*%777rkc lg|]0}t|dDtj1S)c6g|]}tj|Sr$)rKrQrrs rWrz>TestNdBSpline.test_2D_separable... s +++1A+++rk)rsrKrQ)rrrr4rs rWrz3TestNdBSpline.test_2D_separable.. sW    +++++ 2    rkrEFrsr)r\r;rSrtr)TFFr)r<r;r9rr)r<r;)r7r9)r.)r..r9)rrKrQrPrrrr=rMrrrRrrrrr)rSrTrtargetrbspl2rrrrrrwrrr4rs @@@rWtest_2D_separablezTestNdBSpline.test_2D_separable so / / /%%%,, B77B777          Z"* - - - 2 6 <<$)" $ $ $ $ "bA&&&uRyy3r77+----b  6 <<5 J J J JZ^^64 ZZ^^r++RZZ8L8L8L-M-MNNNi##E** [[i[ ( (1 , ZZ^^r|v%%%%jjW%%'6{CK1B///11qs +% 9 9 9 9 9 9rkcjd}gd}d|D}||\}}}|||||fd}d} t||d} | | } t|||| } | jd ksJt | || gd zd | |jt j|dd | jj|dzksJt | ||||jdddfdd| |d} t|| d}|| } | jdksJ| d|j| z}t | |d ||jt j|dd |jj|dzksJt |||||jddddfdddS)Nr9rc6g|]\}}|dz|dzd|zzzSr(r$rs rWrz5TestNdBSpline.test_2D_separable_2.. rrkrrrrr;rSr<r<rsrtr7rEFgvIh%,=rrrr9r9)r9r9) rrrrrrKr3rQrPrr)rSrTrrrrrr4c2_4rbspl2_4r val_singlec2_22bspl2_22 target2_22s rWtest_2D_separable_2z!TestNdBSpline.test_2D_separable_2 s / / /77B777%%%,, BxxRR(qx11 B***)Yr2q))"-- |t####*!122 @ @ @ @ wr{{ BHRLL"$5 8N$NNNNN  6 <. rrkcg|] }|d|zz S)y@r$)rr2s rWrz=TestNdBSpline.test_2D_separable_2_complex.. s +++q!bd(+++rkrrrr9rrr;rSrr<rsrt)rrrrr) rSrTrrrrr4rrrrrs rWtest_2D_separable_2_complexz)TestNdBSpline.test_2D_separable_2_complex s / / /77B777++F+++%%%,, B 6]xxRR(qx11 B***)Yr2q))"-- |t####*a00u > > > > > >rkc tjd}dtjddddtj|ddzddddf tjddddtj|ddzddddf |jz dz jz dz ftf }d }t||t|fd tj gd gdf}t||fd|Dd dS)Nrr;rrrrr<r.rS)r0r0rsrtr.r r9皙?g?@c8g|]}t|fSr$)rs)rrr3r4rprqs rWrz0TestNdBSpline.test_2D_random..- s+CCC""r2h155CCCrk) rKrRrrrrrrrrsr)rSrr2rr3r4rprqs @@@@rWtest_2D_randomzTestNdBSpline.test_2D_random si##E**  U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L KKbgaik2719Q;7K 8 8R!q))) B b"Xq!445 B B B BU;;;"??#$BCCCCCCCCCC" $ $ $ $ $ $rkc|\}}}}gd}d|D}t||||f}||jt|fksJt |||ddS)N)ffffff?r)rBg333333@)rrc6g|]\}}|dz|dzd|zzzSr(r$rs rWrz/TestNdBSpline.test_2D_mixed..3 rrkrSrsrt)rrrr=r)rSrrkxkyrrrs rW test_2D_mixedzTestNdBSpline.test_2D_mixed0 s++--BB 1 1 177B777"bRH---uRyy3r77+----b U , , , , , ,rkc||\}}}}gd}t||||f}dddddd d |zd fd d |zd fdd d|zfdd d|zfdd|zd |zfdi}|D]|\} |fd|D|j} ||| } t | | d|| |} t | | d}dD]|} tt5||| dddn #1swxYwYtt5|| dddn #1swxYwY}dS)NrrrrSrc*d|dzz|dzd|zzzSNr;r9r$rwrs rWrz2TestNdBSpline.test_2D_derivative..@ sQTQTAaCZ!8rkr.r.c$d|dzzd|zdzzSrr$rs rWrz2TestNdBSpline.test_2D_derivative..A sQTQqS1W!5rk)rrc$|dz|dzd|zzzSrr$rs rWrz2TestNdBSpline.test_2D_derivative..B sAA!!4rkr9r.cdSr r$rs rWrz2TestNdBSpline.test_2D_derivative..C ArkrcdSr r$rs rWrz2TestNdBSpline.test_2D_derivative..D rrkr;cdSr r$rs rWrz2TestNdBSpline.test_2D_derivative..E rrkcdSr r$rs rWrz2TestNdBSpline.test_2D_derivative..F rrkcdSr r$rs rWrz2TestNdBSpline.test_2D_derivative..G sqrkc.g|]\}}||Sr$r$)rrwr expected_fns rWrz4TestNdBSpline.test_2D_derivative..L s)222tq!Q""222rkrErrsrt))r7rr7rr.) rritemsrQrPrrrGrI)rSrTrrrrrr test_casesr expected_valsdirect via_methodbad_nurs @rWtest_2D_derivativez TestNdBSpline.test_2D_derivative9 sz++r+22BB 1 1 1"bRH--- 88 55 44 rT1I~~ rT1I~~ "I~~ "I~~ rT1R4L..   *//11 C COB JJ2222r222"*'M U2"%%%F FM > > > >.))"--b11J J E B B B B B# ) )Fz** % %bV$$$$ % % % % % % % % % % % % % % %z** ) )  ((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )  ) )s$D..D2 5D2 E//E3 6E3 c  tjd}d\}}tjddddtj|ddzddddf }tjddddtj|ddzddddf }||j|z d z |j|z d z f}tjgd gd f}t||f|||f }t||f|||f  t|| fd |DddS)Nr)r9r;rrrr;rr<r.rrrSc&g|] }|Sr$r$rrTbspl2_0s rWrz6TestNdBSpline.test_2D_mixed_random..m !222222rkrsrt) rKrRrrrrrrrryr rSrrrrprqr3rrrs @rWtest_2D_mixed_randomz"TestNdBSpline.test_2D_mixed_random_ sii##E**B U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L KKbglQ." q0@AK B B U;;;"??#$2r(A"b222b"XqRH555b 2222r222 @ @ @ @ @ @rkctjd}tjddz}t||dzd}t||dzd|zzd}|jdddf|jdddfz}t |j|jf||j|jf}|dzdddf|dzd|zzdddfz}t||f|}dtj ||D} || } tj |  rJt| || d t| |j|d dS) Nrrr r;rSr9cg|] \}}||f Sr$r$rrrVs rWrz0TestNdBSpline.test_tx_neq_ty..| 9 9 9Aq!f 9 9 9rkrsrt)rKrOr r3rr2r4rrrranyrrr) rSrwrspl_xspl_yrrvaluesrgirbxis rWtest_tx_neq_tyzTestNdBSpline.test_tx_neq_tyo s} IaLL IaLL3 "1ada000"1adQqSjA666 WQQQW aaa 0 0%'57+R%'571CDDQ$4AqD1Q3Jaaa#88%q!ff55 9 9!21a!8!8 9 9 9d2hh8C==$$&&&&&SSWW51111 FL116FFFFFFrkcv|d}|dz}t||d}|dzd|zz}t||d}|dzd|zzdz}t||d}|j|j|jf} |jddddf|jddddfz|jddddfz} | | dfS)Nrr;rSr9r.r) rSrTrwrr2rry_2rrrs rW make_3d_casezTestNdBSpline.make_3d_case s IIaLL qD A+++dQqSj"1cQ///dQqSj1n"1cQ///eUWeg &eAAAtTM"gdAAAtm$%gdD!!!m$%2qyrkctjd}|ddz\}}}|dz|dzd|zzz|dzd|zzdzz}|\}}}t ||d} d t |||D} | | } | jd ksJt| |d dS) Nrr;rDrrr;r9r.rScg|]}|Sr$r$rs rWrz3TestNdBSpline.test_3D_separable..  & & &Aa & & &rkrDrsrt) rKrRrrrrziprr) rSrrwrrrt3c3r4bspl3rrs rWtest_3D_separablezTestNdBSpline.test_3D_separable si##E**++7+++a/1aAA!$1qs Q7%%'' B"bA&&& & &Q1 & & &r|u$$$$U333333rkcp||\}}}t||d}tjd}|ddz\}}} dt ||| D} || } d d d d d d} | D]z\} fd| D}|||j }t|| | |dt| | | |d{dS)Nrrr;rSrrrrcg|]}|Sr$r$rs rWrz4TestNdBSpline.test_3D_derivative.. s)))q)))rkcHd|dzz|dzd|zzz|dzd|zzdzzS)Nr;r9r.r$rwrrs rWrz2TestNdBSpline.test_3D_derivative.. s4q1a4x1a4!A#:'>!Q$1*q.'QrkcBd|z|dzd|zzz|dzd|zzdzzSNrr;r9r.r$rs rWrz2TestNdBSpline.test_3D_derivative.. s0q1u1qs ';q!taczA~'NrkcBd|zd|dzzdzz|dzd|zzdzzSrr$rs rWrz2TestNdBSpline.test_3D_derivative.. s0q1u!Q$ ';q!taczA~'Nrkc*d|zd|dzzdzzdzS)Nrr;r9r$rs rWrz2TestNdBSpline.test_3D_derivative.. s q1u!Q$ ';a'?rkcdSrr$rs rWrz2TestNdBSpline.test_3D_derivative.. ssrk))r.rr)r9rr)r9r.r)r9r.r;)r9r.r<c2g|]\}}}|||Sr$r$)rxi_yi_zi_rs rWrz4TestNdBSpline.test_3D_derivative.. s-SSSMCc[[c377SSSrkrErrsrt) rrrKrRrrrrQrrPrr)rSrTrrr4rrrwrrxi_nprrrrrs @rWtest_3D_derivativez TestNdBSpline.test_3D_derivative su%%%,, B"bA&&&i##E**++7+++a/1a))C1aLL))) ZZ  RQNNNN??**    *//11 Q QOB SSSSUSSSMJJ}BJJGGM EE",,,m% H H H H 0E,,R0044m% P P P P P  Q Qrkc z tjd}d}tjddddtj|ddzddddf }tjddddtj|ddzddddf }tjddddtj|ddzddddf }||j|z dz |j|z dz |j|z dz f}t|||f|| }t|||f||  d }t|| |d tj gd gdgdf}t|| fd|Dd dS)Nrr;rrrrr<r.rS)r0r0r.rsrtrrg?rgffffff?c&g|] }|Sr$r$)rrTspl_0s rWrz0TestNdBSpline.test_3D_random.. s!!9!9!9%%))!9!9!9rk) rKrRrrrrrrryrr) rSrr4rprqtzr3r2rrs @rWtest_3D_randomzTestNdBSpline.test_3D_random si##E**  U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L KKbgaik2719Q; ! DK E ER a1---BB<a000 Br7777 U;;;"??"??#$ B!9!9!9!9b!9!9!9FFFFFFrkc tjd}d}tjddddtj|ddzddddf }tjddddtj|ddzddddf }tjddddtj|ddzddddf }||j|z dz |j|z dz |j|z dz f||j|z dz |j|z dz |j|z dz fd zz}t|||f|| }t|||f|j| }t|||f|j | } tj gd gd gd f} t|| || d | | zzddS)Nrr;rrrrr<r.r/rSrrrrsrt) rKrRrrrrrrrrrr) rSrr4rprqrr3r2rrrs rWtest_3D_random_complexz$TestNdBSpline.test_3D_random_complex si##E**  U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L [[rwqy{BGAIaK1E[ F F [[rwqy{BGAIaK1E[ F Fr IJR a1---BB<1555BB<1555 U;;;"??"??#$ Br Rr ]2 @ @ @ @ @ @rk cls_extrapNT call_extrapcj|\}}}t||d|}gdgdgd} }}ttj||| f\}}} dt ||| D} |dz|dzd|zzz| dzd| zzdzz} || | } t | | d dS) Nr;r4rr7rr1rrr7g@cg|]}|Sr$r$rs rWrz?TestNdBSpline.test_extrapolate_3D_separable.. rrkr9r.rrsrtrrrvrKrQrr) rSrrrrr4rrwrrrrrs rWtest_extrapolate_3D_separablez+TestNdBSpline.test_extrapolate_3D_separable s%%'' B"bA:>>>++a1bj1a),,1a & &Q1 & & &AA!$1qs Q7r{333U333333rkr)FT)TNct|\}}}|\}}t||d|}gdgdgd} } }ttj|| | f\}} } dt || | D} |dz| dzd| zzz| dzd| zzdzz} || | } t | | d dS) Nr;rrrrcg|]}|Sr$r$rs rWrzATestNdBSpline.test_extrapolate_3D_separable_2.. rrkr9r.rrsrtr!)rSrrrr4rrrrwrrrrrs rWtest_extrapolate_3D_separable_2z-TestNdBSpline.test_extrapolate_3D_separable_2 s%%'' B"( K"bA:>>>++a1bj1a),,1a & &Q1 & & &AA!$1qs Q7r{333U333333rkc|\}}}t||d}gdgdgd}}}ttj|||f\}}}dt |||D}|dz|dzd|zzz|dzd|zzdzz} ||d } tj| d sJtj| d sJt| dd | dd d dS)Nr;rS)rr.r)r1rr)r7r rcg|]}|Sr$r$rs rWrzETestNdBSpline.test_extrapolate_false_3D_separable.. rrkr9r.Frrr7rsrt)rrrvrKrQrrr) rSrrr4rrwrrrrrs rW#test_extrapolate_false_3D_separablez1TestNdBSpline.test_extrapolate_false_3D_separable s$%%'' B"bA&&&**nnnnnna1bj1a),,1a & &Q1 & & &AA!$1qs Q7ru---xq """""xr #####qt fQrTl??????rkc |\}}}t||d}tjddtjdddtjg}tjdddtjdd d g}tjd dddtjd d g}d t |||D}|dz|dzd|zzz|dzd|zzdzz} tj|tj|ztj|z} tj| | <||} tj| | sJt| | d dS)Nr;rSrr.r9rr1rrr7rcg|]}|Sr$r$rs rWrz/TestNdBSpline.test_x_nan_3D.. rrkrsrt) rrrKrQrMrrrr) rSrrr4rrwrrrrrrs rW test_x_nan_3DzTestNdBSpline.test_x_nan_3D s]%%'' B"bA&&& JArvq!Q7 8 8 JCBFAsC8 9 9 JCArvsC8 9 9 & &Q1 & & &AA!$1qs Q7x{{RXa[[(28A;;6vt rxt %%))+++++U333333rkc tjd}d\}}tj|ddd}tj|df|z||df|zf}tj|ddd}tj|df|z||df|zf}|dddjjrJ|dddjjrJ||jdz|z d z |jdz|z d z f }|j }|jjrJtj gd gd f}t|ddd|dddf|||f }t|ddd|dddf|||f  t|| fd|DddS)Nrr;r;rr<rr7r9r.rrrrSc&g|] }|Sr$r$rs rWrz7TestNdBSpline.test_non_c_contiguous..; rrkrsrt)rKrRrrrrflags c_contiguousrrrrryrrs @rWtest_non_c_contiguousz#TestNdBSpline.test_non_c_contiguous$ si##E**B WS[[QQR[88 9 9 UBqE8B;RVIbL0 1 WS[[QQR[88 9 9 UBqE8B;RVIbL0 1ccc7=----ccc7=---- KKbgqj2o127A:?Q3FGK H H C7'''' U;;;"??#$2ccc7BsssG,aB8<<<b1gr##A#w/r2h???b 2222r222 @ @ @ @ @ @rkc|\}}}t||d}tdD]}d||j_d|j_t||d}|d|dksJdS)Nr;rSFr.)rrrr0 writeable)rSrrr4rrbspl3_s rW test_readonlyzTestNdBSpline.test_readonly= s%%'' B"bA&&&q * *A$)BqEK ! !"2rQ'''uY66)#4#4444444rkc|\}}}tjgdgdg}t||||||}tj|||||g}|jd|jdksJt ||dtt5tjgd||gdzdddn #1swxYwYttd5tjd d gg||gdzddddS#1swxYwYdS) Nr.)r<rrrgؗҜrrrrGrI)rSrrr4rdmdm1s rWtest_design_matrixz TestNdBSpline.test_design_matrixI s%%'' B ZIII. / / r2q ! ! / /B : :%b"q!Qi88x{bhqk)))) ckkmm%@@@@ : & & : :  #IIIrA3q5 9 9 9 : : : : : : : : : : : : : : ::-> ? ? ? 9 9  #aVHb1#a% 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s$DD D$EEEc tjd}d}tjddddtj|ddzddddf }tjddddtj|ddzddddf }tjddddtj|ddzddddf }||j|z dz |j|z dz |j|z dz f}t|||f|| }d }td ||dS) Nrr;rrrrr<r.rScRtjgdgdgdf}||dS)Nrrr)rKr)rr2rs rWrpz1TestNdBSpline.test_concurrency..worker_fnb s:{{{&&'(B CGGGGGrkre) rKrRrrrrrrr&) rSrr4rprqrr3r2rps rWrqzTestNdBSpline.test_concurrencyX sKi##E**  U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L U1aArws{{{':':;;a?Aq!K L KKbgaik2719Q; ! DK E ER a1---    Is33333rk)r!r"r#rrKrrrrrrrrrrrrrrrrarrr"r%r(r+r2r6r:rqr$rkrWrr} sBBB,!     " & & & &*9*9*9X)$)$)$V>>>&$$$&,,,$)$)$)L@@@ GGG(!& 4 4 4QQQ0GGG&@@@& [\D$<88 []T4L99 4 4:998 4 [X |'DEE44FE4 @@@ 444$@@@2 5 5 5 9 9 944444rkrc6eZdZdZdZejdgddZdZ dZ dZ d Z ejd e je jgd Zd Zejd gddZdZdS) TestMakeNDctjd}tjddz}|dddfdz|dzd|zzdddfz}dtj||D}t ||f|d}t |||d t||dzd}t||dzd|zzd}|jdddf|jdddfz}t ||jd d d d l m } | ||f|d} t | |||d dS)Nrrr;r9cg|] \}}||f Sr$r$rs rWrz7TestMakeND.test_2D_separable_simple..p rrkr.rSrZrtrXrr[)rlinearrrs) rKrOrrr'rravelr r3scipy.interpolater) rSrwrrrrrrrRGIrs rWtest_2D_separable_simplez#TestMakeND.test_2D_separable_simplel s IaLL IaLL3 111d7Q!Q$1*dAAAg!66 9 9!21a!8!8 9 9 9Aq66Q///R&,,..u====#1ada000"1adQqSjA666 WQQQW aaa 0 0DFQ7777 EDDDDDc1a&&222Bb666666rkc8tjd}tjd}dtj||D}|dddfdz|dzd|zzdddfz}tj||||f}t ||f|dt j}||}tj||||ft}|j dksJt| ddd|d | d } t ||f| dt j}||}|j d ksJt| dddd| d d dS) Nrcg|] \}}||f Sr$r$rs rWrz>TestMakeND.test_2D_separable_trailing_dims.. rrkr;r9r4solver)$r<r<rsrtr)rJr9r9) rKrOrrrr'sslspsolverrrrr) rSrwrrrvalues4rrrvalues22s rWtest_2D_separable_trailing_dimsz*TestMakeND.test_2D_separable_trailing_dims s IaLL IaLL 9 9!21a!8!8 9 9 9111d7Q!Q$1*dAAAg!66)VVVV<==Aq67a DDDbFFFF;<<CCEJJ|w&&&&q!Q//U , , , ,??<00Aq68qEEEb|z))))q!Q22|445 B B B B B Brkr4)r-r)r;r.)r.r;rc|tjd}tjddz}dtj||D}|dzdddf|dzd|zzdddfz}t ||f||t j}t|||d dS) Nrrr cg|] \}}||f Sr$r$rs rWrz,TestMakeND.test_2D_mixed.. rrkr;r9rHrZrt) rKrOrrr'rKrLrrB)rSr4rwrrrrs rWrzTestMakeND.test_2D_mixed s IaLL IaLL3  9 9!21a!8!8 9 9 9Q$4AqD1Q3Jaaa#88Aq66Qs{CCCR&,,..u======rkcFtjd}tjddz}tj|d<|dddfdz|dzd|zzdddfz}tt5t ||f|dddddS#1swxYwYdS)Nrrr7r;r9r.rS)rKrOrMrGrIr')rSrwrrs rW test_2D_nanszTestMakeND.test_2D_nans s IaLL IaLL3 "111d7Q!Q$1*dAAAg!66 : & & - - A! , , , , - - - - - - - - - - - - - - - - - -s4BBBc tjgd}tjgd}tjgdgdgdgdgdgdg}|||fS)N)rr>r?r@rr)r.r9r.r9r.r.)r.r9r;r9r.r.)r.r9r9r9r.r.)rKrM)rSrwrrs rW_get_sample_2d_datazTestMakeND._get_sample_2d_data s H... / / H... / / H""""""""""""""""""   !Qwrkc |\}}}t||f|d}t||f|d}tjgdgdgj}t ||||ddS) Nr.rSrArr.gffffff@g333333@rffffff @333333?r;r.rXrYr@r[r0r;rsrtrUr'rrKrMrrrSrwrrrrrs rWtest_2D_vs_RGI_linearz TestMakeND.test_2D_vs_RGI_linear s**,,1aAq61***%q!faAAA X6666668999:  R##b''666666rkc"|\}}}t||f|dtj}t ||f|d}t jgdgdgj}t||||ddS) Nr;rH cubic_legacyrrWrZrsrt rUr'rKrLrrKrMrrr\s rWtest_2D_vs_RGI_cubiczTestMakeND.test_2D_vs_RGI_cubic s**,,1aAq61#+>>>%q!faGGG X6666668999:  R##b''666666rkrIc|\}}}t||f|d|d}t||f|d}tjgdgdgj}t ||||dd dS) Nr;ư>)r4rIr]r_rrWrZrsrr[r[)rSrIrwrrrrrs rWtest_2D_vs_RGI_cubic_iterativez)TestMakeND.test_2D_vs_RGI_cubic_iterative s **,,1aAq61&tDDD%q!faGGG X6666668999:  R##b''DAAAAAArkc"|\}}}t||f|dtj}t ||f|d}t jgdgdgj}t||||ddS) NrrHquintic_legacyrrWrZrsrtr`r\s rWtest_2D_vs_RGI_quinticz!TestMakeND.test_2D_vs_RGI_quintic s**,,1aAq61#+>>>%q!fa8HIII X6666668999:  R##b''666666rkzk, meth))r.rA)r;r_)rrfcVtjd}tj|d}tj|d}tj|d}|d}t |||f||t j}t|||f||} tjd d d } t|| | | d dS)Ni@rrrrr)rHrgffffff?rrCrrsrt) rKrRrcumsumrr'rKrLrr) rSr4methrndmrwrrrrrrs rWtest_3D_random_vs_RGIz TestMakeND.test_3D_random_vs_RGI sy$$V,, Idlll** + + Idlll** + + Idlll** + +9--Aq!9f#+FFF%q!QiEEE Y  3Sw  ? ?R##b''666666rkch|\}}}ddi}ttd5t||f|fddi|dddn #1swxYwYttd5t||ft j||ffddi|ddddS#1swxYwYdS)Nmaxiterr.rIrr4r;)rUrGrIr'rKr)rSrwrr solver_argss rWtest_solver_err_not_convergedz(TestMakeND.test_solver_err_not_converged sg**,,1a !n :X 6 6 6 7 7 A 6 6Q 6+ 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7:X 6 6 6 G G A 1a& 1 1 F FQ F+ F F F G G G G G G G G G G G G G G G G G Gs#AAA3'B''B+.B+N)r!r"r#rErOrarrrrSrUr]rarKgmresgcrotmkrdrgrlrpr$rkrWr>r>k sA777(BBB4 [S"J"J"JKK>>LK>--- 777777 [X 3;'?@@ B BA@ B777 [NNN 7 7 7GGGGGrkr>cJeZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) TestFpcheccd}tjddd}tjd}tjt d5t j|||dddn #1swxYwYtjt d5t j|||ddddS#1swxYwYdS)Nr.rr9rz 1D sequencer)rKrOrrar rI_bfpcheckrSr4r2rws rW test_1D_x_tzTestFpchec.test_1D_x_t s>  IbMM ! !!Q ' ' IbMM ]:] ; ; ;  Jq!Q                  ]:] ; ; ;  Jq!Q                     s$A==BB#CC C cd}d|dzzdz }|dz}tj|}tj|}tj|||dksJt jt d5tj|||dddn #1swxYwYd|dzzdz}||z dz }tj|}tj|}tj|||dksJt jt d5tj|||ddddS#1swxYwYdS)Nr;r9r.rDrez Need k+1*r) rKrOdfitpackfpchecrar rIrvrw)rSr4rUrr2rws rWtest_condition_1zTestFpchec.test_condition_1 s  AY] F IaLL IaLLq!Q''2---- ]:[ 9 9 9  Jq!Q                   qsGaK EAI IaLL IaLLq!Q''2---- ]:[ 9 9 9  Jq!Q                     s$2BBBD66D:=D:cd}dg|dzzdgzdg|dzzz}gd}tj|||dksJtj|||J|}|d|d<tj|||dksJt jtd 5tj|||dddn #1swxYwY|}|d|d<tj|||dksJt jtd 5tj|||ddddS#1swxYwYdS) Nr;rr.r9r)r.r9r;r<rr7rz Last k knots*rzFirst k knots*r{r|rvrwrrar rI)rSr4r2rwtts rWtest_condition_2zTestFpchec.test_condition_2( s  C1IOqc1Q3i '   q!Q''1,,,,z!Q""*** VVXXA2q"a((B.... ]:_ = = = ! ! Jq"a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! !VVXX21q"a((B.... ]:-= > > > ! ! Jq"a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s$(C  CC,EEEcd}dg|dzzddgzdg|dzzz}gd}tj|||dksJtj|||Jdg|dzzddgzdg|dzzz}tj|||dksJt jt d 5tj|||ddddS#1swxYwYdS) Nr;rr.r9rr.r9r;rr<rzInternal knots*r)r{r|rvrwrar rIrxs rWtest_condition_3zTestFpchec.test_condition_3> sF  C1IA !ac * " " "q!Q''1,,,,z!Q""*** C1IA !ac *q!Q''2---- ]:-> ? ? ?  Jq!Q                     s"CC  C cdd}dg|dzzdg|dzzz}gd}tj|||dksJtj|||J|}|d|d<tj|||dksJtj|||J|}|ddz |d<tj|||dksJt jtd5tj|||dddn #1swxYwY|}|d dz|d <tj|||dksJt jtd5tj|||ddddS#1swxYwYdS) Nr;rr.rrr8zOut of bounds*rr7r)rSr4r2rwrhs rWtest_condition_4zTestFpchec.test_condition_4K sG  C1IQqS ! " " "q!Q''1,,,,z!Q""*** VVXX!1r1a((A----z!Q""*** VVXX!q1r1a((B.... ]:-= > > > ! ! Jr1a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! !VVXX22r1a((B.... ]:-= > > > ! ! Jr1a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s$:DD!$D!F%%F),F)cd}gd}gd}tj|||dksJtjtd5t j|||dddn #1swxYwYgd}tj|||dksJtjtd5t j|||ddddS#1swxYwYdS)Nr.)rrr.r9r9)rrrrqSchoenberg-Whitney*r)rrrr{r|rar rIrvrwrxs rWtest_condition_5_x1xmz TestFpchec.test_condition_5_x1xmj so  OO OOq!Q''2---- ]:-B C C C  Jq!Q                   OOq!Q''2---- ]:-B C C C  Jq!Q                     s$A%%A),A)+CCCcd}gd}ddg}tj|||dksJtj|||JdS)Nr.)rrr.r.rg333333?r)r{r|rvrwrxs rWtest_condition_5_k1zTestFpchec.test_condition_5_k1x sX  LL #Jq!Q''1,,,,z!Q""*****rkcd}dg|dzzdgzdg|dzzz}dgdz}tj|||dksJtjtd5t j|||dddn #1swxYwYdg|dzzdgzdg|dzzz}dgdz}tj|||dksJtjtd5t j|||ddddS#1swxYwYdS) Nr;rr.r9rrqrrrrxs rWtest_condition_5_1zTestFpchec.test_condition_5_1 s  C1IOqc1Q3i ' CEq!Q''2---- ]:-B C C C  Jq!Q                  C1IOqc1Q3i ' CEq!Q''2---- ]:-B C C C  Jq!Q                     s$A::A>A>C==DDcd}dg|dzzddgzdg|dzzz}dgdzdgz}tj|||dksJtjtd 5t j|||dddn #1swxYwYdgdzddgz}tj|||dksJt j|||JdS) Nr;rr.r9rrr<rqrrrrxs rWtest_condition_5_2zTestFpchec.test_condition_5_2 s2  C1IA !ac * E!GqcMq!Q''2---- ]:-B C C C  Jq!Q                  E!Gq!f q!Q''1,,,,z!Q""*****sA??BBcd}gd}gd}tj|||dksJtjtd5t j|||ddddS#1swxYwYdS)Nr.) rrr9r;r<rrrr)r.r.r.@rrrrqrrrrxs rWtest_condition_5_3zTestFpchec.test_condition_5_3 s  ' ' ' ) ) )q!Q''2---- ]:-B C C C  Jq!Q                     sA&&A*-A*N) r!r"r#ryr}rrrrrrrrr$rkrWrtrt s       *!!!,    !!!>    +++     + + +     rkrtctj|||| fdttdz D}dd}tt|D]0}||dz }||xx|z cc<||dzxx|zcc<1|dxxdz cc<t t |t d|fS)z)Split the knot interval into "runs". cfg|]-}||dz.Sr)r)rrrm residualss rWrz_split.. s: G G Gi1b1g &**,, G G Grkr.r7r9rZrt)rKrrr=rr) rwr2r4rfpartscarriesrcarryrms ` @rW_splitr s AadG $ $B G G G G GeCGGAI6F6F G G GF1R4!G 3w<<  Qq U qs u  2JJJ)B-JJJCKKYe<<<< 2:rkct||||\}}d}d}tt|D]0}||dz||z dkr|||kr |}||}1|dkrtd||||dzzdzdz} || } t j|| } tj|d| | || df} | S)z Insert a new knot given reduals.ig}Ô%Ir.z5Internal error, please report it to SciPy developers.r9N)rrr=rIrKrr) rwr2r4rrrmidx_max fpart_maxr idx_newknotnew_knotidx_tt_news rW _add_knotr s1a++JFBGI 3v;;  "" ac7RU?Q  6!9y#8#8Gq I$PQQQg;GAI.2q8K~H OAx ( (E E!FUF)Xqy0 1E Lrkc(eZdZdZejdgddZdZdZ dZ dZ d Z d Z d Zejd gd ejdgddZdZdZdS)TestGenerateKnotsc2tjdt}|dzdd|zz z}d}tjdg|dzzdg|dzzz}t ||||}|||z d z}d d lm}|||||}t||||} t|| d t ||||} | ||z d z} ||||| } t|||| } t| | d dS)NrrEr;r0r.r=r)r4r2r9r_fitpack_reprorZrt) rKrOrrMrrCradd_knotrr)rSrwrr4r2r2r_frnew_tnew_t_pyspl2 residuals2new_t2 new_t2_pys rWtest_split_add_knotz%TestGenerateKnots.test_split_add_knot sH Iau % % % qD2q1u:   HbT1Q3Z2$!*, - -aa1---SVVaZ!O ;;;;;; Q1i00Q1i00xe4444q!qE222d1ggkA% a:66a:66  666666rkr4rcX|d|j}|||jzdz }t t |||dd}t |||dd}|||j}t||ddS)NrrErr4rr7rZrt) rOrPrrrr"rrQr)rSr4rTrwrr2rs rWtest_s0zTestGenerateKnots.test_s0 s IIarzI * * FF1RU719   1Q/// 0 0 4 AqA # # #A & ZZ"*Z - -2E******rkc&d}|||j}|dz}tt||dd}|t t j|d}t|d|ddS) NrerEr;rrr7rZrt)rOrPrr"rQrrKrrSrTrUrwrknotsris rW test_s0_1zTestGenerateKnots.test_s0_1 s  IIarzI * * qD^AqA33344::k"*Q--;;<<b 8%888888rkcd}||}|dz}tt||dd}|t t j|d}t |d|ddS)Nrr;rrr7rZrt)rOrr"rQrrKrrs rW test_s0_n20zTestGenerateKnots.test_s0_n20 s  IIaLL qD^AqA33344::k"*Q--;;<<b 8%888888rkc tjd}|dz}tt5t t ||dddddddS#1swxYwYdS)Nrer;rr4rnest)rKrOrGrIrr"rSrwrs rW test_s0_nestzTestGenerateKnots.test_s0_nest s IbMM qD : & & : : 1QR888 9 9 9 : : : : : : : : : : : : : : : : : :s"AA!$A!c$dj}|tjzdz }d}t t |||d}gdgdgdgd gd g}fd |D}t|t|ksJt||D]\}}t||d t|||d\}} } t|d |d dS)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. rrEr;rr)r=r=r=r=rrrr) r=r=r=r=r@rrrr r=r=r=r=r>r@rrrr) r=r=r=r=r>r@rrrrr) r=r=r=r=r>r?r@r[rrrrcHg|]}|jS)rE)rQrP)rwantrTs rWrz3TestGenerateKnots.test_s_switch..4 s+HHH"**T*44HHHrkrZrtr7N) rOrPrrKrrr"r=rrrrQ) rSrTrwrr4rwantedr2rrs ` rW test_s_switchzTestGenerateKnots.test_s_switch sC. IIarzI * * FF1RU719   ^AqA66677222666:::>>>AAA   IHHHHHH5zzS[[(((('' / /EAr Ar . . . . .Ad+++1ab 2::a==u======rkcttd}t||dd}t|dS)Nrrr.)rr4)rrr"next)rSrwgens rWrz!TestGenerateKnots.test_list_input> s8 qNNQSA... S rkc |d|j}|||jzdz }d}t t ||d|d}t |d|gd|jd tt5t t ||dd ddddS#1swxYwYdS) NrrErr;rerr7rrZrtr<)r4r) rOrPrrrr"rrQrGrI)rSrTrwrrrs rW test_nestzTestGenerateKnots.test_nestD s) IIarzI * * FF1RU719   ^AqA<<<== "I JJ???rzJ R R    : & & 4 4 1222 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s(!CCCc tjd}tj|tjzdz }t t 5t t||tjddddn #1swxYwYt t 5t t||tjd ddddS#1swxYwYdS)NrrDr) rKrOrrrGrIrr"rrs rWrzTestGenerateKnots.test_weightsU sK IaLL F1RU719   : & & 8 8 1 " 666 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8: & & 6 6 1 444 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s$ 2B  BB*3C**C.1C.npts)rrqrr)r{Gz?rctjd}dtj||z}tj|tjzdz tj|dz dz z}d}t||||d}tt||||d }t||d dS) Nrrerrr9r;rrr7rZrt) rKrRrrrrrexprrr"r) rSrrrkrwrr4r2rs rWtest_vs_splrepz TestGenerateKnots.test_vs_splrep_ sy$$U++ rwt|||..// / F1RU72:  !A# !2!2 2  1a1 " " "1 % .Aa000 1 1" 5AE******rkcfd}tj|}|dz}tt||dd}t jt 5}t||dd}dddn #1swxYwYt|dksJt|d|ddS)Nr;Jz5rr.r7r) rKrOrr"rawarnsRuntimeWarningrr=r)rSrUrwrrrr`s rWtest_s_too_smallz"TestGenerateKnots.test_s_too_smallr s  IaLL qD^AqA77788 \. ) ) -QAe,,,C - - - - - - - - - - - - - - -1vv{{{{b 3q6*****sA55A9<A9ctgdgdgdd}tjtd5t |ddddS#1swxYwYdS)N)r=r0r>r?)r@r[rr)r=r=r=r=r.)rrzweights are zeror)r"rar rIr)rSrs rWtest_zero_weightsz#TestGenerateKnots.test_zero_weights s]]]MMM]]]aPPP ]:-? @ @ @   III                  sAAAN)r!r"r#rrarrrrrrrrrrrrrr$rkrWrr s07770 [S///22++32+999999:::*>*>*>X 444"666 [V]]]33 [S...11++2143+" + + +rkrc|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. rr.r9NrrrE) rrKrr r<emptyrrr  factorial) r2r4rUdeltanrintrtiirmatrrs rW disc_naiver sq  A a!eaiL1Q4 E !GaKE 1Q3qs1u9B )B  C!HHHHHH1IIIIII(26!a%!)$$a((333A 8U1Wagaj) 7 7 7D 1agaj! $ $,,!QQQ$!AaCF)+QT111W U5[1 t~a00 00D Krkc eZdZdZddZdZdS)F_densezm The r.h.s. of ``f(p) = s``, an analog of _fitpack_repro.F Uses full matrices, so is for tests only. NTc||_||_||_||_|t j|t n||_||_|jj dksJt|t j |j d|z dz |||}||jdddfz|_ ddlm} t!| |||_|j dksJ||jz} tj| t j|jj df|_||_dS)NrEr.rrr)rwrr2r4rKrnrrrrr r<ra_denserCrrdiscrb_denserr/rr) rSrwrr2r4rrrrrrs rWr zF_dense.__init__ s512au----&v{aU'!RVAGAJNQ$677 TTTUVWW40 ;;;;;;#SXXa^^4<<>> v{{{{ Z%BHT\%7%:;;;<rkctj|j|j|z f}ddlm}m}||d\}}|j|jz}|j d}||d|d|f|d|} t|j | |j |j } tj|jdz| |j|jz dzz} | |_| |jz S)Nr)rreconomic)rtr.rr9)rKvstackrr scipy.linalgrrrrrr r2r4rrrrwrr2r) rSrabrrr2rqyr r3r2fps rWrzF_dense.__call__ s Y dlQ&67 8 8 +*******r":&&&1 S47] WQZ E!CRC"H+r#2#w ' 'dfaT5EFFF VDFAITVtv!5 99 : :DF{rk)NT)r!r"r#rr rr$rkrWrr sA.rkrceZdZdZefdZdZdZej dgddZ dZ ej dgdd Z ej jd Zej dgd d Zd Zej dddgej dgddZej dddgdZej dddgej dgddZej dgdej dgddZej dgdej dgddZdS)_TestMakeSplrepBaseNc|jdkr@|ddtjzd}||}d}|||fS|ddd}||dzdz dz}d}|||fS) Nrrr9regq@H&?rrDgQ @)r:rfrKrr)rSrTrwrrs rW _get_xyktz_TestMakeSplrepBase._get_xykt s <: % % Aqw++AAAa7N Aq"%%A$ ##Q&AAa7Nrkc $tjddd}tjddd}tt5t |||jdddn #1swxYwYtt5t ddd|j dddn #1swxYwYtt5tj|jdddf}t ||d|j dddn #1swxYwYtjd}tt5t ||d z|d|j dddn #1swxYwYtjd }tt5t ||d z|d|j dddn #1swxYwYtj|jddf}tt5t ||d z|d|j dddn #1swxYwYtt5t |ddd |d zd|j dddn #1swxYwYtt5t ||d zd d|jdddn #1swxYwYtt5t ||d zd |j dddn #1swxYwYtt5t ||d zd dd|jdddn #1swxYwYtt5t ||d zdd|jdddn #1swxYwYtt5t tj dtj dd|j ddddS#1swxYwYdS)NrrerDrr^r.r9rrr:r;)rrr:r7rBr4rr:)r4rrr:)rrr:r ) rKrfrGrIr#r:rrrrHrOrSrwrrs rWtest_input_errorsz%_TestMakeSplrepBase.test_input_errors s= K2r " " K2r " " : & & 4 4 1dl 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4: & & ; ; 1T\ : : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;: & & ; ;Aq)**A 1T\ : : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; GBKK : & & C C 1a41T\ B B B B C C C C C C C C C C C C C C CWR[[L : & & C C 1a41T\ B B B B C C C C C C C C C C C C C C C GQWQZO $ $ : & & C C 1a41T\ B B B B C C C C C C C C C C C C C C C: & & D D $$B$Adl C C C C D D D D D D D D D D D D D D D9 % % E E 1a43#t| D D D D E E E E E E E E E E E E E E E: & & = = 1a42t| < < < < = = = = = = = = = = = = = = =: & & K K 1a411T\ J J J J K K K K K K K K K K K K K K K: & & E E 1a412t| D D D D E E E E E E E E E E E E E E E: & & Q Q  ! billc4< P P P P Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q QsA%%A),A)B))B-0B-6D  DD=E&&E*-E*GG GH,,H03H0 %I<<JJKK K#L  LL*MMM3NN #N ;=PP  P c Xtt|||||jd}t|||||j}t||||||j}t |j|jdt |j|jd|j|jksJdS)Nrr7)r2r4rr:rZrt)rr"r:r#rr2r3r4)rSrwrr4rr2spl_autospl_ts rW_test_with_knotsz$_TestMakeSplrepBase._test_with_knotss 1Q EEE F Fr Jq!qAt|DDDAqAaFFF EG%8888 EG%8888zUW$$$$$$rkr4r.r9r;r<rrcf|\}}}|||||dSr)rr)rSr4rwrrs rWtest_with_knotsz#_TestMakeSplrepBase.test_with_knots!s7..""1a aAq)))))rkcPt||||j}t||||j}tt ||||jd}t |j|jdt |j|dt |j|ddS)Nr9r7rZrt)r#r:r rr"rr3r2)rSrwrr4r2spl_ir2s rW_test_default_sz#_TestMakeSplrepBase._test_default_s's!Q!T\:::"1a1dlCCC 14<@@@ A A" EuwU3333qu----//////rkcd|\}}}||||dSr)rr)rSr4rwrrs rWtest_default_sz"_TestMakeSplrepBase.test_default_s0s5..""1a Q1%%%%%rkc d}|jdkr8tjddtjzd}tj|}ntjd}|dz}t j5t jdtt||d||jdk}dddn #1swxYwYt j5t jdtt||d||j }dddn #1swxYwYt|j |dttj|jdg|jd zzf|d d dS) NgKH9rrr9rr;r5r4rrrr.r+rt)r:rKrfrrrOwarningscatch_warnings simplefilterrrr#rr2rr3r4)rSrrwrr`r2s rWrz$_TestMakeSplrepBase.test_s_too_small5s  <: % % Aqw++Aq AA " A1A  $ & & K K  !   Aadlj.HJJJC  K K K K K K K K K K K K K K K $ & & D D  !   aa1dlCCCC  D D D D D D D D D D D D D D D s1v&&&ceaS#%']23AU , , , , , ,s$28B66B:=B:4DDDr.cd}|jdkr8tjddtjz|}tj|}ntj|}|dz}t ||||j}t |||d|j}|jjd ksJ|jjd ksJt ||d d |zz z|d|j}|jjd ksJdS) Nrerrr9r;r9h㈵>rr.) r:rKrfrrrOr#r3r)rSr4rUrwrr2rrs rW test_shapez_TestMakeSplrepBase.test_shapeRs  <: % % Aqw**Aq AA ! A1A!Q!T\:::AqAt|DDDuzQw|q    Aq1ac7{a4NNNw|q      rkcd}ddtjz}}tj|||dz}tj|}t t 5t ||ddddddS#1swxYwYdS)Nrerr9r.:0yE>nonsenser)rKrrfrrGrIr#rSNrrVrwrs rWtest_error_on_invalid_bc_typez1_TestMakeSplrepBase.test_error_on_invalid_bc_typegs !BE'1 K1a!e $ $ F1II : & & : : 1j 9 9 9 9 : : : : : : : : : : : : : : : : : :sA66A:=A:r:rrc tjddtjzd}tj|}tjddt |}t ||||d|dk}t |||d|| }t|j|dttj |j dg|j dzzf|dd dS) Nrr9rer.rrrrr4rrrrr4r:rt) rKrfrrr=rr#rr2rr3r4)rSr:r4rwrrr`r2s rW%test_make_splrep_with_unequal_weightsz9_TestMakeSplrepBase.test_make_splrep_with_unequal_weightsps K1RU7B ' ' F1II K1c!ff % %QQ!t'Z2GIII!Q!tq'BBBs1v&&&ceaS#%']23AT + + + + + +rkc d}tjddtjzd}tj|}tj||fj\}}|jjdusJ|jjdusJtjddt|}tj||fj\}} |jjdusJt|||dd |d k } t|||d d| } || | t|||d d| } || | t|||d d| } || | t|||d d| } || | dS) Nct|j|dttj|jdg|jdzzf|dddS)Nrr.rrt)rr2rKrr3r4)r2r`s rWcheckzO_TestMakeSplrepBase.test_make_splrep_with_non_c_contiguous_input..checks^ CE3q6 * * * BE#%!ceAg"67F / / / / / /rkrr9reFr.rr;rrrr) rKrfrrrrr0r1r=rr#) rSr:rrwrrrrw1rr`r2s rW,test_make_splrep_with_non_c_contiguous_inputz@_TestMakeSplrepBase.test_make_splrep_with_non_c_contiguous_inputs / / / K1RU7B ' ' F1IIq!tBx$----x$---- K1c!ff % %ad Ax$----QQ!t'Z2GIII"a17CCC c3!R17CCC c3!Q"7CCC c3"bB$!")+++ c3rkc tjddd}tjdtjz|z}|d|d}}d}d}tj|g|dzz|g|dzzz}t |||||||d| } t | tsJ| jj dd|dzzksJ| j |ksJt| jd|dztj |g|dzzt| j|dz dtj |g|dzzdS) Nrr.rer9r7r;r)xbxer4rr2rr:) rKrfrrrMr#rr r2rr4rrQ) rSr:r4rwrrrrr2r2s rW%test_make_splrep_impl_no_optimizationz9_TestMakeSplrepBase.test_make_splrep_impl_no_optimizationsR K1b ! ! F1ru9q= ! !1quB   HbTQU^rda!en4 5 5!Q2"qtW>>>#w'''''u{1~a!e,,,,uzzzzdqsd RZA%?%?@@@!fgg B41q5>(B(BCCCCCrkrU)r3r rDr)rerrr.rc `tjd}tjdtj|ddtjz|dz dtjzf}tj|tj|z}t||||j dk\}}}t||||j } |dkr|d kr |j dks/*CE>>>>>>rkchd}|\}}}tt||||d}t||||}t|||||}t |j|jdt |j|jd|j|jksJdS)Nr;rr7)r2r4rrZrt)rrr"r#rr2r3r4)rSr4rwrrr2rrs rWrzTestMakeSplrep.test_with_knotss ..""1a 1Q/// 0 0 4q!qA...AqAa000 EG%8888 EG%8888zUW$$$$$$rkcd}|||j}|dz}d}t|||d}|jjdd|dzzksJdS)NrerEr;r.rrr9)rOrPr#r2r)rSrTrUrwrr4r2s rWtest_no_internal_knotsz%TestMakeSplrep.test_no_internal_knots*si  IIarzI * * qD !Q!q)))u{1~AaC((((((rkcd}|||j}|dz}t||d}t||d}t |j|jddS)NrerEr;rSrZrt)rOrPr#r rr3)rSrTrUrwrr2rs rWrzTestMakeSplrep.test_default_s3sn  IIarzI * * qD!Q!$$$"1a1---uwU333333rkcd}tj|}|dz}tjt5}t ||dd}t ||dd}t|j|dttj |j dg|j dzzf|dddddn #1swxYwYt|d ksJdS) Nrr;rrrr.r+rtr9)rKrOrarrrr#rr2rrr3r4r=)rSrUrwrrr`r2s rWrzTestMakeSplrep.test_s_too_small<s  IaLL qD \. ) ) 0QAe,,,Caa5111C CE3q6 * * * BE#%!ceAg"67F 0 0 0 0  0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1vv{{{{{{sA=B>>CCcntjgdtj}tjgdtj}tjdg|jdztj}t t 5t |||ddddddS#1swxYwYdS) 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@rCghs@rE)gJ@gH@gLH@gffffffG@gE@g)\(D@gC@gB@g333333@@g<@g3333339@g5@g2@g,@g%@g@g @r=g5{2?rr9r)rr4r)rKrQrPrrGrIr#rs rWtest_issue_22704zTestMakeSplrep.test_issue_22704Js  J%%%-/J 8 8 8 J&(Z 1 1 1 Jy171:-RZ @ @ @ : & & . . 1Q" - - - - . . . . . . . . . . . . . . . . . .sB**B.1B.c@d\}}|||j}|dz}t|||}t|||d}|jjdksJ|jjdksJt||dd|zz z|d}|jjdksJdS)Nrer;rEr;rSr rr.)rOrPr#r3r) rSrTrUr4rwrr2rrs rWr zTestMakeSplrep.test_shapeXs1 IIarzI * * qD!Q!$$$AqA...uzQw|q    Aq1ac7{a4888w|q      rkc^d\}}|||j}|dz}t||dd}t||dd}|jjdksJ|jjdksJ|jjd||zdzksJ|jjdd|dzzksJdS)NrFrEr;rrr.r9)rOrPr#r3rr2r)rSrTrUr4rwrrrs rWtest_s0_vs_notzTestMakeSplrep.test_s0_vs_noths1 IIarzI * * qDAqA+++AqA+++w|q    w|q    w}Q1q519,,,,w}Q1A;......rkN)r!r"r#rarrr1r5r:r<rr?rrrDr rHr$rkrWr)r)s [S"4"4"455 6 665 6 [S"4"4"455@@65@ 0 0 0 ? ? ? % % %)))444    . . .!!! / / / / /rkr)ceZdZdZejdgddZejdgddZdZ dZ ejdd d gd Z ejd gd ejddgdZ dS)TestMakeSplrepPeriodicrr4rc|ddd}|d}t|||d|j}|jjdd|dzzksJdS)NrrerDrr.rr9)rfrr#r:r2r)rSr4rTrwrr2s rWr?z-TestMakeSplrepPeriodic.test_no_internal_knots}si KK2r " " GGENN!Q!q$,???u{1~AaC((((((rkcd}tjddtjz|}tj|tj|z}t |||d|j}t |||d|j}|jjdksJ|jjdksJ|j j d|d|zzksJdS)Nrerr9rr.) rKrfrrrr#r:r3rr2r)rSr4rUrwrrrs rWrHz%TestMakeSplrepPeriodic.test_s0_vs_nots  K1RU7A & & F1IIq !AqADLAAAAqADLAAAw|q    w|q    w}Q1q1u9,,,,,,rkcd}dd|jz}}||||dz|j}||}t ||d|j}t t|||dd ||||z}t ||d |j}t t|||dd d ||z||dzz}t ||d|j}t t|||dd dS)Nrerr9r.rErrr rYr[rrcrr;) rrfrPrr#r:rrr)rSrTrrrVrwrr2s rW test_periodic_with_periodic_dataz7TestMakeSplrepPeriodic.test_periodic_with_periodic_datas3 !BE'1 KK1a!e2:K 6 6 FF1II!Q$ ===a qt$???? FF1IIq !!Q%>>>a qt$???? bffQiiK"&&))A+ %!Q$ ===a qt$??????rkcd}ddtjz}}tj|||dz}tj|}t t 5t ||d|jddddS#1swxYwYdS)Nrerr9r.rr)rKrrfrrGrIr#r:rs rW$test_periodic_with_non_periodic_dataz;TestMakeSplrepPeriodic.test_periodic_with_non_periodic_datas !BE'1 K1a!e $ $ F1II : & & < < 1dl ; ; ; ; < < < < < < < < < < < < < < < < < TestMakeSplrepPeriodic.test_make_splrep_periodic_m_eq_2_k_eq_1s Hc3Z  Hc3Z $%EEBHc3Z t!Q!q*BBBQQ!a000 q55 CE3q6 * * *ceaS#%']23A     rkk_fp))r.-C6)r9rT)r;gAȘrYc dd}tjdd|}tjdtjz|z}|\}}t ||||d}d} t ||dddf|d||} t j5t jdtt| d d| |z ftj ||z ff|d z} dddn #1swxYwY| j } tj d d g} | | }t|d|ddS) Nrerr.r9rrrr5r>MbP?r=r0)rKrfrrrr rrr rr!rNr2rMr)rSrSrrUrwrr4rr`fp0splinerbsx_checky_checks rWtest_fperiodic_basic_fitz/TestMakeSplrepPeriodic.test_fperiodic_basic_fits  K1a  F1ru9q= ! !2QQ!+++1a4j#a&A;;;  $ & & T T  !(N ; ; ;&#C!G rvrAv6F'GUSSA T T T T T T T T T T T T T T T Z(C:&&"W++ GAJ/////sAC!!C%(C%N) r!r"r#r:rarrr?rHrNrPrRr\r$rkrWrJrJxs+G [S"4"4"455))65) [S"4"4"455 - -65 -@@@"<<< [S1e*--   .-   [V%Q%Q%QRR [S4&))00*)SR000rkrJceZdZddefdZejdgddZejdgddZ dZ d Z d Z d S) TestMakeSplpreprer;c|||j|jz|z }||||g}|||fS)NrE)rOrPrrr)rSrr4rTrwrs rW_get_xykzTestMakeSplprep._get_xyksN IIarzI * *RU 2Q 6 VVAYYq "!QwrkrrrrVr c `d\}}tj|tjz|z }tj|tj|g}ddddd}t ||\\}}}} t ||\} } t| | d t| j|d t|||ksJtj |j } t| j | d t| | t|||d | d dS) NrFrrrrerarrZrtr.r)rKrOrrrrr$rr2r=rQrr3r ) rSrrr4rwr num_knotsr2r3u_r2rrs rWtest_simple_vs_splprepz&TestMakeSplprep.test_simple_vs_splprepsJ1 IaLL25 1 $ VAYYq "%u==  Q Aq2a1%%%Q 2E**** qu----1vv1%%%%Z]]_r.... A01aa00033% A A A A A Arkc|\}}}t|tsJtj|ddksJt ||\}}t tj||\}}t|||t|d|ddt|dt|dksJt|d|dd|d|dksJtjt||tj|ksJt||\}}t||dt|j |ddt|j j |dd|j|dksJ||jtj|ksJttj||\}}t||dt|j |ddt|j j |dd|j|dksJ||jtj|ksJtt 5ttj|j |ddddS#1swxYwYdS)Nrr9rcrtrZr.)r`rrrKrrrQrr=rr$r2r3rr4rGrI) rSrrrr`rtck_au_ar2s rWtest_array_not_listz#TestMakeSplprep.test_array_not_lists--//1a!T"""""x{{1~""""aQRZ]]a000 s3Q''''Aau55553q6{{c%(mm++++Aau55551vq!!!!xa &&"(1++5555a1%%%Q3U++++uQxe4444q6666ua    s1vv|rx{{****bjmmq111Q3U++++uQxe4444q6666ua    s1vv|rx{{**** : & & / / AA . . . . / / / / / / / / / / / / / / / / / /s:)K00K47K4c|d|\}}}t|\}}t||||ddS)NrerrTrZrtr`r$rrrSrTrwrr4r2rs rWtest_default_s_is_zeroz&TestMakeSplprep.test_default_s_is_zerosY--"-,,1aaQA %888888rkc|d|\}}}t|d\}}t|d\}}t||dt||||dt||||d|j|jksJ|jj|jjksJdS)NrerlrrcrZrtr)r`r$rrrr3r) rSrTrwrr4ru_ispl_nu_ns rWtest_s_zero_vs_near_zeroz(TestMakeSplprep.test_s_zero_vs_near_zero%s--"-,,1a!!q))) s!!u--- sSu----c BHHQKKe<<<<c BHHQKKd;;;;zUZ''''w} ------rkc^tjdt}tt5t |dddn #1swxYwYtt5t |ddddn #1swxYwYtt5t |ddddn #1swxYwYt |gd\}}t |gd\}}||jdksJt|||gd dS) NrrErrcrr r.rrZrt) rKrOrrGrIrr$rr)rSrwr`rer2rs rWrzTestMakeSplprep.test_1D2s Iau % % % : & &   AJJJ               : & & ! ! a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! !: & & # # c " " " " # # # # # # # # # # # # # # #1#&&&RqcT***Qs1vv|v%%%%A%000000s5A  AA+B  B B (CC  C N r!r"r#rKr`rarrrfrjrortrr$rkrWr^r^sqR  [S"6"6"677AA87A: [S"6"6"677//87/B999 . . .11111rkr^ceZdZddefdZejdgddZejdgddZ dZ d Z d Z d S) TestMakeSplprepPeriodicrer;c|dd|jz||j}||||g}|||fS)Nrr9rE)rfrrPrr)rSrUr4rTrwrs rWr`z TestMakeSplprepPeriodic._get_xykGsL KK1RU7ARZK 8 8 VVAYYq "!QwrkrrrYr rcc d}tjddtjz|}tj|tj|g}ddddd}t ||d\\}}}} t ||d \} } t| | d t| j ||ksJt| | t|||d | dddS)Nrerr9rr.r{r.r"rrrZrtrrcr[) rKrfrrrrr$rr=r2r ) rSrrUrwrrdr2r3r4rer2rs rWrfz.TestMakeSplprepPeriodic.test_simple_vs_splprepLs   K1RU7A & & VAYYq ""Bb99  QA... Aq2a1j999Q 2E****35zzYq\)))) A 81aa 8 8 8 ; ;" 0 0 0 0 0 0rkc4|\}}}t|tsJtj|ddksJt ||d\}}t tj||d\}}t|||t|d|ddt|dt|dksJt|d|dD]\}} t|| d|d|dksJtjt||tj|ksJt||d\} }t||d| j |dksJ| |jtj|ksJttj||d\} }t||d| j |dksJ| |jtj|ksJtt5ttj|j|dddddS#1swxYwYdS) Nrr9r.r"rtrZrr)r`rrrKrrrQrr=rrr$r4rGrIr) rSrrrr`rrhrirrr2s rWrjz+TestMakeSplprepPeriodic.test_array_not_listes--//1a!T"""""x{{1~""""aQ'''QRZ]]aQ777 s3Q''''Aau55553q6{{c%(mm++++#a&%(++ 0 0FB B / / / / /1vq!!!!xa &&"(1++5555a1j999Q3U++++ua    s1vv|rx{{****bjmmq*EEEQ3U++++ua    s1vv|rx{{**** : & & C C AAz B B B B C C C C C C C C C C C C C C C C C Cs*J  JJc|d|\}}}t|d\}}t||||ddS)NrerUrTrr^rZrtrmrns rWroz.TestMakeSplprepPeriodic.test_default_s_is_zeros`--"-,,1aa444QA %888888rkcp|d|\}}}t|dd\}}t|dd\}}t||d||} t||| dt||| d d |j|jksJdS) NrerrrrrrZrtrrcr[)r`r$rrr) rSrTrwrr4rrqrrrsy_arrs rWrtz0TestMakeSplprepPeriodic.test_s_zero_vs_near_zeros--"-,,1a!!q*=== s!!ujAAA sSu---- c E6666c E4@@@@zUZ''''''rkcvtjddtjzd}tj|}t t 5t |ddddn #1swxYwYt t 5t|dddddn #1swxYwYt t 5t|dddddn #1swxYwYt|gd d\}}||jd ksJt|||gd dS) Nrr9rr.rrrrrZrvrt) rKrfrrrGrIrr$rr)rSrwr2rs rWrzTestMakeSplprepPeriodic.test_1Ds K1RU7A & & F1II : & &   A1                   : & & 5 5 a 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5: & & 7 7 c: 6 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7qcUJ???Qs1vv|v%%%%A%000000s6 A**A.1A. B((B,/B,C&&C*-C*Nrwr$rkrWryryDsqR  [S"7"7"788009800 [S"7"7"788CC98C<999 ( ( (11111rkryc.eZdZdZdZddZdZdZdS) BatchSplinec tj||d}|jdd|_fd|d|jdD|_||_dS)Nr7c$g|] }|fi Sr$r$)ryikwargsrXrws rWrz(BatchSpline.__init__..s-VVVR20000VVVrk)rKmoveaxisr _batch_shaper_splines_axis)rSrwrrrXrs ` ``rWr zBatchSpline.__init__sk K4 $ $GCRCLVVVVVV199RQS;U;UVVV  rkcfd|jD}tj||jjz}jrtj|d|jn|S)Nc&g|] }|Sr$r$)rrXrws rWrz(BatchSpline.__call__..s! 3 3 36VVAYY 3 3 3rkr7)rrKrrrrrrs ` rWrzBatchSpline.__call__s[ 3 3 3 3T] 3 3 3 Jq$+ag5 6 612?r{1b$*---a?rkNcffd|jD}tj||jS)Nc>g|]}|Sr$)r)rrXrrVrs rWrz)BatchSpline.integrate..s+ M M MVV  aK 0 0 M M Mrk)rrKrr)rSrrVrrs ``` rWrzBatchSpline.integrates; M M M M M Mt} M M Mz!T.///rkc`tj|}fd|jD|_|S)Nc:g|]}|Sr$)rrrXrs rWrz*BatchSpline.derivative..s'III&))"--IIIrkrdeepcopyrrSrrs ` rWrzBatchSpline.derivatives4mD!!IIIICLIII  rkc`tj|}fd|jD|_|S)Nc:g|]}|Sr$)rrs rWrz.BatchSpline.antiderivative..s'MMMf--b11MMMrkrrs ` rWrzBatchSpline.antiderivatives4mD!!MMMM MMM  rkr)r!r"r#r rrrrr$rkrWrrsg @@@ 0000 rkrc eZdZejdeifeifeddifeddifeddifgejdgdejd gd d Z d S) TestBatchzmake_spline, kwargsr^r0rrr eval_shape)r$rrrrctjd}d}ddd|f}d}tjg||R} tj||d|} |t kr5d} | df| d zz| df| d zzz} |t | | z}|| | fd |i|} t| | f||d |}|j|d |i} tj | | || | d | d }} tj | | || | d| d}} tj | | || tj | j ||j |dS)NlUU{rer9r;r<)rrer7rr.)r2r4r)rrXr)rKrRrrfrrrJrrtestingassert_allcloserrr)rS make_splinerrrrrUrdomainrwrr4r2rrefs rW test_batchzTestBatch.test_batchsi##$455 Aq!  K # # # # # K 5))2t 4 4 / ) )A11q5!QrUHA$66AdQ!nnn,Fk!Q44T4V44!QHT+HHHH CK 1j 1 1 ""33q6633q66222%%a((#*<*>!$$cnnQ&7&7S ""33q6633q66222 ""=3= =3=&;QRRRRRrkN) r!r"r#rarrr rrrr$rkrWrrs [2 b !  $ % . Hi0 1 Hd+ ,     [\+;+;+;<< [VZZZ00SS10=<  SSSrkrrrr)zrIr|rr rrurrrnumpyrKscipy._lib._array_apirrrrrrscipy._lib.array_api_extra_libarray_api_extrarrar rGrCr r r r rrrrrrrrrrrrrrrrscipy.sparse.linalgsparserKscipy.interpolate._bsplinesrrrrrr.r r!r"r#r$scipy.interpolate._fitpack_implr7 _fitpack_implrscipy._lib._utilr%scipy._lib._testutilsr&scipy.interpolate._ndbspliner'r(r{r)rvr*rrr,rrrrrrrurr_rrr?rrrrrrr!rr%r'rOrQrsr%rPryrr>rtrrrrrrr)rJr^ryrrr$rkrWrs     )((((((((******  !!!!!!!!!EEEEEEEEEEEEEEBAAAAAAAGGGGGGGGGG/////////&&&&&&999999544444333333------&&&&&&;/7_ )_ )_ )_ )_ )_ )_ )_ )D7S$S$S$S$S$S$S$S$l666,$ G G G>>>   $b++++$_)_)_)_)_)_)_)_)D%&&`?`?`?`?`?`?`?'&`?F   88!+11(Y))))X      (nLnLnLnLnLnLnLnLb*** ())J3J3J3J3J3J3J3*)J3^      777 :":":":":":":":"z9j4j4j4j4j4j4j4j4ZWGWGWGWGWGWGWGWGta a a a a a a a L(.>""lllllll#"l^2,,,,,,,,^Q7Q7Q7Q7Q7Q7Q7Q7h;R/R/R/R/R/(R/R/ R/j;\0\0\0\0\00\0\0 \0~<  h1h1h1h1h1h1h1! h1V<  b1b1b1b1b1b1b1! b1J8$S$S$S$S$S$S$S$S$S$Srk