ó î&]\c@`s˜ddlmZmZmZddlZddlZddlmZddl m Z ddl m Z m Z mZmZmZddlmZddlmZmZmZmZmZmZddlmZd dd „ƒYZd dd „ƒYZd dd„ƒYZdd d„ƒYZdd!d„ƒYZdefd„ƒYZ defd„ƒYZ!defd„ƒYZ"dd"d„ƒYZ#de fd„ƒYZ$dS(#i(tdivisiontprint_functiontabsolute_importN(t block_diag(t csc_matrix(tTestCasetassert_array_almost_equaltassert_array_lesstassert_allclosetassert_(traises(tNonlinearConstrainttLinearConstrainttBoundstminimizetBFGStSR1(tsuppress_warningstMaratoscB`sJeZdZdddd„Zd„Zd„Zd„Zed„ƒZ RS(s²Problem 15.4 from Nocedal and Wright The following optimization problem: minimize 2*(x[0]**2 + x[1]**2 - 1) - x[0] Subject to: x[0]**2 + x[1]**2 - 1 = 0 i<cC`si|dtj}tj|ƒtj|ƒg|_tjddgƒ|_||_||_d|_ dS(Ni´gð?g( tnptpitcostsintx0tarraytx_optt constr_jact constr_hesstNonetbounds(tselftdegreesRRtrads((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyt__init__s !  cC`s(d|dd|ddd|dS(Niii((Rtx((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pytfun"scC`s'tjd|ddd|dgƒS(Niii(RR(RR"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pytgrad%scC`sdtjdƒS(Nii(Rteye(RR"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pythess(scC`sgd„}|jdkr$d„}n |j}|jdkrHd„}n |j}t|dd||ƒS(NcS`s|dd|ddS(Niii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#-scS`sd|dd|dggS(Niii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pytjac1scS`sd|dtjdƒS(Nii(RR%(R"tv((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR&7si(RRRR (RR#R'R&((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pytconstr+s     N( t__name__t __module__t__doc__RR!R#R$R&tpropertyR)(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyRs    tMaratosTestArgscB`sSeZdZdddd„Zd„Zd„Zd„Zd„Ze d„ƒZ RS( s²Problem 15.4 from Nocedal and Wright The following optimization problem: minimize 2*(x[0]**2 + x[1]**2 - 1) - x[0] Subject to: x[0]**2 + x[1]**2 - 1 = 0 i<cC`s{|dtj}tj|ƒtj|ƒg|_tjddgƒ|_||_||_||_ ||_ d|_ dS(Ni´gð?g( RRRRRRRRRtatbRR(RR/R0RRRR ((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR!Gs!    cC`s.|j|ks|j|kr*tƒ‚ndS(N(R/R0t ValueError(RR/R0((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyt _test_argsQscC`s8|j||ƒd|dd|ddd|dS(Niii(R2(RR"R/R0((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#UscC`s7|j||ƒtjd|ddd|dgƒS(Niii(R2RR(RR"R/R0((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR$YscC`s!|j||ƒdtjdƒS(Nii(R2RR%(RR"R/R0((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR&]scC`sgd„}|jdkr$d„}n |j}|jdkrHd„}n |j}t|dd||ƒS(NcS`s|dd|ddS(Niii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#cscS`sd|dd|dggS(Niii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR'gscS`sd|dtjdƒS(Nii(RR%(R"R(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR&msi(RRRR (RR#R'R&((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR)as     N( R*R+R,RR!R2R#R$R&R-R)(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR.?s    tMaratosGradInFunccB`sPeZdZdddd„Zd„Zed„ƒZd„Zed„ƒZ RS(s²Problem 15.4 from Nocedal and Wright The following optimization problem: minimize 2*(x[0]**2 + x[1]**2 - 1) - x[0] Subject to: x[0]**2 + x[1]**2 - 1 = 0 i<cC`si|dtj}tj|ƒtj|ƒg|_tjddgƒ|_||_||_d|_ dS(Ni´gð?g( RRRRRRRRRRR(RRRRR ((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR!}s !  cC`sQd|dd|ddd|dtjd|ddd|dgƒfS(Niiii(RR(RR"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#…s'cC`stS(N(tTrue(R((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR$‰scC`sdtjdƒS(Nii(RR%(RR"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR&scC`sgd„}|jdkr$d„}n |j}|jdkrHd„}n |j}t|dd||ƒS(NcS`s|dd|ddS(Niii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#’scS`sd|dd|dggS(Niii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR'–scS`sd|dtjdƒS(Nii(RR%(R"R(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR&œsi(RRRR (RR#R'R&((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR)s     N( R*R+R,RR!R#R-R$R&R)(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR3us   tHyperbolicIneqcB`sGeZdZddd„Zd„Zd„Zd„Zed„ƒZ RS(sProblem 15.1 from Nocedal and Wright The following optimization problem: minimize 1/2*(x[0] - 2)**2 + 1/2*(x[1] - 1/2)**2 Subject to: 1/(x[0] + 1) - x[1] >= 1/4 x[0] >= 0 x[1] >= 0 cC`sIddg|_ddg|_||_||_tdtjƒ|_dS(Nig–~TÃ>ÿ?g þ~1[²¶?(RRRRR RtinfR(RRR((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR!­s   cC`s(d|dddd|dddS(Niiigà?gà?gà?((RR"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#´scC`s|dd|ddgS(Niiigà?((RR"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR$·scC`s tjdƒS(Ni(RR%(RR"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR&ºscC`sjd„}|jdkr$d„}n |j}|jdkrHd„}n |j}t|dtj||ƒS(NcS`sd|dd|dS(Nii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#¿scS`sd|ddddggS(Niÿÿÿÿiii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR'ÃscS`s;d|dtjd|ddddgddggƒS(Niiii(RR(R"R(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR&És*gÐ?(RRRR RR6(RR#R'R&((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR)½s     N( R*R+R,RR!R#R$R&R-R)(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR5¤s    t RosenbrockcB`sGeZdZddd„Zd„Zd„Zd„Zed„ƒZRS(sRosenbrock function. The following optimization problem: minimize sum(100.0*(x[1:] - x[:-1]**2.0)**2.0 + (1 - x[:-1])**2.0) iicC`sItjj|ƒ}|jdd|ƒ|_tj|ƒ|_d|_dS(Niÿÿÿÿi( Rtrandomt RandomStatetuniformRtonesRRR(Rtnt random_statetrng((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR!ÙscC`sPtj|ƒ}tjd|d|d ddd|d dddƒ}|S(NgY@iiÿÿÿÿg@taxisi(Rtasarraytsum(RR"tr((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#ßs4 cC`sÐtj|ƒ}|dd!}|d }|d}tj|ƒ}d||dd||d|dd||dd+d|d|d|dddd|d|dtj|d|ƒddg|_ddg|_d|_dS(Niiÿÿÿÿgà¿gn£¼à?g$¹ü‡ôÛÏ?(R7R!RRRR(RR=((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR! scC`s)ddgg}d}t|tj |ƒS(Nii(R RR6(RtAR0((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR)s(R*R+R,R!R-R)(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyROs tBoundedRosenbrockcB`seZdZdd„ZRS(sRosenbrock subject to inequality constraints. The following optimization problem: minimize sum(100.0*(x[1] - x[0]**2)**2.0 + (1 - x[0])**2) subject to: -2 <= x[0] <= 0 0 <= x[1] <= 2 Taken from matlab ``fmincon`` documentation. icC`sMtj|d|ƒddg|_d|_tddgddgƒ|_dS(Nigš™™™™™É¿gš™™™™™É?iþÿÿÿi(R7R!RRRR R(RR=((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR! s (R*R+R,R!(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyRQs tEqIneqRosenbrockcB`s)eZdZdd„Zed„ƒZRS(s*Rosenbrock subject to equality and inequality constraints. The following optimization problem: minimize sum(100.0*(x[1] - x[0]**2)**2.0 + (1 - x[0])**2) subject to: x[0] + 2 x[1] <= 1 2 x[0] + x[1] = 1 Taken from matlab ``fimincon`` documentation. icC`s>tj|d|ƒddg|_ddg|_d|_dS(Niiÿÿÿÿgà¿gæWs€`ŽÚ?gÙ|\*ÆÅ?(R7R!RRRR(RR=((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR!1scC`sPddgg}d}ddgg}d}t|tj |ƒt|||ƒfS(Nii(R RR6(RtA_ineqtb_ineqtA_eqtb_eq((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR)7s (R*R+R,R!R-R)(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyRR's  tEleccB`s_eZdZddd d d„Zd„Zd„Zd„Zd„Zd„Z e d „ƒZ RS( sªDistribution of electrons on a sphere. Problem no 2 from COPS collection [2]_. Find the equilibrium state distribution (of minimal potential) of the electrons positioned on a conducting sphere. References ---------- .. [1] E. D. Dolan, J. J. Mor'{e}, and T. S. Munson, "Benchmarking optimization software with COPS 3.0.", Argonne National Lab., Argonne, IL (US), 2004. iÈic C`sì||_tjj|ƒ|_|jjddtj|jƒ}|jjtj tj|jƒ}tj|ƒtj|ƒ}tj|ƒtj|ƒ}tj|ƒ} tj ||| fƒ|_ d|_ ||_ ||_d|_dS(Nii(t n_electronsRR8R9R>R:RRRthstackRRRRRR( RRXR=RRtphitthetaR"tytz((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR!Os ""   cC`sB||j }||jd|j!}|d|j}|||fS(Ni(RX(RR"tx_coordty_coordtz_coord((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyt_get_cordinates_s cC`ss|j|ƒ\}}}|dd…df|}|dd…df|}|dd…df|}|||fS(N(RaR(RR"R^R_R`tdxtdytdz((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyt_compute_coordinate_deltases cC`ss|j|ƒ\}}}tjddƒ#|d|d|dd}WdQXd|tj|ƒttr_interior_pointiisInvalid termination condition.(s2-points3-pointR‚(ii(!RRR3R5RR7RORRRQRWR$tFalseR&R4Rtfiltert UserWarningRR#RRR)RRRR"tstatusRt optimalityt tr_radiusRƒtbarrier_parametert RuntimeError(Rtlist_of_problemstprobR$R&tsuptresult((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyttest_list_of_problemsÂsb         !   cC`sMd„}d g}t|ddgd|ddƒ}t|jd d d ƒdS( NcS`s |ddS(Nii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#ÿsiþÿÿÿiRgø¿RRƒs trust-constriR…i(iþÿÿÿi(RRR"(RR#Rtres((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyttest_default_jac_and_hessþs  !c C`sSd„}dg}t|ddgd|ddd d ƒ}t|jd d d ƒdS(NcS`s |ddS(Nii((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR#siþÿÿÿiRgø¿RRƒs trust-constrR's2-pointiR…i(iþÿÿÿi(RRR"(RR#RR”((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyttest_default_hesss    c C`sÝtƒ}t|j|jddd|jd|jƒ}t|j|jddddƒ}tjtƒ&t|j|jddddƒ}WdQXt |j |j dd ƒt |j |j dd ƒt |j |j dd ƒdS( NRƒs trust-constrR'R&sL-BFGS-Bs2-points3-pointR…i( R7RR#RR$R&tpytesttwarnsR‰RR"R(RRR’tresult1tresult2((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyttest_no_constraints s  c `stƒ‰‡fd†}tˆjˆjdddˆjd|dˆjdˆjƒ}ˆjdk rt |j ˆjdd ƒn|j d kr¡t |j d ƒn|j d kråt |jd ƒ|jd kråt |jd ƒqån|j dkrtdƒ‚ndS(Nc`sˆj|ƒ}|j|ƒS(N(R&tdot(R"tpRM(R(sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pythessp sRƒs trust-constrR'RžRR„R…iig:Œ0âŽyE>R†iisInvalid termination condition.(ii(RRR#RR$RR)RRRR"RŠRR‹RŒRƒRRŽ(RRžR’((RsMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyt test_hessps"   cC`stddƒ}t|j|jdddd|jd|jd|jd|jƒ}|jdk r|t |j |jd d ƒn|j d kržt |jd ƒn|j d krât |jd ƒ|jd krât |jd ƒqân|j dkrtdƒ‚ndS(NR/iêRƒs trust-constrR'R&RR„R…iig:Œ0âŽyE>R†iisInvalid termination condition.(R/iê(ii(R.RR#RR$R&RR)RRRR"RŠRR‹RŒRƒRRŽ(RRR’((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyt test_args:s   c C`sAtƒ}ttt|j|jddddddd|jƒdS(NRƒs trust-constrR's2-pointR&R„(RR R1RR#RR)(RR((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyttest_raise_exceptionRs c C`s‹d„}td„dgdd„dd„d|d d ƒ}t|jd ƒƒt|jd d ƒdkƒt|jdd ƒdkƒdS(NcS`s$td|kƒtd|kƒdS(Ntnittniter(R (R"tinfo((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pytcallback]scS`s|dS(Ni((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pytasiR'cS`sd|S(Ni((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR¦asR&cS`sdS(Ni((R"((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR¦bsR¥Rƒs trust-constrtsuccessR¢iÿÿÿÿiR£(RR tget(RR¥R’((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyttest_issue_9044Xs  ( R*R+R—tmarktslowR“R•R–R›RŸR R¡R©(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyR~Às<      (((((((%t __future__RRRtnumpyRR—t scipy.linalgRt scipy.sparseRt numpy.testingRRRRR R tscipy.optimizeR R R RRRtscipy._lib._numpy_compatRRR.R3R5R7RORQRRRWR~(((sMlib/python2.7/site-packages/scipy/optimize/tests/test_minimize_constrained.pyts$  (.-6/..