ó šßČ[c@`sčddlmZmZmZmZddlZddlZddlZddl m Z m Z m Z ddl m Z mZddlmZddlmZmZyddlZWnek rÇeZnXeZd efd „ƒYZdS( i(tabsolute_importtdivisiontprint_functiontunicode_literalsNi(tconvolvet convolve_ffttconvolve_modelsi(tmodelstfitting(tNumpyRNGContext(tassert_allclosetassert_almost_equaltTestConvolve1DModelscB`s(eZejjdddgƒejjdƒd„ƒƒZejjdddgƒejjdƒd„ƒƒZejjdddgƒejjdƒd„ƒƒZejjdddgƒd„ƒZ ejjdddgƒd„ƒZ ejjdddgƒejjdƒd „ƒƒZ RS( umodeu convolve_fftuconvolveu not HAS_SCIPYcC`sƒtjdddƒ}tjdddƒ}t||d|ƒ}tjddƒ}tdj|ƒƒ}t|||ƒddƒdS( Niitmodeiű˙˙˙iu{}(model(x), kernel(x))tatolgńh㈾řä>(Rt Gaussian1DRtnptarangetevaltformatR (tselfR tkerneltmodelt model_convtxtans((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyt,test_is_consistency_with_astropy_convolutions cC`sŻddlm}tjdddƒ}tjdddƒ}t||d|ƒ}tjddƒ}|||ƒ||ƒddƒ}t|||ƒ||ƒjƒdd ƒdS( Ni(t fftconvolveiR iű˙˙˙iusameRgńh㈾řä>( t scipy.signalRRRRRRR tsum(RR RRRRRR((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyttest_against_scipy"s!cC`sĽddlm}tjdddƒ}tjdddƒ}t||d|dtƒ}tjddƒ}|||ƒ||ƒddƒ}t|||ƒd d ƒdS( Ni(RiR tnormalize_kerneliű˙˙˙iusameRgńh㈾řä>( RRRRRtFalseRRR (RR RRRRRR((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyt+test_against_scipy_with_additional_keywords/s !cC`sÔtjdtjdtjƒddƒ}tjdtjdtjƒddƒ}t||d|dtƒ}tjddtjtjƒdtjdƒƒ}tjddƒ}t ||ƒ||ƒd d ƒd S( uä Test that convolving N(a, b) with N(c, d) gives N(a + c, b + d), where N(., .) stands for Gaussian probability density function, in which a and c are their means and b and d are their variances. iiiR Riiű˙˙˙iRgüŠńŇMbP?N( RRtmathtsqrtRtpiRR RR (RR RRRRR((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyttest_sum_of_gaussians=s)) 2cC`s†tjƒ}tjƒ}t||d|ƒ}tjdddƒ}|d|dk| d|dk}t|||ƒddƒdS(NR i˙˙˙˙iiciRgüŠńŇMbP?(RtBox1DRRtlinspaceR (RR RRRRR((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyttest_convolve_box_modelsNs   'c C`sŕtjƒ}tjƒ}tjdddƒ}tjddƒ}tdƒ*||ƒtjjdt|ƒƒ}WdQXt ||d |d t ƒ}t j ƒ}||||ƒ} tj | |ƒ|ƒ} t| d d d ƒdS(u: test that a convolve model can be fitted iű˙˙˙iict amplitudei i{tsizeNR Rgtdecimali(RR&RRR'R trandomtnormaltlenRR RtLevMarLSQFittertmeanR ( RR tb1tg1Rt fake_modelt fake_datat init_modeltfittert fitted_modeltme((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyttest_fitting_convolve_modelsXs   + ( t__name__t __module__tpytesttmarkt parametrizetskipifRRR!R%R(R9(((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyR s   $$ (t __future__RRRRR"tnumpyRR<RRRtmodelingRRt utils.miscR t numpy.testingR R tscipyt ImportErrorR t HAS_SCIPYtTruetobjectR (((sMlib/python2.7/site-packages/astropy/convolution/tests/test_convolve_models.pyts"