ó šßÈ[c@`sddlmZmZmZmZddlZddlZddlj Z ddl m Z m Z ddl m Z mZddlZgZxJdddd gD]6Zx-dddd gD]Zejeefƒq³WqšWdd d d gZd d gZeegZeegZeeeje fƒeƒƒe d fe d fgZeZyddl Z Wne!k rpeZnXde"fd„ƒYZ#de"fd„ƒYZ$de"fd„ƒYZ%ej&j'deƒd„ƒZ(ej&j'ddƒd„ƒZ)d„Z*ej&j+dƒd„ƒZ,dS(i(tabsolute_importtdivisiontprint_functiontunicode_literalsNi(tconvolvet convolve_fft(tassert_array_almost_equal_nulptassert_array_almost_equalu>f4uf8uf8R N(R R ttestingtassert_allclose(R R tconvfuncRRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_1_noneKscC`s²tjdddgddƒ}tjdddgddƒ}t||d|ƒ}|dkr“tj|tjdddgddƒkƒs®t‚ntj||kƒs®t‚dS( u| Test that a unit kernel with three elements returns the same array (except when boundary is None). gð?g@g@Ru>f8gR N(R R RR RR(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyt test_unity_3Ys  6cC`sTtjdddgddƒ}tjdddgddƒ}t||d|dtƒ}|d kr™tj|tjdddgddƒkƒsPt‚n·|d krÛtj|tjdddgddƒkƒsPt‚nu|d krtj|tjdddgddƒkƒsPt‚n3tj|tjd dd gddƒkƒsPt‚d S(u„ Test that the different modes are producing the correct results using a uniform kernel with three elements gð?gg@Ru>f8R tnormalize_kernelg@ufilluwrapg@g@N(R R RtFalseR RR(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_uniform_3ks 6 6 6u nan_treatmentunormalize_kernelu preserve_nanc C`stjdtjdgddƒ}tjdddgddƒ}t||d|d|d|d |ƒ}|rˆtj|d ƒsˆt‚ntj|ƒ}tj|ƒ}|d krètj|tjdddgddƒkƒst‚ntj||kƒst‚d S( u½ Test that a unit kernel with three elements returns the same array (except when boundary is None). This version includes a NaN value in the original array. gð?g@Ru>f8gR t nan_treatmentRt preserve_naniN( R R tnanRtisnanRt nan_to_numR R(R R R"RR#RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_3_withnans!  6c C`sptjdtjdgddƒ}tjdddgddƒ}t||d|d|d|d|ƒ}|rˆtj|d ƒsˆt‚ntj|ƒ}id d d gdd tf6d d d gdd t f6d dd gddtf6d dd gddt f6dd dgdd tf6dd dgdd t f6dddgddtf6d ddgddt f6d d d gdd tf6d d d gdd t f6dd d!gddtf6dddgddt f6d d dgdd tf6dd dgdd t f6d"d#d$gddtf6d dd gddt f6|||f}|rMd |d f8R R"RR#iiiu interpolateiiufillg@ii uwrapuextendi NgUUUUUUõ?gà?gø?gø?g@gUUUUUUÕ?gUUUUUUõ?gð?gUUUUUUõ?gUUUUUUõ?gUUUUUUõ?gUUUUUUå?gUUUUUUõ?g@( R R R$RR%RR&R tTrueR R( R R R"RR#RRRtrslt((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_uniform_3_withnan¢s:!  c C`sL|rtjdƒndddddddd d g }d d d d d d d d d g }tjt|ƒd d dƒs|t‚t||d|d|ƒ}idddddddddg d+6dddddddddg d6dddddddd d!g d"6d#ddddd$d%d&dg d'6|}t|tj |d(d)ƒd*ƒd+S(,uK Test that convolve works correctly with zero sum kernels. u(You can't normalize by a zero sum kerneliiiiiiiii iÿÿÿÿitatolg:Œ0âŽyE>R RggÀgÀgð¿g$@g5@g€@@gG@ufillgBÀg;Àg2Àg"Àg"@g2@g;@gB@uwrapg$Àgð?g@g@uextendRu>f8i N( tpytesttxfailR tisclosetsumRRR RR (R R RRRRR)((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_zero_sum_kernelÕs!!$""""c C`sX|rtjdƒndddddddd d g }tjd d d d d d d d d g d dd d d d d d d d g ddƒ}t||d|d|ƒ}idddddddddg d-6dddddddddg d6dddddddd d!g d"6d#dd$d%ddd&d'd(g d)6|}t|tj|d*d+ƒd,ƒd-S(.uQ Test that convolve works correctly with integer masked kernels. u(You can't normalize by a zero sum kerneliiiiiiiii iÿÿÿÿtmaskit fill_valuegR Rg"@gð¿g@g@g @g$@g5@g€@@gG@ufillg?Àg5Àg&Àg4@g>@gD@uwrapgÀg@g@g(@g.@g3@uextendRu>f8i N(R,R-tmaR RR RR (R R RRRRR)((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_int_masked_kernelðs!Q""""(u dtype_arrayu dtype_kernel(uconvfuncuboundary(uboundaryu nan_treatmentunormalize_kernelu preserve_nan(uboundaryu nan_treatmentunormalize_kernelu preserve_nan(uboundaryunormalize_kernel(uboundaryunormalize_kernel(t__name__t __module__RRR,tmarkt parametrizet VALID_DTYPESRtBOUNDARIES_AND_CONVOLUTIONSRtBOUNDARY_OPTIONSRR!t itertoolstproducttNANHANDLING_OPTIONStNORMALIZE_OPTIONStPRESERVE_NAN_OPTIONSR'R*R0R4(((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyR%s0     .    tTestConvolve2DcB`seZd„Zejjd eƒd„ƒZejjdeƒd„ƒZ ejjdeƒd„ƒZ ejjdeƒd„ƒZ ejjdeƒd„ƒZ ejjdeƒd „ƒZ ejjdeƒd „ƒZejjdeƒd „ƒZRS( c C`s dddgdddgdddgg}t||dddddtƒ}t||dƒt||dddddtƒ}t|tj|tƒddƒdS( uJ Test that convolve works correctly when inputs are lists iR ufillR2Ri i N(RR(RR R R tfloat(R RR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyR s  !!u dtype_arrayu dtype_kernelcC`stjdddgdddgddd ggd |ƒ}tjd d d gd dd gd d d ggd |ƒ}t||ƒ}|j|jks™t‚d S( uG Test that 32- and 64-bit floats are correctly handled gð?g@g@g@g@g@g@g @g"@RgN(R R RRR(R RRRRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyRs  uboundarycC`sˆtjdddgdddgddd ggd d ƒ}tjdggd d ƒ}t||d |ƒ}tj||kƒs„t‚d S(uD Test that a 1x1 unit kernel returns the same array gð?g@g@g@g@g@g@g @g"@Ru>f8R N(R R RRR(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_1x1_none+s  cC`stjdddgdddgddd ggd d ƒ}tjd d d gd dd gd d d ggd d ƒ}t||d |ƒ}|dkrätj|tjd d d gd dd gd d d ggd d ƒkƒsÿt‚ntj||kƒsÿt‚dS(ul Test that a 3x3 unit kernel returns the same array (except when boundary is None). gð?g@g@g@g@g@g@g @g"@Ru>f8gR N(R R RR RR(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_3x3;s    *cC`sÊtjdddgdddgdddggddƒ}tjdddgdddgdddggddƒ}t||d|dtƒ}|dkrßt|tjdddgdd dgdddggddƒd ƒnç|d kr1t|tjdd dgdd d gdddggddƒd ƒn•|dkrƒt|tjd d d gd d d gd d d ggddƒd ƒnCt|tjdddgd d dgd d d ggddƒd ƒdS(uu Test that the different modes are producing the correct results using a 3x3 uniform kernel. gg@gð?g@Ru>f8R Rg@i ufillg@g@uwrapg@g(@g @N(R R RR R R(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_uniform_3x3Ss,    "  "  " c C`sItjdddgdtjdgdddggd d ƒ}tjd d d gd dd gd d d ggd d ƒ}t||d |d ddtƒ}tj|dƒs¯t‚tj|ƒ}tj|ƒ}|dkr*tj |tjd d d gd d d gd d d ggd d ƒkƒsEt‚ntj ||kƒsEt‚dS(u­ Test that a 3x3 unit kernel returns the same array (except when boundary is None). This version includes a NaN value in the original array. gð?g@g@g@g@g@g @g"@Ru>f8gR R"ufillR#iN(ii( R R R$RR(R%RR&R R(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_3x3_withnanus     *c C`sîtjdddgdtjdgdddggddƒ}tjdddgdddgdddggddƒ}t||d|dd d tƒ}|dkrèt|tjdddgdd dgdddggddƒd ƒn|d kr:t|tjdd dgdd dgdddggddƒd ƒn°|dkrŒt|tjd d d gd d d gd d d ggddƒd ƒn^|dkrÞt|tjdddgd d dgd ddggddƒd ƒn tdƒ‚dS(u¶ Test that the different modes are producing the correct results using a 3x3 uniform kernel. This version includes a NaN value in the original array. gg@gð?g@Ru>f8R R"ufillRg @i g@g@uwrapuextendg@g"@g0@g&@g@uInvalid boundary specificationN(R R R$RR R Rt ValueError(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_uniform_3x3_withnanfilled“s2    "  "  "  "c C`sîtjdddgdtjdgdddggddƒ}tjdddgdddgdddggddƒ}t||d|dd d tƒ}|dkrèt|tjdddgdddgdddggddƒd ƒn|d kr:t|tjdddgdddgd d!d"ggddƒd ƒn°|dkrŒt|tjdddgdddgdddggddƒd ƒn^|dkrÞt|tjd#d$d%gd&d'd(gd)d*d+ggddƒd ƒn tdƒ‚dS(,u¶ Test that the different modes are producing the correct results using a 3x3 uniform kernel. This version includes a NaN value in the original array. gg@gð?g@Ru>f8R R"u interpolateRi ufillig@g @g@uwrapuextendg@g"@g0@g&@g@uInvalid boundary specificationNgÀ?gä?gà?gà?gð?gì?gà?gà?gØ?gÐ?gò?g@gä?gð?gö?gð?gì?gè?(R R R$RR(R RRG(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyt test_uniform_3x3_withnaninterped¹s2    "  "  "  "c C`sëtjdddgdddgdddggddƒ}tjdddgdddgdddggddƒ}t||d |d d d tƒ}|dkråt|tjdddgdddgdddggddƒd ƒn|d kr7t|tjdddgdddgdddggddƒd ƒn°|dkr‰t|tjdddgdddgdddggddƒd ƒn^|dkrÛt|tjdddgdddgdddggddƒd ƒn tdƒ‚dS(Ngg@gð?g@g@Rufloatgð¿R R"ufillRi gÀgÀgÀuwrapg Àg@g@gÀuextenduInvalid boundary specification(R R RR R RRG(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_non_normalized_kernel_2Dßs2     "  "  "  "(u dtype_arrayu dtype_kernel(R5R6RR,R7R8R9RR;RCRDRERFRHRIRJ(((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyRA s "&&tTestConvolve3DcB`sïeZd„Zejjd eƒd„ƒZejjdeƒd„ƒZ ejjdeƒd„ƒZ ejjdeƒd„ƒZ ejjd e j eeƒƒd „ƒZejjdeƒd „ƒZejjdeƒd „ƒZRS(c C`s´dddgdddgdddggdddgdddgdddggdddgdddgdddggg}t||dddddtƒ}t|d|dƒdS( uJ Test that convolve works correctly when inputs are lists iR ufillR2Rii N(RR R(R RR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyRs      !u dtype_arrayu dtype_kernelcC`stjdddgdddgddd ggd |ƒ}tjd d d gd dd gd d d ggd |ƒ}t||ƒ}|j|jks™t‚d S( uG Test that 32- and 64-bit floats are correctly handled gð?g@g@g@g@g@g@g @g"@RgN(R R RRR(R RRRRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyRs  uboundarycC`sÜtjdddgdddgdddggdddgdddgdddggdddgd ddgd ddgggd d ƒ}tjdgggd d ƒ}t||d |ƒ}tj||kƒsØt‚dS(uF Test that a 1x1x1 unit kernel returns the same array gð?g@g@g@g@gg@g@g @g"@Ru>f8R N(R R RRR(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_1x1x1_none%s -'6c C`s‹tjdddgdddgdddggdddgdddgdddggdddgd ddgd ddgggd d ƒ}tjdd d ƒ}d|df8iiR N(iii(iii(R R tzerosRR RR(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_3x3x35s-'6  6'Ec C`s;tjdddgdddgdddggdddgdddgdddggdddgd ddgd ddgggd d ƒ}tjd=d d ƒ}t||d|dtƒ}|d<kr]t|tjdddgdddgdddggdddgdddgdddggdddgdddgdddgggd d ƒdƒnÚ|dkrt|tjdddgdddgdddggdddgdddgddd ggd!ddgdd"d#gdd$dgggd d ƒdƒn7|d%kr£t|tjdddgdddgdddggdddgdddgdddggdddgdddgdddgggd d ƒdƒn”t|tjd&d'd(gd)d*d+gd,d-d.ggd/d.dgd0dd'gd1d2d3ggd4d5d6gd7d/d8gd9d:d;gggd d ƒdƒd<S(>uu Test that the different modes are producing the correct results using a 3x3 uniform kernel. gð?g@g@g@g@g@g@gg @g"@Ru>f8iR Rg@T@i ufillg7@g<@g0@g€A@gG@g9@gA@g2@gD@gI@g€O@gB@gN@g;@g@@g€N@g6@gF@uwrapg@P@gK@g€E@gÀR@g€P@g€L@g@U@g€S@gÀQ@gX@g[@g^@gÀV@gO@gÀ_@gV@g€H@g a@g€I@g`c@gZ@g€J@N(iii(R R tonesRR R R(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_uniform_3x3x3Ls(-'6 3'= 3'= 3'=3'u nan_treatmentc C`sÑtjdddgdddgdddggdddgdtjdgdddggdddgd ddgd ddgggd d ƒ}tjdd d ƒ}d|df8iiR R"R#N(iii(iii(iii( R R R$RMRR(R%RR&R R(R R R"RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_unity_3x3x3_withnanls -*6   6'Ec C`s_tjdddgdddgdddggdddgdtjdgdddggdddgd ddgd ddgggd d ƒ}tjd@d d ƒ}t||d|dddtƒ}|d?krft|tjdddgdddgdddggdddgdddgdddggdddgdddgdddgggd d ƒdƒnõ|dkr t|tjdddgdddgdddggdddgdddgdd d!ggd"ddgdd#d$gdd%dgggd d ƒdƒnR|d&kr¬t|tjdddgdddgdddggdddgdddgdddggdddgdddgdddgggd d ƒdƒn¯|d'krOt|tjd(d)d*gd+d,d-gd.d/d0ggd1d0dgd2dd)gd3d4d5ggd6d7d8gd9d1d:gd;d<d=gggd d ƒdƒn td>ƒ‚d?S(Au¶ Test that the different modes are producing the correct results using a 3x3 uniform kernel. This version includes a NaN value in the original array. gð?g@g@g@g@g@g@gg @g"@Ru>f8iR R"ufillRg€S@i g4@g9@g*@g@@g€E@g6@g?@g.@g€B@g€G@gN@g€@@g€L@g8@g=@gM@g3@g€D@uwrapuextendgO@g€I@gD@gR@g€O@gK@g€T@gÀR@gQ@g@W@g@Z@g@]@gV@g€M@g_@g@U@gG@g@a@gH@gc@g@Y@gI@uInvalid Boundary OptionN(iii( R R R$RORR R RRG(R R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyt!test_uniform_3x3x3_withnan_filled‹sH-*6  3'=      % 3'=      c C`s2tjdddgdddgdddggdddgdtjdgdddggdddgd ddgd ddgggd d ƒ}tjdBd d ƒ}t||d|dddtƒ}|jƒd}|tj|ƒjƒ|}|dAkr—t |tjdddgdddgdddggdddgdddgdddggdddgdddgdddgggd d ƒ|dƒn—|dkr>t |tjdddgdddgdddggdddgd dd!gdd"d#ggd$ddgdd%d&gd!d'dgggd d ƒ|dƒnð|d(kr{t |tj |j d ƒd d d gƒdƒn³|d)kr"t |tjd*d+d,gd-d.d/gd0d1d2ggd3d2dgd4dd+gd5d6d7ggd8d9d:gd;d3d<gd=d>d?gggd d ƒ|dƒn t d@ƒ‚dAS(Cu¶ Test that the different modes are producing the correct results using a 3x3 uniform kernel. This version includes a NaN value in the original array. gð?g@g@g@g@g@g@gg @g"@Ru>f8iR R"u interpolateRig€S@i ufillg4@g9@g*@g@@g€E@g6@g?@g.@g€B@g€G@gN@g€@@g€L@g8@g=@gM@g3@g€D@uwrapuextendgO@g€I@gD@gR@g€O@gK@g€T@gÀR@gQ@g@W@g@Z@g@]@gV@g€M@g_@g@U@gG@g@a@gH@gc@g@Y@gI@uInvalid Boundary OptionN(iii( R R R$RORR(R/tisfiniteR RttiletastypeRG(R R RRRtkernsumtmid((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyt#test_uniform_3x3x3_withnan_interped¼sL-*6  3'-       1      (u dtype_arrayu dtype_kernel(uboundaryu nan_treatment(R5R6RR,R7R8R9RR;RLRNRPR<R=R>RQRRRX(((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyRKs   1uboundarycC`s7tjdddgddƒ}tjdddgddƒ}t||d |d tƒ}|d krŽt|tjd d dgddƒdƒn¥|dkrÅt|tjdd dgddƒdƒnn|dkrüt|tjdd dgddƒdƒn7|dkr3t|tjdd dgddƒdƒndS(uo Regression test for #6264: make sure that asymmetric convolution functions go the right direction g@ggð?Ru>f8iiiR Rufillg@g$@g@ufloati uextendg.@uwrapg"@g@N(R R RR RR (R RRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_asymmetric_kernelñs + + + undimsiicC`stjjdƒtjjdg|Œ}tjjdƒtjjdg|Œ}t||ddƒ}t||ddƒ}t||dƒdS(NiiR ufilli(R trandomtseedtrandntrandRRR(tndimsR tkerneltconv_ftconv_d((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyttest_convolution_consistencyscC`stjdddgƒ}tjdddddgƒ}ttj||dƒt||dtƒƒttj||dƒt||dtƒƒdS(NiiiiiusameR(R R RRR R(RR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyt&test_astropy_convolution_against_numpys u not HAS_SCIPYcC`sšddlm}tjdddgƒ}tjdddddgƒ}t|||dƒt||d tƒƒt|||dƒt||d tƒƒdS( Ni(t fftconvolveiiiiiusameR(t scipy.signalRdR R RRR R(RdRR((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyt&test_astropy_convolution_against_scipy s(iii(-t __future__RRRRR,tnumpyR tnumpy.maR3RRt numpy.testingRRR<R9RRtappendR R;R>R(R R?R@tlisttziptcycleR:t HAS_SCIPYtscipyt ImportErrortobjectRRARKR7R8RYRbRctskipifRf(((sFlib/python2.7/site-packages/astropy/convolution/tests/test_convolve.pyts>"          æöð