&]\c@`sdZddlmZmZmZddlmZmZmZm Z m Z ddl m Z ddlZddlmZmZmZddlmZddlmZdd lmZdd lmZmZmZdd lmZmZdd l m!Z!d e"fdYZ#dZ$dZ%dZ&dZ'dZ(de"fdYZ)de"fdYZ*de"fdYZ+de"fdYZ,de"fdYZ-de"fdYZ.dS( s7 Unit tests for optimization routines from minpack.py. i(tdivisiontprint_functiontabsolute_import(tassert_tassert_almost_equaltassert_array_equaltassert_array_almost_equaltassert_allclose(traisesN(tarraytfloat64tmatrix(t ThreadPool(toptimize(tlambertw(tleastsqt curve_fitt fixed_point(t _assert_warnstsuppress_warnings(tOptimizeWarningt ReturnShapecB`s eZdZdZdZRS(sThis class exists to create a callable that does not have a '__name__' attribute. __init__ takes the argument 'shape', which should be a tuple of ints. When an instance it called with a single argument 'x', it returns numpy.ones(shape). cC`s ||_dS(N(tshape(tselfR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt__init__scC`stj|jS(N(tnptonesR(Rtx((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt__call__s(t__name__t __module__t__doc__RR(((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRs cC`s tj|S(sUA function that returns an array of ones of the given shape. `x` is ignored. (RR(RR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt dummy_func!scC`s;tt|}z|jd|SWd|jXdS(NcS`s|S(N((tf((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt+s(R tlentmapt terminate(tfstpool((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pytsequence_parallel(scC`s=||d}tj|d|d|j|f}|S(sEvaluate non-linear equation system representing the pressures and flows in a system of n parallel pipes:: f_i = P_i - P_0, for i = 1..n f_0 = sum(Q_i) - Qtot Where Q_i is the flow rate in pipe i and P_i the pressure in that pipe. Pressure is modeled as a P=kQ**2 where k is a valve coefficient and Q is the flow rate. Parameters ---------- flow_rates : float A 1D array of n flow rates [kg/s]. k : float A 1D array of n valve coefficients [1/kg m]. Qtot : float A scalar, the total input flow rate [kg/s]. Returns ------- F : float A 1D array, F[i] == f_i. iii(Rthstacktsum(t flow_ratestQtottktPtF((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pytpressure_network4s+cC`st|}tj|dd|dd|d|d}tj||f}|d|d|dd|df((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt#test_pressure_network_with_gradient{s cC`s2td}ddg}tttj||dS(Nig?g@(Rt assert_raisest TypeErrorR R;(Rtfunctx0((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_wrong_shape_func_callables  cC`s,ddg}tttjt|dddS(Ng?g@R9i(i((i(RERFR R;R (RRH((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_wrong_shape_func_functions cC`sAtd}td}tttj|dddgd|dS(NiiRHiRC(ii(RRERFR R;(RRGt deriv_func((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt test_wrong_shape_fprime_callables  cC`s;d}d}tttj|dddgd|dS(NcS`s t|dS(Ni(i(R (R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"scS`s t|dS(Ni(ii(R (R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"sRHiiRC(RERFR R;(RRGRK((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt test_wrong_shape_fprime_functions  cC`s<d}ttddtj|ddgWdQXdS(NcW`stddS(NsI raised(t ValueError(R9((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRGstmatchsI raisedRHi(RERNR R;(RRG((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_func_can_raises c C`sKd}d}ttdd!tj|ddgd|WdQXdS(NcS`s|tjdgS(Ni (RR (R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"scW`stddS(NsI raised(RN(R9((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRKsROsI raisedRHiRC(RERNR R;(RRGRK((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_Dfun_can_raises  cC`sSd}tj|tjddgtj}t||ddgdddS(NcS`s0tj|dd|ddgdtjdS(Niidiitdtypei(RR tfloat32(R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"siitatolgMbP?(R R;RR RSR(RRGtp((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt test_float32s 'c `sfd}tjdd}d}tddddg}tj||d||fdt\}}}}t|tjdt|dk|dS( Nc`sjt|S(N(RBR0(R9(R(s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRGs ig?g@gR9R:i(RRR R R;R<RR( RRGR-R,R=R>R?R@RA((Rs@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_reentrant_funcsc`s~fd}tjdd}d}tddddg}tjt|d||fd|}t|tjddS(Nc`sjt|S(N(RDR6(R9(R(s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRKs ig?g@gR9RC(RRR R R;R0R(RRKR-R,R=R>((Rs@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_reentrant_Dfuncs cC`st|jgdS(Ni (R(RB(R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_concurrent_no_gradientscC`st|jgdS(Ni (R(RD(R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_concurrent_with_gradients(RRRBRDRIRJRLRMRPRQRVRWRXRYRZ(((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR8os         t TestRootHybrcB`s#eZdZdZdZRS(cC`srtjdd}d}tddddg}tjt|ddd||fj}t|tjddS(Nig?g@gtmethodthybrR9(RRR R trootR0RR(RR-R,R=R>((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRBs c C`sxtjdd}d}tddddg}tjt|d||fdddtj}t|tjddS( Nig?g@gR9R\R]R5( RRR R R^R0R6RR(RR-R,R=R>((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRDs c C`sxtjdd}d}tddddg}tjt|d||fdddtj}t|tjddS( Nig?g@gR9R\R]R5( RRR R R^R7R<RR(RR-R,R=R>((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt,test_pressure_network_with_gradient_combineds (RRRBRDR_(((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR[s t TestRootLMcB`seZdZRS(cC`srtjdd}d}tddddg}tjt|ddd||fj}t|tjddS(Nig?g@gR\tlmR9(RRR R R^R0RR(RR-R,R=R>((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRBs (RRRB(((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR`st TestLeastSqcB`seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZRS(cC`stjddd}d \}}}||_|||f|_||d|||}tjjd|dtjj|j|_dS( Nii i(g@i*g33333sig{Gz?(g@i*g33333s( RtlinspaceRtabctrandomtseedtstandard_normalRty_meas(RRtatbtcty_true((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt setup_methods cC`s1|\}}}|||d|||}|S(Ni((RRUtyRRiRjRkterr((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt residuals scC`s'tj|d|tj|gj S(Ni(Rtvstackt ones_liketT(Rt_pt_yR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pytresiduals_jacobianscC`sptdddg}t|j|d|j|jf\}}t|d kd|t||jdddS( NiR9iiiissolution not found (ier=%d)tdecimal(iiii(R RRpRhRRRRd(Rtp0t params_fitR@((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt test_basics cC`sytdddg}t|j|d|j|jfd|j\}}t|d kd|t||jd ddS( NiR9tDfuniiiissolution not found (ier=%d)Rw(iiii( R RRpRhRRvRRRd(RRxRyR@((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_basic_with_gradients cC`sotdddg}t|j|d|j|jfdt}|\}}}}}t|d kd|dS( NiR9R:iiiissolution not found: %s(iiii(R RRpRhRR<R(RRxR:Rytcov_xtinfodictRAR@((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_full_output#s  c C`stdddgdt}t|dt}t|j|d|j|jfdt}|\}}}}}t|d kd |t||dS( NiRRtcopyR9R:iiiissolution not found: %s(iiii( R R R<RRpRhRRR( RRxtp0_copyR:RyR}R~RAR@((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_input_untouched+s cC`s2td}ddg}tttj||dS(Nig?g@(RRERFR R(RRGRH((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRI5s  cC`s,ddg}tttjt|dddS(Ng?g@R9i(i((i(RERFR RR (RRH((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRJ<s cC`sAtd}td}tttj|dddgd|dS(NiiRHiR{(ii(RRERFR R(RRGRK((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_wrong_shape_Dfun_callableBs  cC`s;d}d}tttj|dddgd|dS(NcS`s t|dS(Ni(i(R (R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"HscS`s t|dS(Ni(ii(R (R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"IsRHiiR{(RERFR R(RRGRK((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_wrong_shape_Dfun_functionGs  c C`sd}tjdddddddd d d g d tj}tjd dddddddddg d tj}tjddddg}tj||d||f\}}t|dkt||||djd||||djkdS(NcS`sD|dtj||dd d|dd|d}||S(Niiig@i(Rtexp(RURRntq((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRGNs<g?gw/?g%C?gNbX9?gHzG?gM?gZd;O?gS?g~jt?gV-?RRgJ4?gN@aÓ?g^K=?gM O?g ?g?W[?gl?g{Pk?g rh?gׁsF?g?R9iiiig-C6?(iiii(RR RSR RRR*(RRGRRnRxtp1tsuccess((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRVLs !$cC`s<d}ttddtj|ddgWdQXdS(NcW`stddS(NsI raised(RN(R9((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRG]sROsI raisedRHi(RERNR R(RRG((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRP\s c C`sKd}d}ttdd!tj|ddgd|WdQXdS(NcS`s|tjdgS(Ni (RR (R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"dscW`stddS(NsI raised(RN(R9((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRKfsROsI raisedRHiR{(RERNR R(RRGRK((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRQcs  c`s|fd}tdddg}t||djjf\}}t|d kd|t|jd ddS( Nc`sjj|S(N(RzRp(R9(R(s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRGms iR9iiiissolution not found (ier=%d)Rw(iiii(R RRhRRRRd(RRGRxRyR@((Rs@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRWls  c`sfd}tdddg}tj|djjfd|\}}t|d kd |t|jd ddS( Nc`sjj|S(N(RzRv(R9(R(s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRKys iR9R{iiiissolution not found (ier=%d)Rw(iiii(R RRpRhRRRRd(RRKRxRyR@((Rs@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_reentrant_DfunxscC`st|jgdS(Ni (R(Rz(R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRYscC`st|jgdS(Ni (R(R|(R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRZs(RRRmRpRvRzR|RRRIRJRRRVRPRQRWRRYRZ(((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRbs$           t TestCurveFitcB`seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZRS(cC`s:tddddg|_tddddg|_dS(Ng?g @g#@gffffff+@g@g@g@(R RnR(R((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRmsc C`sd}t||j|j\}}tt|dkt|jd kt|ddddt|d dddt||j|jdd}|\}}}}} t||dS( NcS`s||S(N((RRi((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRGsiig3ı.n?Rwig-C6Z?R:(ii(ii(RRRnRR#RRR( RRGtpopttpcovtrestpopt2tpcov2R~terrmsgR@((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_one_arguments cC`sd}t||j|j\}}tt|dkt|jd kt|ddgddt|ddgdd ggdddS( NcS`s |||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRGsigc]K?g -?Rwig䃞ͪϵ?gT㥛 gb=y?(ii(RRRnRR#RR(RRGRR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_two_arguments cC`sdtfdY}|}t|j|j|j\}}t|jd kt|ddgddt|dd gd d ggdddS( Nt test_selfcB`seZdZdZRS(sThis class tests if curve_fit passes the correct number of arguments when the model function is a class instance method. cS`s |||S(N((RRRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRGs(RRRRG(((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyRsigc]K?g -?Rwig䃞ͪϵ?gT㥛 gb=y?(ii(tobjectRRGRRnRRR(RRttest_self_instRR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_func_is_classmethods !c C`sddddddddd d g }d d d dddddddg }dddddd g}ddddddg}d}t||||d d!\}}t||d"d#dS($NgB`"@gZd;@gQ@g#~j@g/$@g;O@g @g@gZd;@g)\@g؊@g(@g@gX@gt@g@g@g8@g@gH@g^w8}@gU]}@gX@g29l?g䃞j@gj@gT{7@g|'fɘ@gƞ?g'ˊ@cS`sP|tj||d d|d|tj||d d|d|S(Nig@(RR(RRHtx1tA0tA1tsigmaRk((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pytf_double_gaussstmaxfevi'trtolgh㈵>(RR(RRRntguesstgoodRRR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_regression_2639s         !c C`stjddddddg}tjdddddd g}tjddddddg}d }xd d d gD]}t|||dddgd|d|\}}tjtj|}t|ddgddt|||dddgdd|d|\}}tjtj|}t|ddgddt|||dddgd|dtd|\}}tjtj|} t| ddgddt|||dddgdd|dtd|\}}tjtj|} t| ddgddq|Wd} tjtjgdjdd} t m} | j t dt| ||dddgd|\}}t||d |d dddg\} }WdQXt |j dkt|| t |j dkt|| dS(Niiiiiiiii cS`s |||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR!sRattrftdogboxRxRR\gN"q?g~uM/N?RgMbP?tabsolute_sigmagO=N?g5=`6?cS`s||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pytf_flatss3Covariance of the parameters could not be estimatedg[u|?g-/i@(ii(ii(RR RtsqrtR1RR<tinftreshapeRtfilterRRRR(RtxdatatydataRR!R\RRt perr_scaledtperrRt pcov_expectedtsuptpopt1tpcov1((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt test_pcovs@!!! !%!% %  *2 cC`sZd}ddddg}ddddg}tt|||d ddgd d dS( NcS`s |||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pytf_linearsiiiiiii iRTg|=(RR(RRRRn((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_array_likes cC`s_tjddddddg}tjddddddg}tttd||dS( Niiiiiig@cS`s||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR" s(RR RRR(RRR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_indeterminate_covariances!! c C`stjdtjdg}tjdddg}tttd||tttd||tttd||itd6dS(NiiicS`s |||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"scS`s |||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"scS`s |||S(N((RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR"st check_finite(RR tnanRERNRR<(RRR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_NaN_handling s  cC`sd}tjddd}||dd}xKddddgD]7}t|||d |\}}t|ddgqCWttt|||d d dS( NcS`s|tj| |S(N(RR(RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR!siii g@RRRaR\tunknown(RRctNoneRRRERN(RR!RRR\RR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_method_arguments c C`sd}tjddd}||dd}ddgddgf}xLdd d gD];}t|||d |d |\}}t|ddqXWt|||d d d d dgdtjgf\}}t|ddttt|||d |d ddS(NcS`s|tj| |S(N(RR(RRiRj((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR!(siii g@g?g?g@RRtboundsR\gg333333?Ra(RRcRRRRRERN(RR!RRRR\RR((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyt test_bounds's $c C`sd}tjdtjdtjd}tj|}dtjdtjf}xwddgD]i}t|||d d tj\}}t|||d d tjd |d |\}}t||qbWdS( NcS`stj||S(N(Rtsin(RRi((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyR!Bsiii(iiRRRxg@RR\(RRctpiRRR( RR!RRRR\tpopt_1t_tpopt_2((s@lib/python2.7/site-packages/scipy/optimize/tests/test_minpack.pyttest_bounds_p0?s #%c C`s}d}d}tjddd}||dd}xeddgD]W}xNd d d gD]=}t|||d |d |\}}t|ddgq\WqFWxNdddgD]=}t|||d |d |\}}t|ddgqWd|ds0(     f  <