&]\c@`sdZddlmZmZmZddlZddlmZmZm Z m Z ddlm Z ddl ZddlmZmZmZmZdefdYZd efd YZdS( s# Unit test for SLSQP optimization. i(tdivisiontprint_functiontabsolute_importN(tassert_tassert_array_almost_equaltassert_allcloset assert_equal(traises(t fmin_slsqptminimizetNonlinearConstrainttBoundst MyCallBackcB`s eZdZdZdZRS(sJpass a custom callback function This makes sure it's being used. cC`st|_d|_dS(Ni(tFalset been_calledtncalls(tself((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt__init__s cC`st|_|jd7_dS(Ni(tTrueRR(Rtx((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt__call__s (t__name__t __module__t__doc__RR(((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyR s t TestSLSQPcB`seZdZdZddZddZddZddZddZddZ dd Z dd Z dd Z d Z d ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ d Z!d!Z"d"Z#d#Z$d$Z%d%Z&d&Z'd'Z(d(Z)d)Z*d*Z+d+Z,d,Z-d-Z.RS(.s Test SLSQP algorithm using Example 14.4 from Numerical Methods for Engineers by Steven Chapra and Raymond Canale. This example maximizes the function f(x) = 2*x*y + 2*x - x**2 - 2*y**2, which has a maximum at x=2, y=1. cC`sitd6|_dS(Ntdisp(R topts(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt setup_method$sg?cC`s@|d}|d}|d||d||dd|dS(s Arguments: d - A list of two elements, where d[0] represents x and d[1] represents y in the following equation. sign - A multiplier for f. Since we want to optimize it, and the scipy optimizers can only minimize functions, we need to multiply it by -1 to achieve the desired solution Returns: 2*x*y + 2*x - x**2 - 2*y**2 iii((RtdtsignRty((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytfun's  cC`sZ|d}|d}|d|d|d}|d|d|}tj||gtS(so This is the derivative of fun, returning a numpy array representing df/dx and df/dy. iiiii(tnptarraytfloat(RRRRRtdfdxtdfdy((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytjac7s   cC`s"|j|||j||fS(N(RR%(RRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt fun_and_jacCscC`stj|d|dgS(s Equality constraint ii(R R!(RRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytf_eqconFscC`stjddggS(s! Equality constraint, derivative ii(R R!(RRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt fprime_eqconJscC`s|j||dS(s Scalar equality constraint i(R'(RRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytf_eqcon_scalarNscC`s|j||djS(s( Scalar equality constraint, derivative i(R(ttolist(RRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytfprime_eqcon_scalarRscC`s tj|d|ddgS(s Inequality constraint iig?(R R!(RRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytf_ieqconVscC`stjddggS(s# Inequality constraint, derivative ii(R R!(RRR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt fprime_ieqconZscC`s tj|S(s Vector inequality constraint (R tasarray(RR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt f_ieqcon2^scC`stj|jdS(s* Vector inequality constraint, derivative i(R tidentitytshape(RR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytfprime_ieqcon2bsc C`s\t|jddgdd ddd|j}t|d|dt|jd d gdS( Ngg?targstmethodtSLSQPtoptionstsuccesstmessageii(g(R RRRRR(Rtres((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt$test_minimize_unbounded_approximatedgsc C`set|jddgdd d|jddd|j}t|d|d t|jd d gdS( Ngg?R3R%R4R5R6R7R8ii(g(R RR%RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_minimize_unbounded_givennscC`stjdd>t|jddgdddddfd d d |j}WdQXt|d |d t|jddgtd|jdkt|jddkdS(Ntinvalidtignoregg?R3tboundsg@g?R4R5R6R7R8ii(g(g@N(Ng?( R terrstateR RtNoneRRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt"test_minimize_bounded_approximatedus c C`sbt|jddgdd dtddd|j}t|d|d t|jd d gdS( Ngg?R3R%R4R5R6R7R8ii(g(R R&RRRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt test_minimize_unbounded_combinedsc C`szt|jddgdddidd6|jd6dd6dd d |j}t|d |d t|jd d gdS(Ngg?R3t constraintsteqttypeRR4R5R6R7R8i(g(g(R RR'RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt#test_minimize_equality_approximateds   c C`st|jddgd|jdddddidd 6|jd 6dd6d |j}t|d |d t|jddgdS(Ngg?R%R4R5R3RCRDRERR6R7R8i(g(g(R RR%R'RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_minimize_equality_givens   c C`st|jddgddd|jdddidd 6|jd 6dd6|jd6d |j}t|d |d t|jddgdS(Ngg?R4R5R%R3RCRDRERR6R7R8i(g(g( R RR%R'R(RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_minimize_equality_given2s    c C`st|jddgddd|jdddidd 6|jd 6dd6|jd6d |j}t|d |d t|jddgdS(Ngg?R4R5R%R3RCRDRERR6R7R8i(g(g( R RR%R)R+RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt(test_minimize_equality_given_cons_scalars    c C`st|jddgddd|jdddidd 6|jd 6dd6d |j}t|d |d t|jddgdddS(Ngg?R4R5R%R3RCtineqRERR6R7R8iitatolgMbP?(g(g(R RR%R,RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_minimize_inequality_givens    c C`st|jddgd|jdddddidd 6|jd 6|jd6d |j}t|d |d t|jddgdS(Ngg?R%R4R5R3RCRJRERR6R7R8ii(g( R RR%R/R2RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt1test_minimize_inequality_given_vector_constraintss     cC`st|jddgddd|jdddddgd id d 6|jd6dd6|jd6d|j}t|d|dt|jd d gddtd|jdkodkntd |jdkod kndS(Ngg?R4R5R%R3R>gig?RCRDRERR6R7R8RKgMbP?ii(g(gg?(ig?(g( R RR%R'R(RRRR(RR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt#test_minimize_bound_equality_given2s     'c C`sit|jddgdd dddd}|\}}}}}t|dk|t|ddgdS( Ngg?R3tiprintit full_outputii(g(RRRR(RR9Rtfxtitstimodetsmode((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_unbounded_approximateds c C`srt|jddgdd d|jdddd}|\}}}}}t|dk|t|d dgdS( Ngg?R3tfprimeROiRPii(g(RRR%RR(RR9RRQRRRSRT((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_unbounded_givens  c C`sut|jddgdd d|jgdddd}|\}}}}}t|dk|t|ddgdS( Ngg?R3teqconsROiRPi(g(RRR'RR(RR9RRQRRRSRT((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_equality_approximateds  c C`s~t|jddgd|jdd d|jgdddd }|\}}}}}t|dk|t|d d gdS( Ngg?RVR3RXROiRPi(g(RRR%R'RR(RR9RRQRRRSRT((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_equality_givens cC`st|jddgd|jdd d|jd|jddd d }|\}}}}}t|dk|t|d d gdS( Ngg?RVR3tf_eqconst fprime_eqconsROiRPi(g(RRR%R'R(RR(RR9RRQRRRSRT((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_equality_given2s   c C`st|jddgd|jdd d|jgdddd }|\}}}}}t|dk|t|d d gd d dS(Ngg?RVR3tieqconsROiRPiitdecimali(g(RRR%R,RR(RR9RRQRRRSRT((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_inequality_given s cC`st|jddgd|jdddddgd |jd |jd d d d}|\}}}}}t|d k|t|ddgddtd|d kodkntd|dkodkndS(Ngg?RVR3R>gig?R[R\ROiRPiR_i(g(gg?(ig?(RRR%R'R(RR(RR9RRQRRRSRT((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_bound_equality_given2s   $cC`sotddgddgdd}t|dgtddgd d dd}t|dgdS( NcS`s|dS(Ni((tz((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt'sg@R^cS`s |ddS(Nii((Rb((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRc(sROig?cS`s|dS(Ni((Rb((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRc,st f_ieqconscS`s|ddgS(Nii((Rb((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRc-s(RR(RR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_scalar_constraints%s    cC`s,tddgdddggdddS(NcS`s |ddS(Nii((Rb((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRc3siR>iRO(R(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_integer_bounds1sc C`s0tttddddgWdQXdS(NcS`s ddgS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRc9siii(t assert_raisest ValueErrorR(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_obj_must_return_scalar5s cC`s#tddddgdddS(NcS`sdgS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRc?siiiROi(R(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_obj_returns_scalar_in_list;sc C`svt}t|jddgdd ddd|d|j}t|d|d t|jt|j|d dS( Ngg?R3R4R5tcallbackR6R7R8tnit(g(R R RRRRRR(RRkR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt test_callbackAs   c C`sddg}d}d}td|didd6|d 6id d6|d 6fd ddfd d }|j}t||dddt||dkt|j|dS(NiicS`s|d|ddS(Niii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcUscS`s|dddS(Niii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcVscS`s|dd|ddS(Niii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcXsRCRDRERRJR>R4R5RKg:0yE>g:0yE(iN(iN(R R@RRRR7(RRtf1tf2tsol((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_inconsistent_linearizationJs      c C`ssddg}td|didd6dd6id d6d d6fd ddfd d}t|j |dS(NiicS`s|dd|ddS(Niii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcisRCRDREcS`s|d|ddS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcksRRJcS`s |ddS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRclsR>iR4R5(iN(iN(R R@RR7(RRRp((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_regression_5743ds   cC`sAd}t|dddgdd}t|jjdkdS(NcS`s8|dddd|dddd|dddS(Niiig?((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytfuncrsiR4R5i(i(R RR%R1(RRsRp((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt test_gh_6676qs c C`sd d fd d fd dftjdftjdffdtj fdfg}xC|D];}tt't|jddgd|ddWdQXqVWdS(Niiigg?R>R4R5(ii(ii(ii(ii(ii(ii(ii(R tinfRgRhR R(Rt bounds_listR>((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_invalid_boundsxs     cC`sd}t|dgdddd g}t|jt|jdddt|d gddddg}t|jt|jd ddt|d gddddg}t|jt|jdddt|dgddddg}t|jt|jd ddt|d gddddg}t|jt|jdddt|dgddddg}t|jt|jddddS(NcS`s|dddS(Niii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytfsi R4tslsqpR>iRKg|=iigi(Ni(iN(Ni(iN(ii(ii(R R@RR7RR(RRxRp((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_bounds_clippings& ! ! ! ! ! ! cC`sd}idd6dd6g}idd6dd6g}idd6dd6idd6dd6g}t|d gd d d |}t|jt|jd ddt|dgd d d |}t|jt|jdddt|dgd d d |}t|jt|jd ddt|d gd d d |}t|jt|jdddt|dgd d d |}t|jt|jd ddt|d gd d d |}t|jt|jd dddS(NcS`s|\}||d|dS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRxs RJREcS`sd|S(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcsRcS`s|dS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcscS`sd|S(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcscS`s|dS(Ni((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcsi R4RyRCiRKg|=iig(R RR7RR(RRxtcons_utcons_ltcons_ulRp((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_infeasible_initials.       c C`sd}d}d}d}ddf}tddd |tddd |f}t||d d d |d |}t|j dS(NcS`sd|dd|dS(Niiii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytcostscS`s|d|ddS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt ineqcons1scS`s|d|dS(Nii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyt ineqcons2siiiRERJRR4R5R>RC(ii(ii(ii(tdictR RR7(RRRRtx0R>tconsR9((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_inconsistent_inequalitiess    *!cC`srd}tddgtjtjg}t|ddgddd|}t|jt|jddgdS(NcS`s|dd|ddS(Niii((R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRcsiiR4RyR>(R R RuR RR7RR(RRxR>Rp((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_new_bounds_types  !! cC`s*dddY}|}|jdS(Nt NestedProblemcB`s,eZdZdZdZdZRS(cS`s d|_dS(Ni(t F_outer_count(R((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRscS`s|jd7_|jdkr-tdnt|jd dd}t|jt|jddg|dd |dd |d d S( Niis(Nested minimization failed to terminate.iiR4R5ii(ii(Rt ExceptionR tF_innerRR7RR(RRt inner_res((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytF_outers cS`s |ddd|dddS(Niii((RR((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRscS`sBt|jddd}t|jt|jdddgdS(NiR4R5i(iii(R RRR7RR(Rt outer_res((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pytsolves (RRRRRR(((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRs  ((R(RRtproblem((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyttest_nested_minimizations (/RRRRRR%R&R'R(R)R+R,R-R/R2R:R;RARBRFRGRHRIRLRMRNRURWRYRZR]R`RaReRfRiRjRmRqRrRtRwRzR~RRR(((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyRsZ                         #  (Rt __future__RRRtpytestt numpy.testingRRRRRRgtnumpyR tscipy.optimizeRR R R tobjectR R(((s>lib/python2.7/site-packages/scipy/optimize/tests/test_slsqp.pyts " "