[c@`sOdZddlmZmZmZmZddlZddlZddl m Z ddl m Z ddl m Z ddlmZmZmZdd lmZd d lmZydd lmZeZWnek reZnXe jdgfe jdgfe jdgfe jdgfe j dgfgZ!e j"dgfe j#d dgfe j$d dgfgZ%de&fdYZ'de&fdYZ(de&fdYZ)defdYZ*de&fdYZ+de&fdYZ,defdYZ-de&fdYZ.defdYZ/d Z0d!Z1d"Z2dS(#uD This module tests fitting and model evaluation with various inputs i(tabsolute_importtunicode_literalstdivisiontprint_functionN(tassert_allclosei(tmodels(tfitting(tModelt FittableModeltFittable1DModel(t Parameteri(trange(toptimizeit TestInputTypecB`sSeZdZdZejjdedZejjde dZ RS(u This class tests that models accept numbers, lists and arrays. Add new models to one of the lists above to test for this. cC`sqd|_d|_tjddd|_tjddd|_tjddddf\|_|_dS(Ng333333@g@ii g?i( txtytnptarangetx1ty1tmgridty2tx2(tself((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyt setup_class0s   umodeluparamscC`s7||}||j||j||jdS(N(RRR(Rtmodeltparamstm((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyt test_input1D7s   cC`sI||}||j|j||j|j||j|jdS(N(RRRRRR(RRRR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyt test_input2D>s (umodeluparams(umodeluparams( t__name__t __module__t__doc__Rtpytesttmarkt parametrizetmodel1d_paramsRtmodel2d_paramsR(((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyR )s t TestFittingcB`seZdZdZdZdZdZdZdZdZ dZ e j j d d Ze j j d d Ze j j d d Ze j j d d ZRS(u/Test various input options to fitting routines.cC`sAtjd|_tjddddf\|_|_dS(Ni (RRRRRR(R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRIscC`stjddddg}tjd}ddddg|_||j}tj}|||j|}t|j|dddS( u1 set 1D x, 1psetiiitatoli iNgHz>( RtarrayRt Polynomial1Dt parametersRRtLinearLSQFitterR(Rtexpectedtp1RtpfitR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_linear_fitter_1setMs cC`stjddgddgddgddgg}tjddd}ddddddd d g|_i}x+td D]}||g||j|( RR(RR)R*R t param_namesRtFalseRR+Rt param_sets(RR,R-RtiRR.R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_linear_fitter_NsetXs3! cC`stjddddggj}tjd}ddddg|_||j}tj}|||j|}t |j |d d d S(u71 pset, 1 set 1D x, 1 set 1D y, Chebyshev 1D polynomialg@g@gC@gq@iiiiR'i iNg{Gz?( RR(tTRt Chebyshev1DR*RRR+RR4(RR,tch1RR.R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_linear_fitter_1dchebgs cC`stjddddggj}tjd}ddddg|_||j}tj}|||j|}t |j |d d d S(uH 1 pset, 1 set 1D x, 1 set 1D y, Legendre 1D polynomial g@g@gm@gv@iiiiR'i iNg-q=( RR(R7Rt Legendre1DR*RRR+RR4(RR,tleg1RR.R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_linear_fitter_1dlegendvs cC`stjd}ddddddg|_ddddddg}||j|j}tj}|||j|j|}t|j|dd t||j|j|dd dS( NiiiiiiR'i ig-q=g-q=(Rt Polynomial2DR*RRRR+R(Rtp2R,tzR.R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_linear_fitter_1set2ds cC`sntjtYtjd}||j}tjddd}tj}|||j|}WdQXdS(u A ValueError is raised if a 1 data set (1d x, 1d y) is fit with a model with multiple parameter sets. iR0iN(R!traisest ValueErrorRR)RRR+(RR-RR.R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_wrong_numpsets  cC`stjddgddgddgddgddgddgg}tjddd}i}x+td D]}d|g||j|( RR(RR)R R2RR3RR+RR4(RR,R-RR5RR.R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_wrong_psets     u not HAS_SCIPYcC`shtjddddd}||j}tj}|||j|}t|jdddgdS(u.1 set 1D x, 1 set 1D y, 1 pset NonLinearFitteri tmeanitstddevg?N(Rt Gaussian1DRRtLevMarLSQFitterRR*(Rtg1RtgfitR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_nonlinear_lsqt_1set_1ds  c C`stjtntjddgdddgdddgd d }||jd t}tj}|||j|}Wd QXd S( u51 set 1D x, 1 set 1D y, 2 param_sets, NonLinearFittergffffff$@i RFig @RGgq= ףp?g?R0iR1N( R!RBRCRRHRR3RRI(RRJRRKR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_nonlinear_lsqt_Nset_1ds *  c C`stjddddddddd d d }||j|j}tj}|||j|j|}t|jddddd d gd S( u.1 set 2d x, 1set 2D y, 1 pset, NonLinearFitteri tx_meanity_meanitx_stddevg333333?ty_stddevg?tthetaiN(Rt Gaussian2DRRRRIRR*(Rtg2R@RKR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_nonlinear_lsqt_1set_2ds  cC`stjttjddgddgddgdddgdddgdd d gd d }||jj|jj}tj }|||j|j|}Wd QXd S( u41 set 2d x, 1set 2D y, 2 param_sets, NonLinearFitteri iiRPg333333?RQg?RRiR0iN( R!RBRCRRSRtflattenRRRI(RRTR@RKR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_nonlinear_lsqt_Nset_2ds 0!! (RRR RR/R6R:R=RARDRER!R"tskipifRLRMRURW(((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyR&Fs        tTestEvaluationcB`s_eZdZdZdZdZdZdZdZdZ dZ d Z RS( us Test various input options to model evaluation TestFitting actually covers evaluation of polynomials cC`sAtjd|_tjddddf\|_|_dS(Nii (RRRRRR(R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRscC`stjddgddgddgdd}||jdt}tj|dddf|d ddfjtjgkstdS( uY This case covers: N param sets , 1 set 1D x --> N 1D y data i ig?R0iR1iNi( RRHRR3RtalltnonzeroR(tAssertionError(RRJR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_non_linear_NYsets-cC`stjddgddgddgdd}tj|j|jg}||}t|dddf|dddfd d dS( uU This case covers: N param sets , N sets 1D x --> N N sets 1D y data i ig?R0iNiiR'ig-q=(RRHRR(RR(RRJtxxR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_non_linear_NXYsets- cC`s7tjd}||j}|jdks3tdS(u 1 data set, 1 pset, Polynomial1DiiN(i(RR)RtshapeR\(RR-R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_p1_1set_1psetscC`stjddd}|tj|j|jgjdd}|jd ksTtt|dddf|d ddfd ddS(u'N data sets, N param_sets, Polynomial1DiR0iR1iiiNiR'i i(iig-q=( RR)RR(RR7R`R\R(RR-R((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_p1_nset_npsets*cC`s=tjd}||j|j}|jdks9tdS(u#1 pset, 1 2D data set, Polynomial2Dii N(i i (RR>RRR`R\(RR?R@((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_p2_1set_1pset scC`sstjddd}tj|j|jg}tj|j|jg}|||}|jdksotdS(u$N param_sets, N 2D data sets, Poly2diR0ii N(ii i (RR>RR(RRR`R\(RR?R^tyyR@((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_p2_nset_npsets cC`stj|j|jgj}d|deZdZdZeZeZeZedZ RS(uxuyuzcC`s|||||||fS(N((RR-R?tp3((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRl{s(ux(uyuz( RRtinputstoutputsR R-R?RRmRl(((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRss    t&TestSingleInputDoubleOutputSingleModelcB`sVeZdZdZdZdZdZdZdZdZ dZ RS( u A suite of tests to check various cases of parameter and input combinations on models with n_input = 1 but n_output = 2 on a toy model with n_models=1. As of writing there are not enough controls to adjust how outputs from such a model should be formatted (currently the shapes of outputs are assumed to be directly associated with the shapes of corresponding inputs when n_inputs == n_outputs). For now, the approach taken for cases like this is to assume all outputs should have the same format. cC`stddd}|d\}}t|ts9tt|tsNttj|tj|kowdknst|dkst|dkstdS( uO Scalar parameters with a scalar input should return a scalar. ii iidiioiWN(RRoRpR\RRq(RRrRR@((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRss4cC`stddd}|tjdd\}}t|tjsItt|tjsattj|tj|kod knsttj|dddd d gksttj||dkstd S( u Scalar parameters should broadcast with an array input to result in an array output of the same shape as the input. ii iiidi ioii7iN(i(RRRRoRtR\R`RZ(RRrRR@((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRus4*cC`stddd}|tjdjddd\}}t|tjsUtt|tjsmttj|tj|kodknsttj|dd d gd d d ggksttj||dkstdS(u Scalar parameters should broadcast with an array input to result in an array output of the same shape as the input. ii iiiiidi ioii7iiN(ii( RRRRvRoRtR\R`RZ(RRrRR@((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRws+4cC`stddd}|tjdjdddd\}}t|tjsXtt|tjspttj|tj|kodknsttj|dd gd d gd d ggddgddgddgggksttj||dkstdS(u Scalar parameters should broadcast with an array input to result in an array output of the same shape as the input. ii ii iiidi ioii7iiicii+iiiWN(iii( RRRRvRoRtR\R`RZ(RRrRR@((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRxs.4'0cC`stddgddgddg}|d\}}t|tjsNtt|tjsfttj|tj|kod knsttj|dd gksttj|d d gkstd S(u Array parameters should all be broadcastable with each other, and with a scalar input the output should be broadcast to the maximum dimensions of the parameters. iii iiiidioiziWiJN(i(RRoRRtR\R`RZ(RRrRR@((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRys$4!cC`stddgddgddg}|ddg\}}tj|tj|koedknspttj|d d gksttj|d d gkst|dgdgg\}}tj|tj|kodknsttj|d d gdd ggks1ttj|d dgdd ggks^ttjt |dddg\}}WdQXdS(ue When given an array input it must be broadcastable with all the parameters. iii iiiidiioiiWiiziiJii,N(i(ii(RRR`R\RZR!RBRC(RRrRtz1Rtz2Rztz3((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyR{s$4!!4--c C`stddgddggddgddggd d gd d gg}|d dgddgg\}}tj|tj|kod4knsttj|ddgddggksttj|ddgddggkst|d ggdgggdggdgggg\}}tj|tj|ko_d5knsjttj|ddgddggddgddgggdd gdd!ggd"d#gd$dggggksttj|dd%gd&d'ggd(dgd)d*gggd+d,gdd-ggd.d/gd0dggggksTttjt/|d ddgdd1d2gg\}}Wd3QXd3S(6ue When given an array input it must be broadcastable with all the parameters. iiiii iii(iii iidii,iioiiMiiWii i\iziiiiii7iBiXiiiiJi= i0ii iii iiiv ii iiXN(ii(iiii(RRR`R\RZR!RBRC(RRrRRRRRzR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyR|s$-$4--<4**c C`s tdddgdddggddd gd d d gggd ddgdddg}|dddg\}}tj|tj|kod_knstt|dddgdddggdddgdd d!gggt|d"d#d$gd%d&d'ggd(d)d*gd+d,d-ggg|dgggdgggdgggg\}}tj|tj|kod`knstt|dd.d/gdd0d1ggdd2d3gdd4d5gggd6dd7gd8dd9ggd:dd;gd<d d=gggd>d?dgd@dAdggdBdCdgdDdEd!ggggt|d"dFdGgd%dHdIggd(dJdKgd+dLdMgggdNd#dOgdPd&dQggdRd)dSgdTd,dUgggdVdWd$gdXdYd'ggdZd[d*gd\d]d-ggggd^S(aue When given an array input it must be broadcastable with all the parameters. g{Gz?g{Gz?gQ?g{Gz?g?gQ?gQ?g{Gz?g ףp= ?g?g)\(?gQ?iiiidii,i iigQ&@gQ6@gp= ׃@@gGz&@g 6@gHz@@gp= #&@gGz6@gQ@@g333333&@g\(6@g(\@@gq= ף[@gq= ףk@gGzt@g(\[@gk@g)\t@gGz[@g(\k@g= ףpt@gfffff[@gQk@gRt@g ףp= (@g(\*@g(@gQ*@g)\((@gGz.*@gQ8(@g= ףp=*@g(\5@gHz7@g ףp= 5@g(\7@gR5@gףp= 7@g5@gQ7@g(\?@g(\@@g ףp= ?@gfffff@@gR?@g ףp= @@g?@gGz@@gq= ףj@gGzs@gj@g)\s@g(\j@g= ףps@gQj@gRs@gq= ף@^@gGz0t@g(\B^@g)\0t@gGzD^@g= ףp1t@gfffffF^@gR1t@gQ``@gq= ףm@gzGa`@gm@g ףp=b`@g(\m@g33333c`@gQm@N(iii(iiii(RRR`R\R(RRrRRRR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyR}sJ4!" 64          ( RRR RsRuRwRxRyR{R|R}(((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRs       tTInputFormattercB`s)eZdZdZdZedZRS(u6 A toy model to test input/output formatting. uxuycC`s ||fS(N((RR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyRl>s(uxuy(uxuy(RRR RRRmRl(((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyR6scC`s.t}|dd}|dks*tdS(Nii(ii(RR\(Rtresult((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_format_input_scalarsCs cC`sYt}|ddgddg}t|tjddgtjddgfdS(Nii(RRRR((RR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyttest_format_input_arraysIs cC`s\t}tjddggjtjddggjf}||}t||dS(Nii(RRR(R7R(RtinputR((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pyt#test_format_input_arrays_transposedOs 6 (3R t __future__RRRRR!tnumpyRt numpy.testingRtRRtcoreRRR R*R textern.six.movesR tscipyR tTruet HAS_SCIPYt ImportErrorR3R)R;R8tShifttScaleR$R>t Legendre2Dt Chebyshev2DR%tobjectR R&RYRkRnR~RRRRRR(((s@lib/python2.7/site-packages/astropy/modeling/tests/test_input.pytsD"     W