ó +Šm\c@`sùddlmZmZmZddlZddlmZmZm Z ddl m Z ddl m Zddl Z ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZddljZ ddl!m"Z"ddl#m$Z$m%Z%ddl&j'j(Z)dd l*m+Z+d e,fd „ƒYZ-d „Z.d „Z/d„Z0d„Z1d„Z2d„Z3dd„Z4ddd„Z5d„Z6de,fd„ƒYZ7de,fd„ƒYZ8d„Z9d„Z:dd„Z;de,fd„ƒYZ<dS( i(tdivisiontabsolute_importtprint_functionN(t assert_equaltassert_allclosetassert_(tsuppress_warnings(traises(tBSplinetBPolytPPolytmake_interp_splinetmake_lsq_splinet_bspltsplevtsplreptsplpreptspldert splantidertsproottsplinttinsert(t NumpyVersion(t _not_a_knott_augknt(t_splintt TestBSplinecB`s eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS(cC`s·tttfttdddgddgddƒtjdd ƒ6ttttddtjgddgddƒWdQXttttddtjgddgddƒttttddd gddgddƒttttddgdggddgddƒttttdddd gddgddƒttttdddd d d gdddgdd ƒttttdddddddgddddgddƒttttdddddddgddddgddƒttttdddddd d gddddgdd ƒd\}}tj ||dƒ}tj j |ƒ}t|||ƒ}t ||j ƒt ||j ƒt||jƒdS(Nttiyð?tcgð?tkitinvalidtignoreiÿÿÿÿiiigg@g@g@tcubicg@i (i i(t assert_raisest TypeErrort ValueErrorRtdicttnpterrstatetnantinftarangetrandomRRRRR(tselftnRRRtb((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_ctors.%71.41 1 7 7 7 cC`s‰tƒ}|j}t|j|dddddƒt|j|dddddƒt|j|dƒtjt ƒd|_WdQXdS(NitatolgV瞯Ò(R2R3RAtFalseR(R+R-Rt_RRTRUtextrap((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_endpoints¨s  cC`sftƒ}|j\}}}t|||d| d!dƒ|||d| d!dƒddƒdS(Nig»½×Ùß|Û=R/g•Ö&è .>(R2R3R(R+R-RR]R((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_continuity±s @cC`sÜtƒ}|j\}}}|d|d}tj||||| d|dƒ}|||k||| dk@}t|||dtƒ|||dtƒƒt||dtƒt||||fddƒƒdS(Niÿÿÿÿiii2R>text(R2R3R%R7RRAR\R(R+R-RRRRZR9tmask((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_extrap¸s *#cC`sdtƒ}|j\}}}|dd|ddg}||ƒ}ttjtj|ƒƒ ƒdS(Niiiÿÿÿÿ(R2R3RR%talltisnan(R+R-RR]RR9tyy((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_default_extrapÇs   c C`sOtjjdƒtjtjjdƒƒ}tjjdƒ}d}t|||ddƒ}|j|d}|d|d }tj||||||d ƒ}|||||||||}t||ƒt||||fƒƒdd d dg}|||||||||}t ||ddƒ||dt ƒƒdS( NiÒiiiR>tperiodiciiÿÿÿÿii2gà?( R%R*RLRMRRRR7RRRRA( R+RRRR-R,RZR9txy((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_periodic_extrapÏs%&%&cC`sztƒ}|j\}}}tj|||fƒ}tj|||| dƒ}t||ƒ||ƒddddƒdS(NidR/g›+¡†›„=R0(R2R3R t from_splineR%R7R(R+R-RRRtppR9((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_ppolyás  cC`sÍtƒ}|j\}}}tj|d|ddƒ}tj||f}xXtd|dƒD]C}t||||fd|ƒ}t|||d|ƒddƒq_Wt||d|dƒdddƒdS( Niiÿÿÿÿi2itdertnuR/g›+¡†›„=(R2R3R%R7RWtrangeRR(R+R-RRRR9Rntyd((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_derivative_rndmés #c C`sÈd}ddddddddddddg }tjjd ƒtjddtjjd ƒddf}t|||ƒ}tjddddgƒ}t|||dkd ƒ|||dkd ƒƒttj|dƒ|dƒƒ ƒtjddgƒ}t||d d dƒ||d d dƒƒtjddgƒ}ttj tj||d d dƒ||d d dƒƒƒ ƒttj tj||d d dƒ||d d dƒƒƒ ƒdS(NiiÿÿÿÿiiiiiiiÒig»½×Ùß|Û=g@Rog2Hþÿÿÿ@gη@( R%R*RLRWRR?RRtallcloseRd(R+RRRR-R<tx0tx1((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_derivative_jumpsös"*(#""cC`s÷tjdddƒ}tjdddddgƒ}t||ƒt||j|j|jfƒd d ƒt||ƒt |ƒd d ƒtjdddddgƒ}tjddd ƒ}t||ƒtj |dk||d |dƒd d ƒdS( NiÿÿÿÿiiRiiiiR/g›+¡†›„=i g@( R%R7Rt basis_elementRRRRRtB_0123R8(R+R9R-((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_basis_element_quadratics (  cC`sitƒ}|j\}}}tj|||| ddƒ}t||ƒt||||ƒddƒdS(NiiR/g›+¡†›„=(R2R3R%R7Rt_sum_basis_elements(R+R-RRRR9((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_basis_element_rndms "c C`sÑtƒ}|j\}}}|d}t|||ƒ}t||jj|ƒ}t||jj|ƒ}tj|||| ddƒ}t||ƒj||ƒddƒt||ƒj||ƒddƒdS(Ngð?y@iiR/g›+¡†›„=yð?@( R2R3RRtrealtimagR%R7R( R+R-RRRtcctb_retb_imR9((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_cmplx%s  ""cC`s;tjddddgƒ}ttj|tjƒƒƒdS(Niii(RRwRR%ReR'(R+R-((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_nan2scC`s§tddƒ}|j\}}}t|||ƒ}tj|||| ddƒ}xKtd|ƒD]:}|jƒ}t|||ƒ||ƒddddƒqeWdS(NRiiiR/gê-™—q=R0(R2R3RR%R7Rpt derivativeR(R+R-RRRtb0R9tj((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_derivative_method7s" cC`sàtƒ}|j\}}}tj|||| ddƒ}t|jƒjƒ|ƒ||ƒddddƒtj|||f}tj||fƒ}t |||ƒ}t|jƒjƒ|ƒ||ƒddddƒdS(NiiR/g›+¡†›„=R0( R2R3R%R7RtantiderivativeRƒtc_tdstackR(R+R-RRRR9((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_antiderivative_method@s "cC`s&tjdddgƒ}t|jddƒdƒt|jddƒdƒt|jddƒdƒt|jddƒdƒt|jdddtƒdƒt|jdddtƒdƒt|jdddtƒdƒ|j\}}}t|jdddtƒt|||ddƒdƒd|_|j ƒ}|dƒ|dƒ}t|jddƒ|ƒt|jddƒd|ƒt|jd d ƒ|ƒt|jd d ƒd|ƒt|jdd ƒ|d ƒ|dƒƒt|jd dƒ|dƒ|dƒ|dƒ|d ƒƒt|jddƒ|dƒ|dƒ|dƒ|d ƒƒt|jd dƒ|dƒ|dƒ|dƒ|d ƒd|ƒt|jddƒ|dƒ|dƒƒt|jd dƒ|dƒ|dƒƒt|jdd ƒ|dƒ|dƒd|ƒdS(Niiigà?iÿÿÿÿgà¿R>Rhi÷ÿÿÿiùÿÿÿiøÿÿÿiüÿÿÿgø?ii iiöÿÿÿigà¿gà¿g+@ii( RRwRt integrateRAR\R3RR>R‡(R+R-RRRtit period_int((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_integralNs6  )=+3))cC`s…dddddg}t||ƒ}d|_tj|ƒ}xBd d dgD]1\}}t|j||ƒ|j||ƒƒqLWdS(NiiiiiRhiûÿÿÿgà?iiüÿÿÿi (iûÿÿÿgà?(gà?i(iüÿÿÿi (R R>R RkRR‹(R+R<R-tpRtRu((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_integrate_ppolyts cC`sqdtfd„ƒY}|jddddgƒ}t|j|ƒt|jƒj|ƒt|jƒj|ƒdS(NtBcB`seZRS((t__name__t __module__(((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR‘siii(RRwRt __class__RƒR‡(R+R‘R-((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_subclassings c C`såd\}}tjdd||dƒ}dddg}x¦tdƒD]˜}|}|j||ƒtjjd |ƒ}t|||d |ƒ}t|jj||g|| ||dƒtjjdƒ} t|| ƒj|| t | jƒ||dƒxBd |j gD]1} t t tt d |d|d|d | ƒqWx“t|||d |ƒjƒt|||d |ƒjdƒt|||d |ƒjƒt|||d |ƒjdƒgD]} t| j|jƒqÀWqEWdS(NiiiiiiiiRRtaxisiiÿÿÿÿRRRi(ii(iii(R%R7RpRR*RRRRStlisttndimR!R#R$RƒR‡R–( R+R,RRtsh0R–tshRR-txptaxtb1((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_axis‰s(  "$/((R’R“R.R6R:R=RDRJRKRORQRVR[R_R`RcRgRjRmRrRvRyR{RR‚R†RŠRŽRR•Rž(((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyRs: #               & c C`s©dddd„}xdddddgD]y}td |ƒ}xatt|ƒƒD]M\}}|||ƒx1td|dƒD]}||||d d ƒq}WqPWq(WdS( Nig›+¡†›„=c S`s¥|j\}}}tj|ƒ}tj|ddd|d|d |ddf}tt||||f|ƒ|||ƒd|d|dd ||jfƒdS( Nigš™™™™™¹?gà?iiÿÿÿÿR/R0terr_msgsder = %s k = %s(R3R%tuniqueRWRRR( R-R…RnR/R0RRRR<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt check_splev©s 6*iiiiiRgê-™—q=(R2t enumeratet_make_multiplesRp(R¡RR-R…RRn((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_knots_multiplicity¥s  cC`s|dkr<|||ko/||dknr8dSdS|||||kr]d}n9||||||||t||d||ƒ}|||d||dkr¿d}nM|||d||||d||dt||d|d|ƒ}||S(sw Naive way to compute B-spline basis functions. Useful only for testing! computes B(x; t[i],..., t[i+k+1]) iigð?g(t_naive_B(R<RRŒRtc1tc2((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¥¼s 0 9  Mc`s¸ˆˆˆkrˆ‰ntjˆˆƒd‰ˆˆˆkoRˆˆdkns]t‚ˆˆkrˆtˆƒˆks…t‚t‡‡‡‡‡fd†tdˆdƒDƒƒS(s= Naive B-spline evaluation. Useful only for testing! ic3`s4|]*}ˆˆ|tˆˆˆ|ˆƒVqdS(N(R¥(t.0R…(RRŒRRR<(sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pys Ùsi(R%t searchsortedtAssertionErrortlentsumRp(R<RRR((RRŒRRR<sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyREÏs  .(c`s•tˆƒˆd}|ˆdks*t‚tˆƒ|ksBt‚ˆˆˆkoaˆ|knslt‚t‡‡‡‡fd†t|ƒDƒƒS(s'Naive B-spline evaluation, another way.ic3`s,|]"}ˆ|tˆˆ|ˆƒVqdS(N(R¥(R¨RŒ(RRRR<(sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pys âs(R«RªR¬Rp(R<RRRR,((RRRR<sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyRFÜs *cC`s¨t|ƒ|d}||dks*t‚t|ƒ|ksBt‚d}xYt|ƒD]K}tj||||d!dtƒ|ƒ}|||tj|ƒ7}qUW|S(NigiR>(R«RªRpRRwR\R%t nan_to_num(R<RRRR,tsRŒR-((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyRzås*cC`smtj|ƒ}tj||dk|dkB|dk|dk@|dk|dk@gd„d„d„gƒS(s+ A linear B-spline function B(x | 0, 1, 2).iiicS`sdS(Ng((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pytöscS`s|S(N((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯öscS`sd|S(Ng@((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯ös(R%t atleast_1dt piecewise(R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR;ðs cC`s±tj|ƒ}|dk|dk|dk@|dkg}|dkrad„d„d„g}n7|dkrˆd„d„d „g}ntd |ƒ‚tj|||ƒ}|S( s0A quadratic B-spline function B(x | 0, 1, 2, 3).iiicS`s ||dS(Ng@((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯þscS`sd|ddS(Ng@iigè?gø?((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯ÿscS`sd|ddS(Ng@i((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯scS`sdS(Ngð?((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯scS`sdS(NgÀ((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯scS`sdS(Ngð?((R<((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR¯ssnever be here: der=%s(R%R°R#R±(R<Rntcondstfuncstpieces((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyRxùs+  i#icC`sXtjjdƒtjtjj||dƒƒ}tjj|ƒ}tj|||ƒS(Ni{i(R%R*RLRMRtconstruct_fast(R,RRR((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyR2 s#cc`s»|j|j}}|jjƒ}|d|dd+|d|d 1 is not recommended.ii(i(R»R-R¼RRR3RtfiltertDeprecationWarningttupleRpRR˜t transposeRR( R+R»R-R¼R<tsupRšR~R3((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_splev5s "  .c C`s'|j|j}}t||ƒ}tj||ƒ\}}}t|d|ddƒt|d|ddƒt|d|ƒt||dtƒ\}}}}t|d|ddƒt|d|ddƒt|d|ƒt||ƒ} t|| ddƒt|Œ} t|| |ƒddƒdS(NiR/gV瞯Ò= 1.14.0c C`sÌ|j|j}}tj||f}ttƒt||ƒWdQXttƒtj||ƒWdQXttddƒt|d |d ƒWdQXttddƒtj|d |d ƒWdQXdS(Ntmatchsm > k must holdi( R9RfR%RˆR!R#RRÅR"(R+R<RNty2((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_splrep_errorsjs  c C`sçtjdƒjd ƒ}t|ƒ\}}tj|ƒ\}}t||ddƒtt||ƒ|ddƒtt||ƒ|ddƒt|dddtƒ\\}}}}}t||ddƒtt||ƒ|ddƒdS( NiiiR/gV瞯Ò k must holdg– Ð>IÀg– Ð>KÀsInvalid inputsiig333333Ó?gš™™™™™É?i i<(iii( R%R)R@R!R#RRÅR7R"tNone(R+R<RÌ((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_splprep_errorsŠs&cC`s\|j|j}}tjddddgƒtj}tt|ƒ|ddddƒtt|j|j|j fƒ|ddddƒt ƒ)}|j t dƒt|d d ƒ}WdQXtj |ƒ}t|jdƒt||dddƒ|jjdd dƒ}tj t|j||j fd d ƒƒ}t|jdƒt||dddƒdS(Ngà?gø?g@g @R/gH¯¼šò×z>R0sICalling sproot.. with BSpline objects with c.ndim > 1 is not recommended.tmesti2iiiigê-™—q=i(iii(iii(R-R¼R%tarrayR¹RRRRRRR¾R¿R?RRSRÁ(R+R-R¼trootsRÂtrtc2rtrr((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_sproot§s"1  *c C`s.|j|j}}ttdd|ƒtdd|jƒddƒttdd|ƒ|jddƒddƒtƒB}|jtdƒttdd|ƒ|jddƒddƒWdQX|j j dddƒ}t j tdd|j ||jfƒƒ}t|jdƒt|tdd|ƒddƒdS( NiiR/g›+¡†›„=sICalling splint.. with BSpline objects with c.ndim > 1 is not recommended.ii(ii(R-R¼RRR3R‹RR¾R¿RRÁR%R?RRRRS(R+R-R¼RÂRØtintegr((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_splint¾s  1*cC`s4x-|j|jgD]}t|jƒt|jƒ}|dkrwtj|jtj|f|jjdƒf|_nx²dddgD]¡}t |ƒ}t j |j|j|j fƒ}t |j|dddƒt |j|dddƒt |j |dƒtt|tƒƒtt|tƒƒq‡WqWdS(NiiiiR/gV瞯Ò<(R-R¼R«RRR%RWtzerosRSRRÅRRRRt isinstanceRRÀ(R+R-tctR,tbdttck_d((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyt test_splderÓs 6 !cC`s4x-|j|jgD]}t|jƒt|jƒ}|dkrwtj|jtj|f|jjdƒf|_nx²dddgD]¡}t |ƒ}t j |j|j|j fƒ}t |j|dddƒt |j|dddƒt |j |dƒtt|tƒƒtt|tƒƒq‡WqWdS(NiiiiR/gV瞯Ò<(R-R¼R«RRR%RWRÝRSRRÅRRRRRÞRRÀ(R+R-RßR,RàRá((sDlib/python2.7/site-packages/scipy/interpolate/tests/test_bsplines.pyttest_splantiderãs 6 !c C`sŒ|j|j|j}}}|jjd}d|j||j|d}t||ƒt||j|j|jfƒ}}tt ||ƒt ||ƒddƒt t |t ƒƒt t |t ƒƒt t|jjƒƒ}|jj|ddƒ} t||j| |jfƒ} t||ƒ} ttjt || ƒƒjdddƒ| |ƒddƒt t | t ƒƒt t | t ƒƒdS(Nigà?iR/gV瞯Òtjtjtj gD]#}||ds8$   &&  &cC`s.ttj|||fƒ\}}}|j}|j|d}tj||fdtjƒ}xƒt|ƒD]u}||}|||kr’|} ntj||ƒd} tj |||| ƒ} | ||| || d…fs:  ^ÿ’    êÿ  4