[c @`sOddlmZmZmZmZddlZddlZddlZddl m Z m Z ddl m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlm Z dd l!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'ydd l(m)Z)e*Z+Wne,k rQe-Z+nXdd d dgZ.ddddgZ/ddddgZ0eeeeeeeeeeg Z1ddej2dej3dej4dgZ5ej6dZ7de7ddS(i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tassert_almost_equaltassert_allclosei(tconvolvet convolve_fft(tGaussian1DKerneltGaussian2DKernelt Box1DKernelt Box2DKerneltTrapezoid1DKerneltTrapezoidDisk2DKerneltMexicanHat1DKerneltTophat2DKerneltMexicanHat2DKerneltAiryDisk2DKernelt Ring2DKernelt CustomKernelt Model1DKernelt Model2DKerneltKernel1DtKernel2D(tKernelSizeErrori(tBox2Dt Gaussian1Dt Gaussian2D(tAstropyWarningtAstropyUserWarning(tfiltersiii iiiucenteru linear_interpu oversampleu integrateig?iQi(i=t TestKernelscB`seZdZejjdejjdedZejjdejjdedZ ejjd-e e j e edZejjd.e e j e edZejjdedZejjded Zd Zd Zejjd ed Zejjd edZejjd edZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'd Z(d!Z)d"Z*ejjd/e e j e e+d$Z,ejjde-d%Z.d&Z/d'Z0d(Z1d)Z2d*Z3d+Z4d,Z5RS(0u: Test class for the built-in convolution kernels. u not HAS_SCIPYuwidthcC`st|}|jt|}|jtt|dd}tt|dd}tjt|}tjt|}t||ddt||dddS(uL Test GaussianKernel against SciPy ndimage gaussian filter. tboundaryufilltdecimali N( Rt normalizeR Rtdelta_pulse_1Dtdelta_pulse_2DRtgaussian_filterR(tselftwidthtgauss_kernel_1Dtgauss_kernel_2Dt astropy_1Dt astropy_2Dtscipy_1Dtscipy_2D((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_scipy_filter_gaussian8s    c C`s<t|}t|}tt|dddt}tt|dddt}tjt"}tt|dddt }WdQXd|j j dkst tjt"}tt|dddt }WdQXd|j j dkst t jt| }t jt| }t||ddt||dddS( uY Test MexicanHat kernels against SciPy ndimage gaussian laplace filters. R ufilltnormalize_kernelNusum is close to zeroiR!i(RRRR#tFalseR$tpytesttraisest ExceptiontTruetvaluetargstAssertionErrorRtgaussian_laplaceR( R&R'tmexican_kernel_1Dtmexican_kernel_2DR*R+texcR,R-((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyt"test_scipy_filter_gaussian_laplaceLs  !!u kernel_typecC`s|tkr#t r#tjdn|tks>||}n|||d}|jdkrtt|dddt}t t|dddt}t ||ddnItt |dddt}t t |dddt}t ||ddd S( uI Test smoothing of an image with a single positive pixel u/Omitting AiryDisk2DKernel, which requires SciPyg?iR ufillR/R!i N( Rt HAS_SCIPYR1tskipRt dimensionRR#R0RRR$(R&t kernel_typeR'tkerneltc1tc2((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_delta_datajs cC`s|tkr#t r#tjdn|tks>||}n|||d}|jdkrtt|dddt}t t|dddt}t ||ddnItt |dddt}t t |dddt}t ||ddd S( uA Test smoothing of an image made of random noise u/Omitting AiryDisk2DKernel, which requires SciPyg?iR ufillR/R!i N( RR=R1R>RR?Rtrandom_data_1DR0RRtrandom_data_2D(R&R@R'RARBRC((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_random_datas cC`sVtj||g}tt|dd}tt|dd}t||dddS(u Test smoothing of an image with a single positive pixel Instead of using kernel class, uses a simple, small kernel R ufillR!i N(tnptonesRR$RR(R&R'RARCRB((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_uniform_smallkernelscC`sjtj||g|d}t|}tt|dd}tt|dd}t||dddS(uI Test smoothing of an image with a single positive pixel iR ufillR!i N(RHRIR RR$R(R&R'tkernel1tkernel2RCRB((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_smallkernel_vs_Box2DKernels  cC`sbtd}td}td}t||}tjtj||jdks^tdS(uR Check if convolving two kernels with each other works correctly. iiig{Gz?N(RRRHtalltabstarrayR7(R&tgauss_1tgauss_2t test_gauss_3tgauss_3((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_convolve_1D_kernelss    cC`sbtd}td}td}t||}tjtj||jdks^tdS(uR Check if convolving two kernels with each other works correctly. iiig{Gz?N(R RRHRNRORPR7(R&RQRRRSRT((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_convolve_2D_kernelss    unumbercC`s7td}||}t|j|j|dddS(uN Check if multiplying a kernel with a scalar works correctly. iR!i N(RRRP(R&tnumbertgausst gauss_new((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_multiply_scalars  cC`s2td}||}t|tks.tdS(uN Check if multiplying a kernel with a scalar works correctly. iN(RttypeR7(R&RWRXRY((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_multiply_scalar_types  cC`s2td}||}t|tks.tdS(uN Check if multiplying a kernel with a scalar works correctly. iN(RR[R7(R&RWRXRY((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_rmultiply_scalar_types  cC`s.td}tjt ||WdQXdS(u9Test that multiplying two 1D kernels raises an exception.iN(RR1R2R3(R&RX((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_multiply_kernel1ds cC`s.td}tjt ||WdQXdS(u9Test that multiplying two 2D kernels raises an exception.iN(R R1R2R3(R&RX((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_multiply_kernel2ds cC`s.tjttdtdWdQXdS(ua Test that multiplying a 1D kernel with a 2D kernel raises an exception. iN(R1R2R3RR (R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_multiply_kernel1d_kernel2dscC`s(tjttddWdQXdS(u:Test that adding a scalar to a kernel raises an exception.iiN(R1R2R3R(R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_add_kernel_scalarscC`sud}tdtjdtj|dd|}t|dd}t|dd}t|j|jddd S( u> Check Model1DKernel against Gaussian1Dkernel g@g?iitx_sizeiR!i N(RRHtsqrttpiRRRRP(R&tstddevRXtmodel_gauss_kernelt gauss_kernel((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_model_1D_kernels .cC`srd}tddtj|ddd||}t|dd}t|dd}t|j|jddd S( u> Check Model2DKernel against Gaussian2Dkernel g@g?iiRbiR!i N(RRHRdRR RRP(R&ReRXRfRg((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_model_2D_kernels +cC`srtjd}t|}|jtd}tt|dd}tt|dd}t||dddS(u9 Check CustomKernel against Box1DKernel. iR ufillR!i N(RHRIRR"R RR#R(R&RPtcustomtboxRCRB((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_custom_1D_kernels   cC`srtjd}t|}|jtd}tt|dd}tt|dd}t||dddS(u9 Check CustomKernel against Box2DKernel. iR ufillR!i N(ii(RHRIRR"R RR$R(R&RPRjRkRCRB((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_custom_2D_kernels   cC`s4tdddddg}|jtks0tdS(u9 Check if CustomKernel works with lists. iN(Rtis_boolR4R7(R&Rj((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_custom_1D_kernel_list*scC`sItdddgdddgdddgg}|jtksEtdS(u9 Check if CustomKernel works with lists. iN(RRnR4R7(R&Rj((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_custom_2D_kernel_list1s cC`sYdddddg}t|}|j|jdks@t|jdksUtdS(u] Check if CustomKernel works when the input array/list sums to zero. iiiiigN(RR"t truncationR7t _kernel_sum(R&RPRj((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_custom_1D_kernel_zerosum:s   cC`sndddgdddgdddgg}t|}|j|jdksUt|jdksjtdS(u] Check if CustomKernel works when the input array/list sums to zero. iiigN(RR"RqR7Rr(R&RPRj((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_custom_2D_kernel_zerosumEs *  c C`s0tjttddddgWdQXdS(uH Check if CustomKernel raises if the array size is odd. iN(R1R2RR(R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_custom_kernel_odd_errorPscC`std}td}td}|||}|||}|||}d d d dd ddddg}t|j|ddt|j|ddt|j|ddt|jdddddgddt|jdddgddt|jdgddd S(u: Check if adding of two 1D kernels works. iiig@g@R!i g?Ng?g?gUUUUUU?gUUUUUU?g?g?gUUUUUU?g?gUUUUUU?gUUUUUU?gUUUUUU?(R RRP(R&tbox_1tbox_2tbox_3t box_sum_1t box_sum_2t box_sum_3tref((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_add_1D_kernelsWs   %%cC`std}td}||}||}ddd gd dd d gd ddgg}dddgdddgdddgg}t|jdggddt|j|ddt|j|ddt|j|dddS(u: Check if adding of two 1D kernels works. iig"@R!i Ngqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?gqq?(R RRP(R&RvRwRyRzR|tref_1((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_add_2D_kernelsks       cC`s.tddd}|jjdks*tdS(u> Check if even size for GaussianKernel works. iRbi N(RRPtsizeR7(R&RX((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_Gaussian1DKernel_even_size~scC`s4tddddd}|jjdks0tdS(u> Check if even size for GaussianKernel works. iRbi ty_sizeN(i i (R RPtshapeR7(R&RX((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_Gaussian2DKernel_even_sizescC`sJtdddddg}|jdd|jjdksFtdS(u: Check if normalize works with peak mode. iiitmodeupeakN(RR"RPtmaxR7(R&Rj((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_normalize_peakscC`std}|jdks!tt|jts9t|jddgksTt|jt|j ddd|j stdS(u9 Check if kernel attributes are correct. iiig?R!i N( R RqR7t isinstancetmodelRtcenterR"RRrt separable(R&Rk((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_check_kernel_attributess  umodecC`s|tkr#t r#tjdn|tks>|d}n|dd }|jdkrtt|dddt}t t|dddt}t ||dd nItt |dddt}t t |dddt}t ||dd d S( u Check if the different modes result in kernels that work with convolve. Use only small kernel width, to make the test pass quickly. u/Omitting AiryDisk2DKernel, which requires SciPyig?iR ufillR/R!i Ng433333?( RR=R1R>RR?RR#R0RRR$(R&R@RRARBRC((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_discretize_modess cC`st|}|jdddks)t|jjdksDtt|}tjg|jD]}|ddk^q`st|jjdkstdS(uC Check if BoxKernel work properly with even sizes. iig?N(R RR7RPtsumR RHRN(R&R't kernel_1Dt kernel_2Dt_((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_box_kernels_even_sizes   5cC`sittjd}|jtj|j}|jt||j|jt||jdS(uU Test that repeated normalizations do not change the kernel [#3747]. iN(RRHRIR"tcopyRPR(R&RAtdata((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_kernel_normalizations   cC`s?tjt*ttjd}|jddWdQXdS(uB Test that an error is raised if mode is invalid. iRuinvalidN(R1R2t ValueErrorRRHRIR"(R&RA((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_kernel_normalization_modescC`s*tjttdddWdQXdS(ud Test that an error is raised if ``Kernel1D`` ``x_size`` is not an integer. iRbg333333?N(R1R2t TypeErrorR(R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_kernel1d_int_sizescC`s*tjttdddWdQXdS(ud Test that an error is raised if ``Kernel2D`` ``x_size`` is not an integer. iRbg333333?N(R1R2RR (R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_kernel2d_int_xsizesc C`s0tjttdddddWdQXdS(ud Test that an error is raised if ``Kernel2D`` ``y_size`` is not an integer. iRbiRg333333?N(R1R2RR (R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_kernel2d_int_ysizescC`s!tjt tWdQXdS(un Test that an error is raised if an array or model is not specified for ``Kernel1D``. N(R1R2RR(R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_kernel1d_initializationscC`s!tjt tWdQXdS(un Test that an error is raised if an array or model is not specified for ``Kernel2D``. N(R1R2RR(R&((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyttest_kernel2d_initializations(u kernel_typeuwidth(u kernel_typeuwidth(u kernel_typeumode(6t__name__t __module__t__doc__R1tmarktskipift parametrizet WIDTHS_ODDR.R<tlistt itertoolstproductt KERNEL_TYPESRDRGRJRMRURVtNUMSRZR\R]R^R_R`RaRhRiRlRmRoRpRsRtRuR}RRRRRtMODESRt WIDTHS_EVENRRRRRRRR(((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyR3sT!!00                  0      (iQiQ(i(i((?t __future__RRRRRR1tnumpyRHt numpy.testingRRRRtkernelsRR R R R R RRRRRRRRRRtutilsRtmodeling.modelsRRRtutils.exceptionsRRt scipy.ndimageRR4R=t ImportErrorR0RRRRtfloattfloat32tfloat64RtzerosR#R$trandomtrandRERFtobjectR(((sJlib/python2.7/site-packages/astropy/convolution/tests/test_kernel_class.pyts:"   j   0