[c@`sddlmZmZmZmZddlZddlZddlZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZd d lmZydd lmZeZWnek reZnXd efdYZejjddefdYZ defdYZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+ejjddZ,ejjddZ-ejjdd Z.ejjdd!Z/dS("i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tassert_allclose(t RandomStatei(tFittable1DModel(t Parameter(tmodels(tfittingi(tignore_non_integer_warning(toptimizetTestNonLinearConstraintscB`s}eZdZejjddZejjddZejjddZejjddZ RS(cC`stjdddd|_tjdddd|_tjddd|_|j|j|_|j|j|_t d }|j d |_ |jd |j |_ |jd |j |_ dS( Ni g-@tstddevg333333?i g?ig?iIidi(Rt Gaussian1Dtg1tg2tnptarangetxty1ty2Rtrandntntny1tny2(tselftrsn((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyt setup_classs u not HAS_SCIPYc C`shtjdddddditd6}tj}|||j|j}|jjdksdt dS(Ni tmeang-@R g333333?tfixedu amplitude( RRtTrueR tLevMarLSQFitterRRt amplitudetvaluetAssertionError(RRtfittertmodel((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_fixed_par(s  c C`syd}tjddddddi|d6}tj}|||j|j}t|jjd |j d d dS( NcS`sd|j}|S(Ni2(R (R%R((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttied3s i Rg-@R g333333?R'umeani2trtoligh㈵>( RRR R RRRRR"R (RR'RR$R%((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyt test_tied_par0s  ( c`sytjdddd}tjdddd}tj||gidg|6dg|6dg}tjdd d }||}||}tjjd }|d |}|d |} ||||| ddg} ddg} d} tj| | | f} d fd}t j || d|||| f\}}t |j |ddt |j j|j jdS(Ni g-@R g333333?i g?u amplitudeg#@ig?idicS`s-|tjd|dd||ddS(Ngiii(Rtexp(tAtpR((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyt compmodelNsc`sMtjtj|d|dd!|||d|d||fS(Niii(Rtraveltr_(R,tx1Rtx2R(R-(sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyterrfQs$targsR(igh㈵>(RRR t JointFitterRRtrandomRR/R tleastsqRt fitparamsR!R"(RRRtjfRRRRRRtp1tp2R+R,R2R7t_((R-sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_joint_fitter<s(     *c `stjdddd}dfd}dddg}||j}tjjd}||}tj||d|j|f\}}tj } | ||j|} t | j |d d d dS(Ng#@g-@R g333333?cS`s1|dtjd|dd||ddS(Nigii(RR*(R,R((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pytfunc^sc`s|||S(N((R,Rty(R=(sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyR2asidR3R(ii igMbP?( RRRRR5RR R6R R Rt parameters( RRR2tp0R>RtnytfitpartsR$R%((R=sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_no_constraintsZs  ' ( t__name__t __module__RtpytesttmarktskipifR&R)R<RD(((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyR s   u not HAS_SCIPYt TestBoundscB`s5eZdZdZdZdZdZRS(czC`sd}d}tjddd|_||j|tjjdddd|_tjd d d d d dddddddddddddddddd dd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdId>ddJdKd-dLdMdNdOdPdQdRdSdJdTdUdVdWdXdYdZd[dd[d\d]d^d_d`dadbdcdddedfd3dgd'dhdidjdkdldmdndodpdqdqdrdsd@d'dtdugy}|jdvdv|_dS(wNgg?gg?idtscaleg?tsizeg@g`@g@g@g@gH@g@g@g|@gP}@gz@g0@g`@g@g@gЈ@gH@g@g@g@g}@g`|@g@g@g<@g@g(@g0@g@g@gp~@gP|@gx@g`@g@g@g$@gD@gԗ@g@g@g@g }@g@g0@g@g@gܤ@gΣ@gX@g@g@g@g@g}@gȁ@g@g@gf@gr@g@g؛@g@g@g@g@gđ@gڦ@g @g؝@g8@g@g@g|@g`~@g@g@g@gL@g@gp@g؄@g ~@g@g@g@g@g$@gD@g(@g@g z@g|@g0{@g@g@g@gP@g0@g}@g@g{@gz@gp{@g}@g~@gX@g{@gz@i ( RtlinspaceRR5tnormalR>tarraytreshapetdata(RR+tBRQ((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRqs"-$!'c C`sd}d}idd6dd6}tj||d |}tj}|||j|j}|jj}|jj}|d|dd kst |d|dd kst |d|dd kst |d|dd kst dS(Ng?ggg@uslopegg?u intercepttboundsi iii(gg@(gg?gh㈵>gh㈵>gh㈵>gh㈵>( RtLinear1DR R RR>tslopeR"t interceptR#( Rt guess_slopetguess_interceptRSt line_modelR$R%RURV((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_bounds_lsqs    c C`sd}d}idd6dd6}tj||d |}tj}t|||j|j}WdQX|jj}|j j}|d|dd kst |d|dd kst |d|dd kst |d|dd kst dS(Ng?ggg@uslopegg?u interceptRSi iii(gg@(gg?gh㈵>gh㈵>gh㈵>gh㈵>( RRTR tSLSQPLSQFitterR RR>RUR"RVR#( RRWRXRSRYR$R%RURV((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_bounds_slsqps     c C`stjtjdtjd\}}iddgd6ddgd6ddgd6ddgd 6}tjd d d d dd ddddddd|}tj}|||||j}|jj }|j j }|j j } |j j } |d|ddks t |d|ddks*t |d|ddksHt |d|ddksft | d|ddkst | d|ddkst | d|d dkst | d |d dkst dS(!Ni gg&@ux_meanuy_meang?iux_stddevuy_stddevR!g$@tx_meang@ty_meantx_stddevg@ty_stddevtthetag?RSi iiigh㈵>gh㈵>gh㈵>gh㈵>gh㈵>gh㈵>gh㈵>gh㈵>(RtmeshgridRRt Gaussian2DR R RQR]R"R^R_R`R#( RtXtYRStgausst gauss_fitR%R]R^R_R`((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_bounds_gauss2d_lsqs,*        c C`stjtjdtjd\}}iddgd6ddgd6ddgd6ddgd 6}tjd d d d dd ddddddd|}tj}t|||||j}WdQX|j j }|j j }|j j } |j j } |d|ddkst|d|ddks:t|d|ddksXt|d|ddksvt| d|ddkst| d|ddkst| d|d dkst| d |d dkstdS(!Ni gg&@ux_meanuy_meang?iux_stddevuy_stddevR!g$@R]g@R^R_g@R`Rag?RSi iiigh㈵>gh㈵>gh㈵>gh㈵>gh㈵>gh㈵>gh㈵>gh㈵>(RRbRRRcR R[R RQR]R"R^R_R`R#( RRdReRSRfRgR%R]R^R_R`((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_bounds_gauss2d_slsqps.*         (RERFRRZR\RhRi(((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRJns     tTestLinearConstraintscB`seZdZdZRS(cC`stjd|_d|j_d|j_ddg|j_tjd|_|j|j|_ t d}|j d|_ |j |j |_ dS(Niigg"@i iI(Rt Polynomial1DR9tc0tc1twindowRRRR>RRRRA(RR((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRs   cC`sbt|jj_t|jj_tj}||j|j|j}t |j||jdS(N( RR9RlRRmR tLinearLSQFitterRR>R(RtpfitR%((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttests  (RERFRRq(((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRjs cC`sXtjdddddd}t|j_|jitd6td6td 6ksTtdS( NR!iRiR iu amplitudeumeanustddev(RRRRRtFalseR#(Rf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_set_fixed_1s c C`sGtjddddddditd6}|jjtksCtdS( NR!iRiR iRumean(RRRRRR#(Rf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_set_fixed_2scC`snd}tjdddddd}||j_|jjtk sKtt|jdtjsjtdS( NcS`s d|jS(Ni2(R (R%((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyt tie_amplitudesR!iRiR iu amplitude( RRR!R'RrR#t isinstancettypest FunctionType(RuRf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_set_tied_1s   c C`sJd}tjdddddddi|d 6}|jjsFtdS( NcS`s d|jS(Ni2(R (R%((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRusR!iRiR iR'u amplitude(RRR!R'R#(RuRf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_set_tied_2s c C`setjddddddditd6}t|j_|jitd 6td6td 6ksatdS( NR!iRiR iRumeanu amplitudeustddev(RRRRrRRR#(Rf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_unset_fixed s c C`snd}tjdddddddi|d 6}t|j_|jitd 6td 6td 6ksjtdS( NcS`s d|jS(Ni2(R (R%((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRusR!iRiR iR'u amplitudeumeanustddev(RRRrR!R'R#(RuRf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_unset_tieds   c C`sYtjdddddddid d 6}|jidd 6dd 6dd 6ksUtdS(NR!iRiR iRSiustddevu amplitudeumeang(iN(NN(NN(gN(RRtNoneRSR#(Rf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_set_bounds_1s cC`sXtjdddddd}d|j_|jid d6d d 6d d 6ksTtdS(NR!iRiR igu amplitudeumeanustddev(NN(NN(gN(RRR tminRSR}R#(Rf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_set_bounds_2&s  c C`sqtjdddddddid d 6}d|j_d|j_|jid d 6dd 6dd 6ksmtdS(NR!iRiR iRSiustddevu amplitudeumean(ii(NN(NN(NN(RRR}R RtmaxRSR#(Rf((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_unset_bounds.s  c C`sdtfdY}|jjdks.t|jjdksFt|jjdks^t|jjd ksvt|jjt kst|}|jj dkst|jj dkst|jjdkst|jjdkst|jjt kst|jidd6dd6ks2t|jit d6t d6ksUt|ddd idd6dd6d it d6t d6}|jj dkst|jj dkst|jjdkst|jjd kst|jjdkst|jjdkst|jjt ks4t|jjt ksLt|jidd6dd6ksot|jit d6t d6kstd S(uRegression test for https://github.com/astropy/astropy/issues/2396 Ensure that default constraints defined on parameters are carried through to instances of the models those parameters are defined for. tMyModelcB`sAeZeddZedddddeZedZRS(tdefaultiiRRcS`s |||S(N((Rtatb((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pytevaluateCs(RERFRRRRt staticmethodR(((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyR?siiuaubiiRSiRN(iN(iN(NN(iN(iN(iN(iN(iN(iN(iN( RRRR#RRRSR}RRR"Rr(Rtm((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_default_constraints8s4 ## #c C`stjdddddddidd 6d itd 6}tjd d d }tj|d d}tj}||||}|jdkst |jdkst |j dkst t |j _ ||||}|jdkst |jdks t dS(u Regression test for https://github.com/astropy/astropy/issues/2235 Currently doesn't test that the fit is any *good*--just that parameters stay within their given constraints. R!iRiR iRSiumeanRu amplitudeii ig@N(ii( RRRRRMR*R R RR#R!RrR(RRR>tftfitted_1tfitted_2((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyt)test_fit_with_fixed_and_bound_constraintses    c C`s2dtfdY}|dddd}tjd}||}|}tj}||||}tj|jdsttj|jdst|}d |j_ tj}||||} tj|jdsttj|jds ttj |j d d ks.td S(u Regression test for https://github.com/astropy/astropy/issues/2400 Checks that bounds constraints are obeyed on a custom model that does not define fit_deriv (and thus its Jacobian must be estimated for non-linear fitting). RcB`s5eZeddZeddZedZRS(RiicS`s |||S(N((RRR((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRs(RERFRRRRR(((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyRsRg?RiidiiufjaciN(ii( RRRR R tallcloseRR#RRStanytfit_info( Rtm_realRR>RRRtm2tf2R((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyt1test_fit_with_bound_constraints_estimate_jacobians       c!C`snddddddddd d d d d ddddgddddddddddddddd d!d"gd#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3gd4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDgdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUgdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfgdgdhdidjdkdldmdndodpdqdrdsdtdudvdwgdxdydzd{d|d}d~ddddddddddgdddddKddddddddddddgdddddddddddddddddgdddddddddddddddddgdddddddddddddddddgdddddddddddddddddgdddddddddddddddddgdddddddddddddddddgddddddddddd d d d d ddgddddddddddddddddd gg}tjd!d"d#d"}tj}tjdd$dd$f\}}|||||}t|jj|j jgd%d&gd'd(t|j jd)t|j jd*d'd+t|j jd,d'd-dS(.Ng ףp=*KgQ+@gfffffFAgfffff!@gQagQgM@g(\-g̜igQ>g)\_gR^@gp= ׳Q@gQ^[gzGZgRA@g ףp=J2@g\(_gQHVgq= ףO@gQ^I@g ףp=N@g(\3@glP@g\(|S@g)\H]@gp= ka@g)\HJ@gzGm@gp= #Y@g)\Hn@gp= דfgq= ף@c@g(\ gQV@gX@g(\¥]g(\eg(\]gףp= T@gp= ׳Ug= ףp=-@gq= ףS@g= ףp=@gQ~K@gGzBp@g= ףp agQ4g33333cSgq= ףe@gQ`@gq= ף@@gQ@gQE@g(\Q@gHzwQ@g\(@@gq@gQf@gGz>i@g= ףpu@gq= ףHu@gq= ףxd@gףp= cg= ףp}5@gQbgQW@gfffffFE@gRT@g= ףp5f@g(\:@g)\1@gHzG@gGz.?gHzv@g{Gh@g\(J@g33333Kn@gR{@gq= ף@g@g(\a@gR@gQUg{G:2gHzd@g(\E\@gzGBg(\`@g)\Q@gQZ@gGzh@gGzq@gףp= @gfffff>@g{G@g"@gu@gya@gGzc@gQg= ףp:g̬B@gGzZ@gQ?@g(\?U@g(\r_g`h@gfffff~@gQ@gGzb@g33333@gGzh@gQ@gQJy@gQ\@gףp= [@gGz<@g33333^@gGz.Mgp= f@g\(F@gRqD@gQ[@g)\z@gGzt@gQ@g\(@g ףp=@gچ@gGz@g(\Vy@gfffff3gQ@gag ףp=b@g= ףp=9@g\($a@g̼UgHz.@gGzf@gGz؅@g33333'@gR볏@g{Gp@g= ףp=@gGz|@gQ8o@g33333W@g(\•D@g(\gQf@gGzg@gRA@gQ:@g)\h@gp= Bg= ףp@gzG@gQ@g ףp=@gȅ@g)\~@g(\w@g(\u2gq= ףpPgfffffX@gR(@gQYg33333]@g(\g@gGzt_@g33333ws@g= ףpw@gQS@gRe|@g@g{G@g= ףp@gHzg@gQj@gQ=k@gp= ף@g(\u;@gGzegףp= Tgp= TgRKgQ T@gףp= p@g33333#q@g)\0e@gq= ףpt@gfffff~@gGzX{@g\(,l@gGzk@g ףp= @gGzWgףp= C@g\(L@gQb@gp= Y@g{GzA@gףp= O`@gQgfffff&G@g(\_Y@gq= ף\@g\(,k@g(\:o@g\@gףp= U@gp= sZ@gGzGgzG#gRTgfffffb@g\(T@g= ףp}7@gf@gQ[gQ\gQn@gq= ףY@g(\o@gףp= od@g(\BL@g)\@@gzG@g(\[gQV@gfffff&h@gQ+YgzG.@gGzU@gq= ףp:g33333Chgףp= W[@gq= ףX@g)\8_gffffff0gHzU@gHz\@giPg(\BMgQ87gGzNE@gQb@g{G:jg(\[@gHzP@gfffffv]g(\wbgq= ף^gzG%@g= ףpF@gHz]@gp= CGg)\RgGzRgQ85@g(\?gRU@gQ7@g(\µFg(@gQeg ףp=;@gHzN@gGzU@g ףp=<@gGzNG@gףp= Zg{GZL@gfffffVg(\0g̜Pg{GagQ?g ףp= M@gq= ףq@gGzNgQ"gGz6tg,M@g)\T@gGzNYg4b@R]iR^ig['Ǝ@gSix @R(gTqs*>g!0N%'@g(#@gƠ>gU1?g>( RRcR R RtmgridRR!R"R`R]R^R_(Rqtg_initR$R>Rtg_fit((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyttest_gaussian2d_positive_stddevsV! %c C`s tjdddddd}tj}tjddd }tjddd }tj||\}}|||}tj|j}|j |_ d d l m }|d Ntj j |j} |j | _ |||||d| d |} t| j|jdd|||||d| d d} t| j|jddt|j_|||||d| d |} t| j|jdd|||||d| d d} t| j|jddtjddddddd} | ||}|| |||d| d d} t| j| jdd|| |||d| d |} t| j| jddt| j_|| |||d| d |} t| j| jdd|| |||d| d d} t| j| jddWdQXdS(Ngffffff$@g333333@iig333333?gffffff?iiii(tNumpyRNGContextiItweightsR(g?itc0_0tc1_0tc0_1g @(RRcR R RRMRbtonesRLtshapetutilsRR5RRR?R}RR_Rt Polynomial2DR( tgauss2dR$RdReRR>tztwRRRR:((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pyt test_2d_models@    ## ##!## ##(0t __future__RRRRRwRGtnumpyRt numpy.testingRt numpy.randomRtcoreRR?RtRR RR tscipyR Rt HAS_SCIPYt ImportErrorRrtobjectR RHRIRJRjRsRtRyRzR{R|R~RRRRRRR(((sFlib/python2.7/site-packages/astropy/modeling/tests/test_constraints.pytsB"      Sh      -,6