&]\c@`sddlmZmZmZddlZddlmZmZmZm Z ddl m Z m Z m Z mZddlZddlmZmZmZmZddlmZmZmZddlmZddlmZmZm Z!dd l"m#Z#d eje$j%Z&ee$j%Z'd Z(d Z)d Z*dZ+dZ,dZ-dZ.de/fdYZ0dZ1dZ2dfdYZ3dZ4dZ5dZ6dZ7dZ8dZ9dS(i(tdivisiontprint_functiontabsolute_importN(tsqrttexptsintcos(t assert_warnstassert_tassert_allcloset assert_equal(tfinfotpowertnantisclose(tzerostnewtont root_scalar(tgetargspec_no_self(t get_testst functionstfstrings(tsuppress_warningsicC`s|dd|dS(Nii((tx((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf1scC`s d|dS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf1_1scC`s dd|S(Ng@i((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf1_2#scC`st|t|t|fS(N(RRR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf1_and_p_and_pp'scC`st|t|S(N(RR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf2,scC`st|t|S(N(RR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf2_10scC`st|t|S(N(RR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf2_24st TestBasiccB`sheZddZdZd d dZdeded dZd d dededZdZ dZ d Z d Z d Z d Zd ZedZedZedZedZedZedZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"RS(!ic K`sd}td}dtjtj}dtjtj}xtttD]\}} |dkrx| dkrxqNnt|d|d ||gd |d |d ||} | j } t | j t | d d|d |dd|| fqNWdS(Ng?iiitf4tf5tf6tmethodtbrackettx0txtoltrtolg?tatolterr_msgsmethod %s, function %s(R R!R"( RtnpR tfloattepstzipttstutils_functionsttstutils_fstringsRtrootRt convergedR ( tselftnamet smoothnesstkwargstatbR&R'tfunctiontfnametrtzero((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytrun_check_by_name:s !  c C`sd}td}dt}dt}x}tttD]l\}}||||d|d|dt\} } t| jt| dd|d|d d ||fq6WdS( Ng?iiR&R't full_outputg?R(R)smethod %s, function %s( Rt _FLOAT_EPSR-R.R/tTrueRR1R ( R2R#R3R6R7R&R'R8R9R;R:((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt run_checkIs    c K`s4g}x[|pgD]M}||krOidd6dd6dd6j||}n|j||qWt|}|jitd6td6x"|pgD]}||||lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt _run_one_testUs  *  ic K`s't|}t|d} t|d| } |d| } g} |dkr|dkr| jd|dkr| jdqn||dRIRFR0R-R(R2ttestsR#R3R&R't known_failR5tsigt nDefaultst nRequiredRORPRNtresultstelttnotcvgdt notcvged_IDSttolsR(tcvgdtapproxtcorrectR6tctnotclosetaroottfullouttfvstfv((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt run_testslsB        7 &-!  & "'6;c K`s>t|d|} |j| ||d|d|d||dS(suRun a collection of tests using the specified method. The name is used to determine some optional arguments.R4R&R'RaN(RRs( R2t collectionR#R3R4RaR&R'R5R`((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytrun_collectionscC`s@|jtjd|jd|jdtjddddS(NtbisecttapsR4i(R@RRvR<Ru(R2((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_bisects cC`s@|jtjd|jd|jdtjddddS(NtridderRwR4i(R@RRyR<Ru(R2((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_ridders c C`sL|jtjd|jd|jdtjddddddddS(NtbrentqRwR4iR&g+=R'(R@RR{R<Ru(R2((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_brentqs c C`sL|jtjd|jd|jdtjddddddddS(NtbrenthRwR4iR&g+=R'(R@RR}R<Ru(R2((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_brenths cC`s@|jtjd|jd|jdtjddddS(Nttoms748RwR4i(R@RRR<Ru(R2((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_toms748s c C`sVdg}|ddg7}x6ddgD](}|j|tjdddd |q&WdS( Ns aps.13.00s aps.12.05s aps.12.17RwtcomplexRR4iRa(RuRR(R2RaRt((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_newton_collectionss  cC`smddddddddd d d d d dg}x6ddgD](}|j|tjdddd|q=WdS(Ns aps.12.06s aps.12.07s aps.12.08s aps.12.09s aps.12.10s aps.12.11s aps.12.12s aps.12.13s aps.12.14s aps.12.15s aps.12.16s aps.12.17s aps.12.18s aps.13.00RwRRXR4iRa(RuRR(R2RaRt((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_halley_collectionss    cC`s|dd|dS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscC`s d|dS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscC`s dd|S(Ng@i((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscC`st|t|S(N(RR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscC`st|t|S(N(RR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscC`st|t|S(N(RR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRsc C`s$x|j|j|jf|j|j|jfgD]\}}}tj|ddd}t||dddtj|ddddd}t||dddtj|dd|dd}t||dddtj|dd|d |dd}t||dddq1WdS( NiR[gư>iR(RAiRYRZ( RRRRRRRRR (R2RBtf_1tf_2R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_newtons($c C`svxotttftttfgD]O\}}}t|ddddd|dd}t||jdd dqWd S( s#Invoke newton through root_scalar()R#RR%iRYR&gư>iR(N( RRRRRRRR R0(R2RBRRR:((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_newton_by_names.$c C`sxtttftttfgD]\}}}t|dddddddd}t||jd d dt|dddddd dd}t||jd d dqWd S( s#Invoke secant through root_scalar()R#RWR%iRAiR&gư>iR(iN( RRRRRRRR R0(R2RBRRR:((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_secant_by_names .$$c C`s|xutttftttfgD]U\}}}t|ddddd|d|dd}t||jd d dqWd S( s#Invoke halley through root_scalar()R#RXR%iRYRZR&gư>iR(N( RRRRRRRR R0(R2RBRRR:((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_halley_by_name s.cC`stjt!ttddddddWdQXtjt!ttddddddWdQXtjt'ttddd tddddWdQXtjt'ttddd tddddWdQXdS( NR#RWR%iR&gư>RRXRYRZ(tpytesttraisest ValueErrorRRRR(R2((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_root_scalar_fails""(c C`s d}d}d}tjdddddd d d d d g }tjtddd}||ddddf}dgd}tj||||}d!} t|| tj||||d|}t|| tj||d|}t|| d S("stest newton with arraycW`sO|d||d}|d|dtj||dd||d|S(Niiiiig?i(R*R(RR6R7((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscW`sT|d|d}|d tj|d|d||||d|ddS(Niiiiii(R*R(RR6R7((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyR"scW`sD|d|d}|d tj|d|d|||dS(Niiii(R*R(RR6R7((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyR&sg 4O@gNk@g]0J@g]Qݚt@g~EO5@g$J?g~ 5,@gXCڭ@gͮ9@@gӍ@i g?g@g& .>gMbp?gn2d?gԩ˰@g9~4b'@gaq(@g0p@gk"z?g4se?gU+F@gwQu%@gd6)@g)i!@RZREN( gԩ˰@g9~4b'@gaq(@g0p@gk"z?g4se?gU+F@gwQu%@gd6)@g)i!@(R*tarrayRtrangeRRR ( R2RRRta0ta1RER%Rt x_expected((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_array_newtons*          cC`sGtjddddgdtjddgg}t|d d S( s8test secant doesn't continue to iterate zero derivativescW`s|||dS(Ni((RR6((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytDsR%gˡE}@iREiigf~@g@N(gf~@g@(RRR*RR (R2R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt!test_array_secant_active_zero_derBscC`sttjddgddddgf}t|d tjd d gddd d gf}t|ddS(NcS`s ||dS(Ni((tytz((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRJsg@iREg.@g1@gNO@gf~@cS`s ||dS(Ni((RR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRNsiii(gNO@gf~@(gNO@gf~@(RRR (R2R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_array_newton_integersHs  +c C`stttjdddgdtjtgtjdddgddt}t|jd|j j st |j j st WdQXdS(NcS`s |ddS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRTsgcS`sd|S(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRTscS`s |ddS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRWscS`sd|S(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRXsR=i(RtRuntimeWarningRRRtwarnsR?R R0tzero_dertalltAssertionErrorR1tany(R2Re((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt#test_array_newton_zero_der_failuresQs c C`sd}d}d}d}t|ddddd |}t|ddddd t}t|j|jd d td |j|jt|dd ddd |d|}t|dd dddt}t|j|jd d td|j|jdS(NcS`s|dd|dS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyR^scS`s d|dS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyR_scS`s dd|S(Ng@i((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyR`scS`s%|dd|dd|ddfS(Niig@((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRbsR#RR%iRYR(g:0yE>iRXRZ(RR?R R0R tfunction_calls(R2RRRRtsol0tsol((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_newton_combined]s    $c C`sd}dddg}xtdD]}idd6td 6}x:d |jgd |jgg| D]\}}|||R[R=RYRZRDiitmaxitertmatchs3Failed to converge after %d iterations, value is .*(ii(ii (ii (RR?RRRRRRJRR1R R0t iterationsRRRRt RuntimeError( R2R%texpected_countstderivsR5RRtvRR:titers((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_newton_full_outputos./$   * cC`s/d}d}tttj|d|dS(NcS`s |ddS(Nig@((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscS`sd|S(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRsg(RRRR(R2RCtdfunc((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_deriv_zero_warnings  N(#t__name__t __module__R<R@tNoneRVR>RsRuRxRzR|R~RRRt staticmethodRRRRRRRRRRRRRRRRRR(((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyR8sD    3           '   %c `sdfd}tjtjg}t}}xR|D]J}||ddd|d|}t|d|d|dd |jq8WdS( Ng?c`s|S(N((R(R0(s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRBsgחgcAR&R'R(R)s method %s(RRvRytTOLR R(RBtmethodsR&R'R#tres((R0s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_gh_5555s  c C`swd}d}dt}tjtjg}xE|D]=}||ddd|d|}td|d |d|q2WdS( NcS`s|dkrdS|dSdS(Ng?gg333333?((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRBs gRQ?iiiR&R'g333333?R((R>RR{R}R (RBR(R'RR#R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_gh_5557s   tTestRootResultscB`seZdZRS(c C`sAtjdddddddd}d }tt||dS( NR0g?Ri,Ri.tflagiso converged: True flag: 'converged' function_calls: 46 iterations: 44 root: 1.0(RRLR trepr(R2R:t expected_repr((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_reprs  (RRR(((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRsc C`sd}d}d}tdd}d}tj||d|d |d |d d }t|||d dd |gd}d}tj||d|d |d |d d }t|||d dd dS(s&Test Halley's works with complex rootscW`s$|d|d|d||dS(Niii((RR6((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRBscW`sd|d||dS(Niii((RR6((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRscW`sBd|d}yt|}Wntk r2|SX|g|SdS(Nii(R^t TypeError(RR6tretvaltsize((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRs  g?g@g@g@RERYRZR[gư>iR(i N(g@g@g@(g@g@g@(RRRR (RBRRRtcoeffsR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_complex_halleys   * *c C`s#tjtjd}d|d|}t6}|jtdtjdd|gd}WdQXt |d gddd}t/}|jtd tjd d|}WdQXt |d dd}t/}|jtd tjdd|}WdQXt |ddS(sBTest secant method with a non-zero dp, but an infinite newton stepgQ?gi@g@sRMS ofcS`s |ddS(NgY@i((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRsR%i Nidg-C6?s Tolerance ofcS`s |ddS(Ng?i((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRsigcS`s |ddS(Ng?i((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRsigSt$?g9m4@gSt$g9m4@( R*R R+R,RtfilterRRRR (tdxtp0tsupR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_zero_der_nz_dps  %     c`sd}dd}d}d}||||}fd}tjtTtj|ddd d d gd d d||gdt}|jj stWdQXtj t 9tj|ddgd d d d||gdt}WdQXdS(s(Test that array newton fails as expectedg?ga2U0*#?g@g\mJA?gCl@c`s?dtj|dtjd|d|tj|S(Niig @gGz@(R*Rtlog10(tdarcy_frictiontretdia(t roughness(s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt colebrook_eqnsR%g{Gz?g?g vÖ?g333333?RiRER=N( RRRRRR?R1RRRR(tdiametertrhotmututreynolds_numberRtresult((Rs>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyttest_array_newton_failuress cC`s d}tj|dd}t|ddtjdtjtj|ddgd}t|ddtjdtjd}d}tj|ddd |}t|ddtjdtjtj|ddd |d |}t|ddtjdtjtj|ddgdd |}t|ddtjdtjtj|ddgdd |d |}t|ddtjdtjtj|dd d |}t|ddtjdtjtj|dd gdd |}t|ddtjdtjd S( s@Test that Newton or Halley don't warn if zero derivative at rootcS`s|d|dS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytf_zeroder_root*sR%iR(R'i cS`sd|dd|S(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytfder5scS`s d|dS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytfder29sRYRZg?N(RRR t_xtolt_rtol(RR:RR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt!test_gh8904_zeroder_at_root_fails&s*    " "c `sdfd}fd}fd}d}t||d|dt\}}|jsitt||d|d|dt\}}|jstd S( szTest that Halley's method realizes that the 2nd order adjustment is too big and drops off to the 1st order adjustment.i c`s"t|dtdS(Ng?(R (R(tn(s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRB[sc`st|dS(Ng?(R (R(R(s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytfp^sc`s-t|ddddS(Ng?i(R (R(R(s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pytfppasg?RYR=RZN(RR?R1R(RBRRR%trtR:((Rs>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt test_gh_8881Vs!'c C`s d}d}d}tjdgdtj}t||d|d|dt\}}|jslttjdd gdtj}tjt )t j||d|d|dt}Wd QXd }t j||d|d|dt}|jj std S( s_ Test that shape is preserved for array inputs even if fprime or fprime2 is scalar cS`s|dS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRBuscS`sd|S(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyRxscS`sdS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyR{sitdtypeRYRZR=iNcS`s#dtjtj|dtjS(NiR(R*tonestshapetfloat32(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt fpp_arrays( R*RRRR?R1RRRt IndexErrorRR( RBRRR%RR:tx0_arrayRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyt!test_gh_9608_preserve_array_shapeps   '$ (:t __future__RRRRtmathRRRRt numpy.testingRRR R tnumpyR*R R R Rtscipy.optimizeRRRtscipy._lib._utilRR]tscipy.optimize._tstutilsRRR.RR/tscipy._lib._numpy_compatRR+R,RR>RRRRRRRtobjectRRRRRRRRRR(((s>lib/python2.7/site-packages/scipy/optimize/tests/test_zeros.pyts: "" "       c     " 0