ó šßÈ[c@`sŽdZddlmZmZmZmZddlZddlZddlZ ddl m Z m Z m Z mZddlmZddlmZmZdd lmZmZdd lmZmZd d lmZd „Zd„Zdefd„ƒYZdefd„ƒYZdefd„ƒYZd„Z d„Z!de"fd„ƒYZ#de"fd„ƒYZ$de"fd„ƒYZ%d„Z&d„Z'dS(u Tests models.parameters i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tassert_allcloset assert_equaltassert_array_equaltassert_almost_equali(tirafutili(tmodelstfitting(tModelt FittableModel(t ParametertInputParameterErrori(tget_pkg_data_filenamecC`s|S(N((tval((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pytsetter1scC`s|j|ƒ||jS(N(t do_somethingtp(Rtmodel((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pytsetter2s t SetterModelcB`sYeZd Zd ZedddeƒZedddeƒZd„Z d„Z d„Z RS( uxuyuztdefaultitsettercC`s2||_tt|ƒjƒ||_||_dS(N(RtsuperRt__init__txctyc(tselfRRR((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR(s  cC`s||d||dS(Ni((RtxtyRR((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pytevaluate.scC`sdS(N((Rtv((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR1s(uxuy(uz( t__name__t __module__tinputstoutputsR RRRRRR R(((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR s  t TParModelcB`s8eZdZeƒZeƒZd„Zed„ƒZRS(u2 A toy model to test parameters machinery cK`s&tt|ƒjd|d||dS(Ntcoeffte(RR&R(RR'R(tkwargs((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR=scC`sdS(N((R'R(((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR @s( R"R#t__doc__R R'R(Rt staticmethodR (((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR&5s    t MockModelcB`s,eZeddddƒZed„ƒZRS(tnameualphaRi*cG`sdS(N((targs((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR Hs(R"R#R talphaR+R (((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR,EscC`s@tƒ}|j}|jdks't‚tjtƒd|_WdQX|jtks[t‚t |_|jt ksyt‚|j tksŽt‚d„|_ t|_ |j tks¸t‚|j dksÍt‚d|_ |j dksët‚d|_ |j dks t‚|j dkst‚d|_ |j dks<t‚dS(u8Test if getting / setting of Parameter properties works.ualphaubetaNcS`sdS(Ni((t_((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyt^si*i)(R,R/R-tAssertionErrortpytesttraisestAttributeErrortfixedtFalsetTruettiedtmintNonetmax(tmR((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_parameter_propertiesMs(        cC`sJtƒ}|j}d}d}||||ks8t‚||||ksRt‚||||kslt‚||||ks†t‚||||ks t‚||||ksºt‚|dksÌt‚|dksÞt‚||ksðt‚||kst‚||kst‚| | ks(t‚t|ƒt|ƒksFt‚dS(u0Test if the parameter arithmetic operators work.gE@ii-i)N(R,R/R2tabs(R=tpartnumR((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_parameter_operatorsos"  tTestParameterscB`s¼eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„Zd„Zd„Zd„ZRS(cC`sìtdƒ}t|ƒ}|jƒ}|jdƒ}|jƒtj|dƒ}|j|_t |j dƒ}t j |dƒ|_ t jddddd ƒ|_tjƒ|_|j|_|j|_tj|j|jgƒ|_d S( ué Unit tests for parameters Read an iraf database file created by onedspec.identify. Use the information to create a 1D Chebyshev model and perform the same fit. Create also a gausian model. udata/idcompspec.fitsubeginiuorderitmeanitstddeviN(RtopentreadtsplittcloseRtIdentifyRecordR'ticoefftinttfieldsR t Chebyshev1DRt Gaussian1DtgmodelR tLinearLSQFittert linear_fitterRtzRtnptarraytyy(Rt test_filetftlinestreclisttrecordtorder((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyt setup_classˆs       cC`sVtjdddddgƒ|jj(|jjdddd d gkjƒsRt‚d S( ut Tests updating the parameters attribute with a slice. This is what fitters internally do. iiiiig@g@g@g@g@N(RTRURt parameterstallR2(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_set_slice s%cC`sLdddddg|j_|jjdddddgkjƒsHt‚d S( u'Tests updating parameters using a list.ii(i2i<iFg>@gD@gI@N(RR^R_R2(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_set_parameters_as_listªscC`sUtjdddddgƒ|j_|jjdddd d gkjƒsQt‚d S( u)Tests updating parameters using an array.iiiiig@g@g@g@g@N(RTRURR^R_R2(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_set_parameters_as_array°s$cC`s=d|j_|jjdddddgkjƒs9t‚dS(u(Tests updating parameters using a tuple.iiiiiN(iiiii(RR^R_R2(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_set_as_tuple¶s cC`sXdddddg|j_d|j_|jjdddddgkjƒsTt‚dS(u{ Tests updating the parameters attribute when a model's parameter (in this case coeff) is updated. igiN(RR^tc0R_R2(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_set_model_attr_seq¼s cC`s7d|j_|jjdddgkjƒs3t‚dS(u>Update the parameter list when a model's parameter is updated.iiiN(RPt amplitudeR^R_R2(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_set_model_attr_numÆs cC`stdddddg|j_d|jjd<|jjdddddgkjƒsXt‚|jjdkspt‚d S( u%Update the parameters using indexing.iiiiig$@ii N(RR^R_R2Rd(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyt test_set_itemÌs-cC`s/tjtƒdddg|j_WdQXdS(uz Tests raising an error when attempting to reset the parameters using a list of a different size. iiiN(R3R4RRR^(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_wrong_size1ÔscC`s/tjtƒdddg|j_WdQXdS(uš Tests raising an exception when attempting to update a model's parameter (in this case coeff) with a sequence of the wrong size. iiiN(R3R4RRRd(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_wrong_size2ÝscC`s,tjtƒddg|j_WdQXdS(u‰ Tests raising an exception when attempting to update a model's parameter and the new value has the wrong shape. iiN(R3R4RRPRf(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_wrong_shapeæscC`sc|j|j|j|jƒ}t|j|jƒt|jtjdddddgƒdd ƒd S( u[ Test the fitter modifies model.parameters. Uses an iraf example. g`‡NÚ²@gŸ†g±'Ç@g¼$:PH)@g‘ˆi#6¨ü¿gÿæg+„áì?trtoli iþÿÿÿNg{®Gáz„?( RRRRRtprintRR^RTRU(Rt new_model((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_par_against_irafïs  cC`sTidd6dd6dd6dd6}tjd |}t|jddddgƒdS( Ni uc0i uc1i uc2iuc3i(R t Polynomial1DRR^(Rtdtp1((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttestPolynomial1Ds"cC`s¬tjdddƒ}t|jddddddddddddg ƒt|jdddgƒdddg|_t|jddddddddddddg ƒdS(Nitn_modelsgii g$@(R RpRR^RRd(RRr((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_poly1d_multiple_setsscC`s`tjdddƒ}ddg|jd*t|jddddddddddddg ƒdS( u5 Test assigning to a parameter slice iRti ig$@giN(R RpRdRR^(RRr((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_par_slicingsc C`sMtjddƒ}d|_t|jddddddddddg ƒdS(Ntdegreeiii(R t Polynomial2Dtc0_0RR^(Rtp2((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyt test_poly2ds cC`sžiddgd6ddgd6ddgd6ddgd 6ddgd 6ddgd 6}tjd|}t|jddddddddddddg ƒdS( Niiuc0_0iuc1_0iiuc2_0uc0_1uc0_2uc1_1(R RxRR^(RtkwRz((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_poly2d_multiple_setss **cC`sItjdƒ}d|_|jdks-t‚|jjdksEt‚dS(Nii(R tShifttoffsetR2tvalue(Rtsh1((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_shift_model_parameters1d#s cC`setjddgƒ}ddg|_tj|jddgkƒsHt‚t|jjddgƒdS(Nii(R tScaletfactorRTR_R2RR€(Rtsc1((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_scale_model_parametersnd)s$cC`s8tjdƒ}tjtƒddg|_WdQXdS(Nii(R R~R3R4RR(RR((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_parameters_wrong_shape/s(R"R#R]R`RaRbRcReRgRhRiRjRkRoRsRuRvR{R}R‚R†R‡(((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyRC†s(            tTestMultipleParameterSetscB`s,eZd„Zd„Zd„Zd„ZRS(cC`s›tjdddƒ|_tjdd…dd…f\|_|_tj|j|jgƒj|_t j ddgddgdd d gd d ƒ|_ dS( Nii gš™™™™™¹?ii g @gÍÌÌÌÌÌ@REgš™™™™™Ù?gffffffæ?Rti( RTtarangetx1tmgridRRRUtTtx11R RORP(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyR]7s +!'cC`sxddg|j_t|jjtjddgddgddggƒƒtj|jjddddddgkƒd S( uX Test that a change to one parameter as a set propagates to param_sets. ii gð?g @gÍÌÌÌÌÌ@gš™™™™™Ù?gffffffæ?g$@N(RPRfRt param_setsRTRUR_R^(R((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyttest_change_par>s  cC`svd|jjdRBtobjectRCRˆR’R¹R¿(((sElib/python2.7/site-packages/astropy/modeling/tests/test_parameters.pyts,"   "   " ¯.î