[c@`skdZddlmZmZmZmZddlZddlZddl Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZmZdd lTd d lmZd dlmZddlmZd dlmZd dlmZddlmZddl Z yddl!m"Z"e#Z$Wne%k rQe&Z$nXe#Z'yddl(m)Z)Wn;e%k ryddl)Z)Wqe%k re&Z'qXnXyddl*m+Z+e#Z,Wne%k re&Z,nXe-e.gZ/dZ0de1fdYZ2de1fdYZ3ej4j5dde1fdYZ6de1fdYZ7ej4j5dd e1fd!YZ8ej4j5d"ej4j5d#d$e1fd%YZ9ej4j5dd&e1fd'YZ:ej4j5dd(e1fd)YZ;ej4j5dd*d.d+YZ<ej4j5dd,Z=ej4j5dd-Z>dS(/u! Module to test fitting routines i(tabsolute_importtunicode_literalstdivisiontprint_functionN(tlinalg(tassert_allclosetassert_almost_equali(tirafutili(tmodels(tFittable2DModelt Parameter(t*i(tNumpyRNGContext(tget_pkg_data_filename(tignore_non_integer_warning(t sigma_clip(tAstropyUserWarning(tpopulate_entry_points(toptimize(tmock(t EntryPointi7tTestPolynomial2DcB`sDeZdZdZdZdZejjddZ RS(u Tests for 2D polynomail fitting.cC`sntjd|_tjddddf\|_|_d}||j|j|_t|_ dS(NiicS`s8dd|d|dd|d|dd||S(Niiiiii((txty((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytpoly2?s( Rt Polynomial2DtmodeltnptmgridRRtztLinearLSQFittertfitter(tselfR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt setup_class;s + cC`sx|jjd|jd|j}tj||jjd}|j|j|j|j|j}t |j |dS(NRRi( Rt fit_derivRRRtlstsqRtflattenRRt parameters(R tvtpt new_model((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_poly2D_fittingDs!$cC`sG|j|j|j|j|j}t||j|j|jdS(N(RRRRRR(R R(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt test_evalJs$u not HAS_SCIPYcC`snddddddg|j_t}||j|j|j|j}t|jddd d d d gdS( Ng333333?g?g333333@g @g@g@iiiiii(RR%tLevMarLSQFitterRRRR(R tnlfitterR(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt#test_polynomial2D_nonlinear_fittingNs !( t__name__t __module__t__doc__R!R)R*tpytesttmarktskipifR-(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR8s   t TestICheb2DcB`s_eZdZdZdZdZejjddZ ejjddZ RS(u Tests 2D Chebyshev polynomial fitting Create a 2D polynomial (z) using Polynomial2DModel and default coefficients Fit z using a ICheb2D model Evaluate the ICheb2D polynomial and compare with the initial z cC`s}tjd|_tjddddf\|_|_|j|j|j|_tjdd|_ t |_ dS(Nii( RRtpmodelRRRRRt Chebyshev2Dtcheb2RR(R ((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR!_s +c C`stjd|j_tjdddddddd d g }|j|j|j}|j|j|j|j|}t|j|dS( Ni g@g@gy@g@g @g@@g{@g@g`@( RtarangeR7R%tarrayRRRR(R R'RR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_default_paramsfs  !cC`sM|j|j|j|j|j}||j|j}t|j|dS(N(RR7RRRR(R Rtz1((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_poly2D_cheb2Dns$u not HAS_SCIPYc C`stjdd}tjd|_||j|j}ddddddd d d g |_t}|||j|j|}t|jd d dddddddg dddS(Nii g?g333333?g?g333333@g @g@g@g@g!@iiiiiiiitatoli ig& .>( RR6RR8R%RRR+R(R tcheb2dRR,R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt"test_chebyshev2D_nonlinear_fittingss$ *c C`stjdd}tjd|_||j|j}ddddddd d d g |_t}tj|j}|||j|j|d |}t |jd ddddddddg dddS(Nii g?g333333?g?g333333@g @g@g@g@g!@tweightsiiiiiiiiR=i ig& .>( RR6RR8R%RRR+t ones_likeR(R R>RR,R@R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt/test_chebyshev2D_nonlinear_fitting_with_weights~s$ !*( R.R/R0R!R:R<R1R2R3R?RB(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR4Vs     u not HAS_SCIPYtTestJointFittercB`s)eZdZdZdZdZRS(uA Tests the joint fitting routine using 2 gaussian models cC`s#tjddddd|_tjddddd|_t|j|jgidg|j6dg|j6d g|_tjdd d |_|j|j}|j|j}t t tj j d }Wd QX|d||_ |d||_|j|j|j |j|jd S(u Create 2 gaussian models and some data with noise. Create a fitter for the two models keeping the amplitude parameter common for the two models. i tmeang-@tstddevg333333?i g?u amplitudeg#@ig?idNi(Rt Gaussian1Dtg1tg2t JointFittertjfRR8RR t _RANDOM_SEEDtrandomtrandntny1tny2(R ty1ty2tn((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR!s cC`sFt|jjd|jjdt|jjd|jjddS(uH Tests that the amplitude of the two models is the same iN(RRJt fitparamsRGR%RH(R ((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_joint_parameters!c`sddg}ddg}d}tj|||f}dfd}tj||d|j|j|j|jf\}}t||jj d d d S(uk Tests the fitting routine with similar procedure. Compares the fitted parameters. g-@g333333?i g?g#@cS`s-|tjd|dd||ddS(Ngiii(Rtexp(tAR'R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRsc`sMtjtj|d|dd!|||d|d||fS(Niii(Rtraveltr_(R'tx1RPtx2RQ(R(sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyterrfuncs*targstrtoli iNg{Gz?( RRXRtleastsqRRNRORRJRS(R tp1tp2RVR'R[tcoefft_((RsBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_joint_fitters   '(R.R/R0R!RTRc(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRCs  tTestLinearLSQFittercB`s>eZdZdZdZdZdZdZRS(c C`sttjjdd}t|"}|j}|jd}WdQXtj|d}|j }t |j d}t j |dd|j}t} | ||j|j} t| jtj|dd dS( u@Tests fitting a 1D Chebyshev polynomial to some real world data.udatauidcompspec.fitsubeginNiuordertdomainR]g?(R tostpathtjointopentreadtsplitRtIdentifyRecordRatinttfieldsRt Chebyshev1Dt get_rangeRRRRR%RR9( R t test_filetftlinestreclisttrecordtcoeffstordert initial_modelRt fitted_model((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_chebyshev1Ds    c C`stjdddddgdd}tjd}||dt}|jdksZttt'|tj j dd d |j}Wd QXt }||||}t ||dt|d d d S(u-Tests fitting multiple models simultaneously.tdegreeitc0itn_modelsi tmodel_set_axisig{Gz?tsizeNR]g?(ii ( Rt Polynomial1DRR8tFalsetshapetAssertionErrorR RKRLtnormalRR(R t init_modelRt y_expectedRRRy((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_linear_fit_model_sets$ ( c C`stjdddddgdd}tjd}tjd}|||dt}|jdkslttt'|tj j dd d |j}Wd QXt }|||||}t |||dt|d d d S(u0Tests fitted multiple 2-D models simultaneously.R{itc0_0iR}i R~ig{Gz?RNR]g?(ii ( RRRR8RRRR RKRLRRR(R RRRt z_expectedRRRy((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_linear_fit_2d_model_sets$ ( cC`stjdddd}t|j_tjd}d|d||}t}||||}t|j dddgd d d S( uX Tests fitting a polynomial model with a fixed parameter (issue #6135). R{itc1ii g?g@g?R=g+=N( RRtTrueRtfixedRR8RRR%(R RRRRRy((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_linear_fit_fixed_parameters  cC`stjdddddgdd}t|j_tjd}tjd|d||d|g}t}||||}t |j d d gd d t |jd dgd d t |j dd gd d dS(uV Tests fitting a polynomial model set with a fixed parameter (#6135). R{iRiiR}i g?g@gR=g+=g?gN( RRRRRRR8R9RRR|tc2(R RRtyyRRy((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt)test_linear_fit_model_set_fixed_parameter s$ ) c C`stjdddddgdddgddditd 6td 6}tjd d d d f\}}tjd|d |d||d||d||g}t}|||||}t|||dt|dddS(uX Tests fitting a 2d polynomial model set with fixed parameters (#6135). R{itc1_0itc0_1gR}Ruc1_0uc0_1iig?g?g?R~R=g+=N( RRRRRR9RRR(R RRRtzzRRy((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt-test_linear_fit_2d_model_set_fixed_parameterss'%A (R.R/RzRRRRR(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRds      tTestNonLinearFitterscB`steZdZdZdZdZdZdZej j de dZ dZ d Zd ZRS( u?Tests non-linear least squares fitting and the SLSQP algorithm.cC`sdddg|_tjddd|_dtj|j}tttjjd|}WdQXd}||j|j||_ t j ddd d|_ dS( Nidiiii g?g@cS`s1|dtjd|dd||ddS(Nigii(RRU(R'R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytfunc<sRE( tinitial_valuesRR8txdataRAR RKRLRtydataRRFtgauss(R tsigmatyerrorR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR!3s  cC`st}||j|j|j}tjdddd}t}|||j|jdt}t|j|jdd d S( uk Runs `LevMarLSQFitter` with estimated and analytic derivatives of a `Gaussian1D`. idg@REitestimate_jacobianR]i iNgMbP?( R+RRRRRFRRR%(R RRtg1etefittertemodel((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt test_estimated_vs_analytic_derivBs   cC`sd|jd}t}||j|j|jd|}tjdddd}t}|||j|jd|dt}t|j|jd d d S(uk Runs `LevMarLSQFitter` with estimated and analytic derivatives of a `Gaussian1D`. g?g$@R@idg@REiRR]i iNgMbP?( RR+RRRRFRRR%(R R@RRRRR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt-test_estimated_vs_analytic_deriv_with_weightsOs ! $c`st}||j|j|jdt}dfd}tj||jd|j|jf}t|j |ddd d S( uX Tests results from `LevMarLSQFitter` against `scipy.optimize.leastsq`. RcS`s1|dtjd|dd||ddS(Nigii(RRU(R'R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRgsc`s|||S(N((R'RR(R(sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR[jsR\iR]i iNgMbP?( R+RRRRRR^RRR%(R RRR[tresult((RsBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_with_optimize^s   c C`st}||j|j|jdt}||j|j|jdtdtj|j}t|j|jdd tj |j}d|ddd<|dk}||j|j||j|dt}||j|j|jdtd|}t|j|jdd dS( uD Tests results from `LevMarLSQFitter` with weights. RR@R]i ig?Nig-C6?g-C6?( R+RRRRRRARR%t zeros_like(R RRtwithwR@tmask((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_with_weightsqs     u fitter_classcC`sut}|}t ||j|j|j}WdQX||j|j|j}t|j|jdddS(u@Tests results from non-linear fitters against `LevMarLSQFitter`.NR]i ig-C6?(R+RRRRRR%(R t fitter_classtlevmarRR(R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_fitter_against_LevMars   !cC`stjdddd}t|j_t}t}t|||j|j }WdQX|||j|j }t |j |j dd dS( ub Runs `LevMarLSQFitter` and `SLSQPLSQFitter` on a model with constraints. idiREiNR]i ig-C6?( RRFRRDRR+tSLSQPLSQFitterRRRRR%(R RGRtfslsqpt slsqp_modelR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_LSQ_SLSQP_with_constraintss    c C`sdtfdY}tjddd}}tt?|j||dd}|tjjdd d |j7}Wd QXt }|d d}|||||}t |j ddgd dd S(u)A basic test for the `SimplexLSQ` fitter.t RosenbrockcB`s)eZeZeZedZRS(cS`s ||d|||ddS(Ni((RRtatb((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytevaluates(R.R/R RRt staticmethodR(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRs  gg@idg?gY@gg?RNiR]g{Gz?( R RtlinspaceR RKRRLRRtSimplexLSQFitterRR%(R RRRRRtr_itr_f((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_simplex_lsq_fitters ( c C`skd}d}tt>tjddd}|||tjjt|}WdQXtjtj|tj t|gj }tj j |j ||j tj|j }tj |||jjdt|t|}tj j |j ||}tj||} t} | | ||} t| j|jjt|| jddS(u Tests that the 'param_cov' fit_info entry gets the right answer for *linear* least squares, where the answer is exact iidiiNu param_cov(R RKRRRLRMtlentmatrixtvstacktonestTRtinvtsumRVRWRtLinear1DR+RR%tfit_info( R RRRRtXtbetats2tolscovtmodRtfmod((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_param_covs *008 (R.R/R0R!RRRRR1R2t parametrizetfittersRRRR(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR/s      u not HAS_MOCKu not HAS_PKGtTestEntryPointcB`s_eZdZdZdZdZdZdZdZdZ dZ d Z RS( u4Tests population of fitting with entry point fitterscC`std|_dS(NuCThe test should not have gotten here. There was no exception thrown(t Exceptiontexception_not_thrown(R ((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR!scC`sdtfdY}|S(Nt goodclasscB`seZdZRS(u GoodClass(R.R/(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRs(tFitter(R R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytsuccessfulimportscC`s tdS(N(t ImportError(R ((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytraiseimporterrorscC`s d}|S(NcS`sdS(N((((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytbadfuncs((R R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt returnbadfuncs cC`sdtfdY}|S(NtbadclasscB`seZRS((R.R/(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRs(tobject(R R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytreturnbadclassscC`s5tjt}d|_|j|_t|gdS(uThis should work fineuWorkingN(Rtcreate_autospecRtnameRtloadR(R tmock_entry_working((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt test_workings  c C`stjtjdy5tjt}d|_|j|_t |gWn/t k r}d|j dkrwq|n X|j WdQXdS(uHThis raises an import error on load to test that it is handled correctlyuerroruIErru ImportErroriN( twarningstcatch_warningstfilterwarningsRRRRRRRRR\R(R tmock_entry_importerrortw((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_import_errors     c C`stjtjdy5tjt}d|_|j|_t |gWn/t k r}d|j dkrwq|n X|j WdQXdS(u2This returns a function which fails the type checkuerroruBadFuncuClassiN( RRRRRRRRRRRR\R(R tmock_entry_badfuncR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt test_bad_funcs     c C`stjtjdy5tjt}d|_|j|_t |gWn/t k r}d|j dkrwq|n X|j WdQXdS(u9This returns a class which doesn't inherient from fitter uerroruBadClassumodeling.FitteriN( RRRRRRRRRRRR\R(R tmock_entry_badclassR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_bad_class"s     ( R.R/R0R!RRRRRRRR(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRs        tTest1DFittingWithOutlierRemovalcB`seZdZdZRS(cC`sFtjddd|_d|_d}||j|j|_dS( Ngg@ig@g?g?cS`s1|dtjd||dd|ddS(Nigii(RRU(R'R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR:s(g@g?g?(RRRt model_paramsR(R R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR!6s  cC`sddlj}tjjd|jjdd|jj}|j tjj dd|jj|tjj dd|jj7_ t j dd d dd d }t ttd d dd}|||j|j \}}t|j|jddt ttd d dd}|||j|j \}}t|j|jddt ttd d dd}|||j|j \}}t|j|jdddS(Nig?Rgg?g@g@t amplitudeg?RDREtniteriRR]g?R=(t scipy.statststatsRRLtseedt bernoullitrvsRRRRRRFtFittingWithOutlierRemovalR+RRR%RRR(R Rtctg_inittfitRbRy((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt test_with_fitters_and_sigma_clip?s$"%(R.R/R!R(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR4s tTest2DFittingWithOutlierRemovalcB`s#eZdZdZdZRS(cC`sqtjddddddf\|_|_d |_d}||jtj|j|jg|_dS( Niiy`@g@g?gg?cS`sY|dtjd|d|dd|ddd|d|dd|ddS(Nigiig?ii(RRU(R'tpos((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt Gaussian_2D_s0(g@g?gg?g?(RRRRRR9R(R R((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR![s1  cC`sD|d}|d}||}||}tj|}tj||}tj||} |dj||djd|djd|dd} |dj||djd|djd|dd} tj|| |djdjt} tj| | |djdjt} || | }||| fS(Niig@(RRRtaroundtastypeRm(R tdataRRRtwxtwyttotal_intensitytx_meanty_meant x_to_pixelt y_to_pixeltx_posty_posR((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyt initial_guesses    ;;++c C`sddlj}tjjd|jjdd|jj}|jtjj dd|jj|tjj |jd|jj7_|j |jtj |j |j g}tjd|dd|d d |d d d dd}tttdddd}|||j |j |j\}}t|jdd!|jddtttdddd}|||j |j |j\}}t|jdd!|jddtttdddd}|||j |j |j\}}t|jdd!|jdddS(Nig?Rgg?g@RRiRitx_stddevg?ty_stddevg?RiRg@iR=g?(RRRRLRRRRRRRR9RRRt Gaussian2DRR+RRR%RRR(R RRtguesstg2_initRRbRy((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyR{s0"(* $ $$(R.R/R!RR(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyRYs t%TestWeightedFittingWithOutlierRemovalcB`s_eZdZdZdZdZdZdZdZdZ dZ d Z RS( u Issue #7020 cC`stjddddf\|_|_tj|j|jddd|_tj|j|jddd|_d|jduacci ig333333?iN(R+RRRRFRR8tpop( Rtslsqptsimplextkwargstsimplex_kwargsRRRRt simplex_modeltlm_model((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pyttest_fitters_interfaces   "  ((?R0t __future__RRRRtos.pathRfR1tnumpyRRt numpy.testingRRtRRtcoreR R tfittingtutilsR t utils.dataR RRRtutils.exceptionsRRRtscipyRRt HAS_SCIPYRRtHAS_MOCKtunittestRt pkg_resourcesRtHAS_PKGRRRRKRRR4R2R3RCRdRRRRR RR'(((sBlib/python2.7/site-packages/astropy/modeling/tests/test_fitters.pytsl"              59jW$B%L