ó šßÈ[c@`s)ddlmZmZmZmZddlZddlZddlm Z m Z yddl m Z Wne k r}eZnXeZddlmZmZmZddlmZd „Zejjd ƒd „ƒZd „Zd „Zd„Zd„Zd„Zd„Z d„Z!d„Z"d„Z#dS(i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(t assert_equaltassert_allclose(tstatsi(t sigma_clipt SigmaCliptsigma_clipped_statsi(tNumpyRNGContextc C`sltdƒZtjjdƒ}t|ddddƒ}t|jƒdksRt‚t|jƒ|jksqt‚t|dddddtj ƒ}tj |j|jkƒ s´t‚t|ddddd tj ƒ}tj |j|jkƒ s÷t‚t|dd ddƒ}|j d|dks,t‚|j dcd 7<|j d|dks\t‚t|dd ddd tƒ}|j d|dks—t‚|j dcd 7<|j d|dksÇt‚tjd ƒtjjdddƒtjtjd ƒƒ}t|ddddƒ}|jƒdks2t‚t|ddddƒ}|jƒdksbt‚WdQXdS(Ni90i'tsigmaititersiitstdfunctcenfuncigð?tcopyiggš™™™™™©?taxisgffffff@ii(ii(R tnptrandomtrandnRtsumtmasktAssertionErrortsizetvartalltmeantNonetdatatFalsetarangetnormaltdiagtonestcount(trandvart filtered_datatfiltered_data2tfiltered_data3R((sFlib/python2.7/site-packages/astropy/stats/tests/test_sigma_clipping.pyttest_sigma_clips0 !" ""u not HAS_SCIPYc C`s™tdƒ‡tjjdƒ}t|dddddtjƒ}tj|ddƒd}|j ƒt |ƒkswt ‚t ||j j|ƒWdQXdS(Ni90i'R iR Ri(R RRRRRRRt sigmaclipR"tlenRRRR(R#t astropyrestscipyres((sFlib/python2.7/site-packages/astropy/stats/tests/test_sigma_clipping.pyttest_compare_to_scipy_sigmaclipCs  !cC`sCtjdƒ}t|ddddƒ}|jjdks?t‚dS(u,Test that the returned mask is not a scalar.iR gY@R iN((RRRRtshapeR(Rtresult((sFlib/python2.7/site-packages/astropy/stats/tests/test_sigma_clipping.pyttest_sigma_clip_scalar_maskSsc C`sstdƒatjjdƒ}d|dtjjdgdgƒ}t|ƒ}tjj||ƒdS(u^Make sure a empty masked array is returned when sigma clipping an empty masked array. RRN(RR8R9RRK(RRL((sFlib/python2.7/site-packages/astropy/stats/tests/test_sigma_clipping.pyttest_sigmaclip_empty_masked³s cC`s,tjgƒ}t|ƒ}t||ƒdS(uLMake sure a empty array is returned when sigma clipping an empty array. N(RR4RR(RRL((sFlib/python2.7/site-packages/astropy/stats/tests/test_sigma_clipping.pyttest_sigmaclip_empty½s ($t __future__RRRRtpytesttnumpyRt numpy.testingRRtscipyRt ImportErrorRt HAS_SCIPYR5tsigma_clippingRRR t utils.miscR R'tmarktskipifR,R/R2R>RFRIRJRMRNRO(((sFlib/python2.7/site-packages/astropy/stats/tests/test_sigma_clipping.pyts*"     -