[c@`sdZddlmZmZmZmZyddlZWnek rWddlZnXddl Z ddl Z ddl m Z mZddlmZmZddlmZmZdd lmZdd lmZd d lmZd d lmZd dlmZy&ddlZddlmZe Z!Wnek rAe"Z!nXe!oTeedZ#e j$j%ddddZ&dZ'dZ(dZ)de*fdYZ+de*fdYZ,e ddZ-e j$j.d/e/ej0dd d!e,fd"YZ1e j$j.d0e/ej0dd#d$e+fd%YZ2d&Z3d'Z4d(Z5d)Z6e j$j%d*d+Z7e j$j%d*d,Z8e j$j%d*d-Z9d.Z:dS(1uU Tests for model evaluation. Compare the results of some models with other programs. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tassert_allcloset assert_equali(t models_1Dt models_2Di(tfittingtmodels(t FittableModel(tPolynomialBasei(t minversion(tzip(tNumpyRNGContext(toptimizeu0.14u not HAS_SCIPYic C`sddd}ddd}tj|d|}tjddd}|}|j|dd }|j|dd }tjjd||tjjt |d } t j } | ||| } tj tj | jj| jjgtj ||gd kstdS( NiicS`s |tjdtj||S(u Model function i(tnptsintpi(txt amplitudet frequency((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt sine_model+scS`sbtjdtj||}dtj||tjdtj||}tj||fS(uv Jacobian of model function, e.g. derivative of the function with respect to the *parameters* i(RRRtcostvstack(RRRtdatdf((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt sine_deriv1s1t fit_derivii2g@g@g?gMbP?(R t custom_modelRtlinspacetevaluateRtrandomtseedtrandtlenRtLevMarLSQFittertalltarrayRtvalueRtAssertionError( RRRRt SineModelRt sin_modeltyty_primetdatatfittertmodel((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_custom_model(s  & $cC`s[tjddd}|dddd}|jdksBt|jdksWtdS(NiicS`s |tjdtj||S(uModel functioni(RRR(RRR((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyR)KsRg@Rg?(R RRR(R(R)R*((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_custom_model_initJs cC`stjddd}|}|jjdks9t|jjdksQt|jdksft|jdks{tdS(NiicS`s |tjdtj||S(uModel functioni(RRR(RRR((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyR)Ws(R RRtdefaultR(R(R)R*((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_custom_model_defaultsVs cC`sdddddddd}d tj|fd Y}|}|j}|\}}}tj|d\}}} tjt|d |ddt|d |ddt|d |ddf\} } } tjt|d ||d|dt|d ||d|dt|d | |d| df\} }}|||| }|| | | }t|j|j|jd kst d S(u+Test bounding box evaluation for a 3D modeli i iiiiic S`sF|||d|||d|||d} | dk| } | S(Nii(( RR+tztx0ty0tz0tatbtctamptrsqtval((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt ellipsoidhs2t Ellipsoid3DcB`seZedZRS(cS`s[|j|j|j|jf|j|j|j|jf|j|j|j|jffS(N(R7R:R6R9R5R8(tself((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt bounding_boxns(t__name__t __module__tpropertyRA(((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyR?msigHz>N( R RRARtdifftmgridtslicetabstsumR((R>R?R/tbboxtzlimtylimtxlimtdztdytdxtz1ty1tx1tz2ty2tx2tarrtsub_arr((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_custom_model_bounding_boxes  (& 0tFittable2DModelTestercB`sheZdZdZdZdZdZejj ddZ ejj ddZ RS(u Test class for all two dimensional parametric models. Test values have to be defined in example_models.py. It currently test the model with different input types, evaluates the model at different positions and assures that it gives the correct values. And tests if the model works with non-linear fitters. This can be used as a base class for user defined model testing. cC`sd|_d|_d|_d|_d|_d|_tjddd|_tjddd|_ tj ddddf\|_ |_ dS( Nidg-C6?g?g333333@g@ii i( tNtMt eval_errort fit_errorRR+RtarangeRSRRRFRURV(R@((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt setup_classs      cC`sLt||}||j|j||j|j||j|jdS(u&Test model with different input types.N(t create_modelRR+RSRRRVRU(R@t model_classttest_parametersR/((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_input2DscC`set||}|d}|d}|d}tjtj|||||jksatdS(u*Test model values add certain given pointsux_valuesuy_valuesuz_valuesN(RaRR%RHR]R((R@RbRcR/RR+R4((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_eval2Ds    cC`st||}d d f|_|jd d fks9td|_tjt |jWdQXtjtd |_WdQX|`y |j}Wntk rtjdnX|\}}t j |d\}}t j t |d|ddt |d|ddf\} } t j t |d||d|dt |d||d|df\} } || | } || | }t | j|j| jdkstdS(uTest bounding box evaluationiiNu&Bounding_box is not defined for model.iiigHz>(ii(ii(ii(ii(ii(RaRAR(tNonetpytesttraisestNotImplementedErrort ValueErrortskipRRERFRGRHRI(R@RbRcR/RJRLRMRORPRRRSRURVRWRX((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_bounding_box2Ds,     %&-u not HAS_SCIPYcC`sB|d}|d}|d}t||}t|tr_g|jD]}||^qF}nd|kr|drtj|d|d|j}tj|d|d|j} qn@tj|d|d|j}tj|d|d|j} tj|| \} } tj j dtj j |j|jd} || | d|d| } t j }||| | | }g|jD]}t||^q}g|D]}|j^q}tjgt||D]\}}|s|^q}tjg|D]}|js|j^q}t||d |jd S( u3Test if the parametric model works with the fitter.ux_limuy_limu parametersulog_fitiig?g?tatolN(Rat isinstancetdictt param_namesRtlogspaceR[RtmeshgridR R!R"RR$tgetattrtfixedR&R R'RR^(R@RbRctx_limty_limt parametersR/tnameRR+txvtyvtnoiseR-R.t new_modeltparamstparamRtR=texpectedtfitted((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_fitter2Ds4   #   &   %" cC`sj|d}|d}|jdkr3tjdnt|trRtjdnd|kr|drtj|d|d|j}tj|d|d|j }qn@tj |d|d|j}tj |d|d|j }tj ||\}}yUt ||dt d d } t ||dt d d } t ||dt d d } WnPtk rt ||dt } t ||dt } t ||dt } nXtjjd } |d d} d | | j|j |jd}| |||}tj}|| |||}tj}|| |||dt}t|j|jdd dS(up Test the derivative of a model by fitting with an estimated and analytical derivative. ux_limuy_limu-Derivative function is not defined for model.u'Skip testing derivative of polynomials.ulog_fitiituse_constraintst parameter_keyu deriv_initialiIu parametersg?g?testimate_jacobiantrtolN(RRfRgRkt issubclassR RRqR[R\RRrRatFalsetKeyErrorR t RandomStateR"RR$tTrueRRw(R@RbRcRuRvRR+RyRztmodel_with_derivtmodel_no_derivR/trsnRtnR-tfitter_with_derivtnew_model_with_derivtfitter_no_derivtnew_model_no_deriv((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_deriv_2DsV     &               $       ( RBRCt__doc__R`RdReRlRgtmarktskipifRR(((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyRZs   %&tFittable1DModelTestercB`sheZdZdZdZdZdZejj ddZ ejj ddZ RS(u Test class for all one dimensional parametric models. Test values have to be defined in example_models.py. It currently test the model with different input types, evaluates the model at different positions and assures that it gives the correct values. And tests if the model works with non-linear fitters. This can be used as a base class for user defined model testing. cC`sd|_d|_d|_d|_d|_d|_tjddd|_tjddd|_ tj ddddf\|_ |_ dS( Nidg-C6?g?g333333@g@ii i( R[R\R]R^RR+RR_RSRRRFRURV(R@((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyR`As      cC`s:t||}||j||j||jdS(u&Test model with different input types.N(RaRRSRV(R@RbRcR/((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_input1DLs  cC`sCt||}|d}|d}t|||d|jdS(u; Test model values at certain given points ux_valuesuy_valuesRmN(RaRR](R@RbRcR/RR+((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_eval1DTs  c C`sqt||}d |_d|_tjt |jWdQX|`tjtd|_WdQXy |j}Wntk rtjdnXt|t j rd}d}n d}d}t j |d }t j t|d |dd|}t j t|d ||d|d|} || } ||} t| j| j| j|ksmtdS( uTest bounding box evaluationiiNu&Bounding_box is not defined for model.g{Gz?g?gHz>iii(ii(RaRARfRgRhRiRjRkRnR t Lorentz1DRRERFRGRHRIR(( R@RbRcR/RJRtddxRPRSRVRWRX((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_bounding_box1D]s.      %-  u not HAS_SCIPYcC`s|d}|d}t||}t|trUg|jD]}||^q<}nd|kr|drtj|d|d|j}qn tj|d|d|j}tjj dd}d|tjj t |||} t j } | ||| } g| jD]}t| |^q} g| D]} | j^q;}tjgt||D]\}}|sf|^qf}tjg| D]} | js| j^q}t||d|jdS( uE Test if the parametric model works with the fitter. ux_limu parametersulog_fitiig{Gz?RmN(RaRnRoRpRRqR[RR R!trandnR#RR$RsRtR&R R'RR^(R@RbRcRuRwR/RxRtrelative_noise_amplitudeR-R.R|R}R~RtR=RR((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_fitter1Ds,  #  &   %"cC`s{|d}|jdkr)tjdnt|trHtjdnd|kr|drtj|d|d|j}qn tj |d|d|j}|d}t ||dt }t ||dt }tj j d }d |d|j|jd } ||| } tj} | ||| } tj} | ||| d t}t| j|jd ddS(uk Test the derivative of a model by comparing results with an estimated derivative. ux_limu-Derivative function is not defined for model.u'Skip testing derivative of polynomials.ulog_fitiiu parametersRiIg?g?RRmg333333?N(RRfRgRkRR RRqR[RRaRR RR"RR$RRRw(R@RbRcRuRRwRRRRR-RRRR((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyt test_deriv_1Ds0   &      "    ( RBRCRR`RRRRgRRRR(((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyR5s   '#u parameterscC`si}t|tr#|||St|trd|krUt rUtjdn|rwd|krw|d}qwn||||SdS(uCreate instance of model class.urequires_scipyuSciPy not foundu constraintsN(RR R t HAS_SCIPYRgRk(RbRcRRt constraints((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyRas u model_classutest_parameterstkeycC`st|dS(Ni(tstr(R((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pytstTestFittable1DModelscB`seZRS((RBRC(((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyRscC`st|dS(Ni(R(R((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyRstTestFittable2DModelscB`seZRS((RBRC(((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyRscC`stjd}|ddks'tt|ddgddgtjddgdd}t|dddgt|ddgdtddgdd ggdS( Ni*iiii+i,tn_modelstmodel_set_axisi-(R tShiftR(RR(tm((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_ShiftModelscC`stjd}|ddks'tt|ddgddgtjddgdd}t|dddgt|ddgdtddgdd ggdS( Ni*iiiiTi+RRiV(R tScaleR(RR(R((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_ScaleModelsc C`setjdddddddd}tjd dd }||}|d |jksatd S( u Currently just tests that the model peaks at its origin. Regression test for https://github.com/astropy/astropy/issues/3942 tx_0it amplitude_Li tfwhm_Lg?tfwhm_Gg?ig{Gz?iN(R tVoigt1DRR_tmaxR((RRR+((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_voigt_models$ cC`s1tjddd}t|dks-tdS(Ng?g@g @u1(R t Gaussian1DtreprR((R((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_model_instance_reprsunot HAS_SCIPY_14c C`s;tjdd}dddddg}tjd}|d |d |}d d d ddg}t||dddddg|d |}t||dddddgtjt!|d d d dddgWdQX|d |dtdd}t|d d d dddgddddddgdS(u Test Tabular1D model. iig?i ii-iitpointst lookup_tableggffffff?gffffff?g@g333333@g333333@g333333@g333333@g?gffffff@Nt bounds_errort fill_valueg333333( RR_R t tabular_modelRRgRhRjRRf(Rtvaluest LookupTableR/txnew((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_tabular_interp_1ds""" c C`stjdddddgdddd d gd d d ddgdddddgdddddgg}tjddtjddf}tjddddd g}tjd!}|||}|||}tjdd"d#d$d%g}t||d&d1tjd)jd2}tjdd*dd+f\}} tjd,|} | || }t||t j t /|d-d.d/gd.d0gd*d+gf}WdQXdS(3Ngr:g%Yɿ}g3{`yga.e?g,ypw?g~O2 g|bLg+冿g?g쀐Ѧ?g8gRrwkӷg&2gI+;?g9?c?g`&rg8Qg- rg©ja?g#ɻ?g廙g{Bg>jYg?gd?iiggffffff?gffffff?g@g333333@iggngBog|g>%ړ?Rmi ii iiRRg333333?gffffff@g@gHz>(ii( RR&R_R RRtreshapeRFt Tabular2DRgRhRj( ttableRRRR/tznewtresultR8R+Rtt((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_tabular_interp_2d.s*$% cC`stjdjd}tjddddddf\}}}tjd}|d|}||||}t||dS(NiiiiR(iii(RR_RRFR RR(R8RR+R4ttabRR((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_tabular_ndLs 1c C`stjdtjd@}tjd|B}tdtjjd|_WdQXddf|_tj dd dd f\}}|||\}}|||d t \}}tj |j }t ||||t |||||||d t d d \}}|d kj }t ||||t ||||tjddddd}d|_|d|dd t kstdS(uM Test the option to evaluate a model respecting its bunding_box. iiiii Nii ii twith_bounding_boxRitc0tc1gffffff@i(iiii(ii (ii(ii(R t Polynomial2DtMappingRRR R"RwRARFRtisnantnonzeroRt Polynomial1DR(( tpRRR+R8R9tawtbwtind((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pyttest_with_bounding_boxVs$ %! (u model_classutest_parameters(u model_classutest_parameters(;Rt __future__RRRRtcPickletpicklet ImportErrorRgtnumpyRt numpy.testingRRtexample_modelsRRtRR tcoreR t polynomialR tutilsR textern.six.movesR RtscipyRRRRt HAS_SCIPY_14RRR0R1R3RYtobjectRZRRat parametrizetsortedtitemsRRRRRRRRRR(((sAlib/python2.7/site-packages/astropy/modeling/tests/test_models.pytsZ"       !  "