[c@`sGddlmZmZmZmZddlZddlZddlZddl m Z ddl m Z ddl mZmZmZmZmZmZddlmZmZdd lmZyddlZeZWnek reZnXd d d gZeeegZeeegZ ej!j"de#ej$eedZ%ej!j"dedZ&ej!j"de#ej$e edZ'ej!j"dedZ(ej!j)ddZ*ej!j)ddZ+ej!j)ddZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dS( i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tassert_allclosei(tdiscretize_modeli(t Gaussian1DtBox1Dt MexicanHat1Dt Gaussian2DtBox2Dt MexicanHat2D(t models_1Dt models_2D(t create_modelucenteru linear_interpu oversampleu model_classumodecC`s|tkr(|dkr(tjdnt|}t||}t|t|dd|}t|jt|ddddS( uK Test if the sum of all pixels corresponds nearly to the integral. ucenteru)Non integrating mode. Skip integral test.ux_limtmodeuintegraltatolg-C6?N(RtpytesttskipR RRRtsum(t model_classRt parameterstmodeltvalues((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_pixel_sum_1Ds  cC`s\tddd}tjdd}||}t|d d|}t||ddd S( ut Discretize Gaussian with different modes and check if result is at least similar to Gaussian1D.eval(). iiiiieRRgMbP?N(iie(RtnptarangeRR(RRtxRt disc_values((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_gaussian_eval_1D+s  cC`s|tkr(|dkr(tjdnt|}t||}t|t|dt|dd|}t|jt|dddd S( uK Test if the sum of all pixels corresponds nearly to the integral. ucenteru)Non integrating mode. Skip integral test.ux_limuy_limRuintegralRg-C6?N(R RRR RRRR(RRRRR((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_pixel_sum_2D8s cC`stddddd}tjdd}tjdd}tj||\}}|||}t|d d d|}t||dddS( us Discretize Gaussian with different modes and check if result is at least similar to Gaussian2D.eval() g{Gz?iiiiRRN(ii(ii(R RRtmeshgridRR(RRRtyRR((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_gaussian_eval_2DHsu not HAS_SCIPYcC`stdddddtdddddtdddddg}tjdd}tjdd}tj||\}}xH|D]@}|||}t|d d dd}t||d dqWd S( u1 Discretize Gaussian with integrate mode g{Gz?iiiiiRu integrateRN(ii(ii(R RRRRR(t model_listRR RRR((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyt$test_gaussian_eval_2D_integrate_modeZs cC`s^tddd}t|d dddd }t|jtjdtjdd d d S(uN Test subpixel accuracy of the integrate mode with gaussian 1D model. iig?iiRu integratetfactoridRgh㈵>N(ii(RRRRRtsqrttpi(tgauss_1DR((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_subpixel_gauss_1DnscC`s^tddddd}t|dddddd }t|jdtjd d d d S(uN Test subpixel accuracy of the integrate mode with gaussian 2D model. iig?iiRu integrateR$idg{Gz?Rgh㈵>N(ii(ii(R RRRRR&(tgauss_2DR((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_subpixel_gauss_2DxscC`s9d}t|d}t|tjddddS(u7 Test discretize when a 1d function is passed. cS`s|dS(Ni((R((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pytfsiiiN(ii(RRRR(R+R ((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_discretize_callable_1ds cC`sWd}t|dd }tjd d\}}|d|d}t||dS( u7 Test discretize when a 2d function is passed. cS`s|d|dS(Ni((RR ((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyR+siii iiN(ii(ii(i i (RRtindicesR(R+tactualR Rtdesired((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_discretize_callable_2ds  cC`sKtjt}ttddWdQX|jjddksGtdS(u Test type exception. iii NuModel must be callable.(ii (Rtraisest TypeErrorRtfloattvaluetargstAssertionError(texc((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_type_exceptionsc C`sQd}tjt}t|ddWdQX|jjddksMtdS( u& Test dimension exception 1d. cS`s|dS(Ni((R((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyR+sii Niu)y range specified, but model is only 1-d.(ii (ii (RR1t ValueErrorRR4R5R6(R+R7((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_dim_exception_1ds cC`sNd}tjt}t|dWdQX|jjddksJtdS(u& Test dimension exception 2d. cS`s|d|dS(Ni((RR ((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyR+sii Niu'y range not specified, but model is 2-d(ii (RR1R9RR4R5R6(R+R7((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_dim_exception_2ds cC`sNd}tjt}t|dWdQX|jjddksJtdS(NcS`s|d|dS(Ni((RR ((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyR+sg/$$g(\u&@iuTThe difference between the upper an lower limit of 'x_range' must be a whole number.(g/$$g(\u&@(RR1R9RR4R5R6(R+R7((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_float_x_range_exceptions c C`sQd}tjt}t|dd WdQX|jjddksMtdS( NcS`s|d|dS(Ni((RR ((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyR+sii g/$$g(\u&@iuTThe difference between the upper an lower limit of 'y_range' must be a whole number.(ii (g/$$g(\u&@(RR1R9RR4R5R6(R+R7((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyttest_float_y_range_exceptions (u model_classumode(u model_classumode(4t __future__RRRRt itertoolsRtnumpyRt numpy.testingRtutilsRtmodeling.functional_modelsRRRR R R tmodeling.tests.example_modelsR R tmodeling.tests.test_modelsRtscipytTruet HAS_SCIPYt ImportErrortFalsetmodesttest_models_1Dttest_models_2Dtmarkt parametrizetlisttproductRRRR!tskipifR#R(R*R,R0R8R:R;R<R=(((sHlib/python2.7/site-packages/astropy/convolution/tests/test_discretize.pyts<"   .    0 0