ó ›ßČ[c@s;ddlZddlZddlmZddlmZddlmZmZddl m Z ddl m Z yddl Z eZWnek reZnXejdd d ƒZejd ƒZd eZejjd ƒd„ƒZejjdƒdefd„ƒYƒZejjdƒdefd„ƒYƒZdS(i’’’’N(tma(tassert_allclosei(tImageNormalizet simple_norm(tManualInterval(t SqrtStretchgg.@iigą?tHAS_MATPLOTLIBcCs?tjtƒ}tƒWdQX|jjddks;t‚dS(Nis2matplotlib is required in order to use this class.(tpytesttraisest ImportErrorRtvaluetargstAssertionError(texc((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_normalize_error_messages snot HAS_MATPLOTLIBt TestNormalizecBsPeZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z RS(c Cs9tjtƒ$tdddddtdtƒWdQXdS(Ntvming@tvmaxg$@tintervaltclip(RRt TypeErrorRRtTrue(tself((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_invalid_interval"sc Cs9tjtƒ$tdddddtdtƒWdQXdS(NRg@Rg$@tstretchR(RRRRRR(R((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_invalid_stretch'sc Cstdddddtƒdtƒ}tddd td d ƒdtƒdtƒ}t|dƒd ƒt|dƒ|dƒƒdS( NRg@Rg$@RRtdataiRii gß'Üež ę?(RRRRR(Rtnormtnorm2((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_scalar,s  c Csøtdddddtƒdtƒ}ttdtdd ƒdtƒdtƒ}|tƒ}d d d d ddg}t||ƒt|jddddddgƒt||tƒƒdS(NRg@Rg$@RRRii ggß'Üež Ö?gß'Üež ę?gķ°.źīķ?gš?i(RRRtDATARRtmask(RRRtoutputtexpected((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_clip4s   "c Csßtdddddtƒdtƒ}ttdtdd ƒdtƒdtƒ}|tƒ}tjd d d d dg}t||ƒt|jddddddgƒt|j |tƒƒdtdƒt||tƒƒdS(NRg@Rg$@RRRii gß'Üež Ö?gß'Üež ę?gķ°.źīķ?g½Üķ›wćń?g1C€_eeō?ii( RRtFalseRRtnptnanRRtinverse(RRRR R!((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_noclip?s    "$c CsDtdddddtƒdtƒ}ttdtddƒdtƒdtƒ}|tƒ}|jtjtƒksxt ‚|j dkst ‚t ||tƒƒtdddddtƒdtƒ}ttdtd dƒdtƒdtƒ}|tƒ}|jdkst ‚|j tj tƒks-t ‚t ||tƒƒdS( NRRg$@RRRi g@i( RtNoneRR#RRRR$tminR RRtmax(RRRR ((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_implicit_autoscaleLs     c Csčtjtdddddddgƒ}tdddddtƒd tƒ}t|d td d ƒdtƒd tƒ}||ƒ}d dddddg}t|jdƒ|ƒt|j ddddddgƒt|||ƒƒdS(NRiiRg@Rg$@RRRii ggß'Üež Ö?gš?gķ°.źīķ?iö’’’( RtarrayRRRRRRtfilledR(RtmdataRRR R!((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_masked_clip_s'  "c Cstjtdddddddgƒ}tdddddtƒd tƒ}t|d td d ƒdtƒd tƒ}||ƒ}tjd ddddg}t |j dƒ|ƒt |j ddddddgƒt |j |tƒƒdtdƒt |||ƒƒdS(NRiiRg@Rg$@RRRii gß'Üež Ö?iö’’’gķ°.źīķ?g½Üķ›wćń?g1C€_eeō?( RR,RRRR#RR$R%RR-RR&(RR.RRR R!((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_masked_noclipks'   "$( t__name__t __module__RRRR"R'R+R/R0(((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyR s     tTestImageScalingcBsPeZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z RS(cCs5ttddƒ}t|tƒtddddƒdS(sTest linear scaling.Rtlineartatolitrtolgńh戵ųä>N(RtDATA2RtDATA2SCL(RR((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_linear~scCs>ttddƒ}t|tƒtjtƒddddƒdS(sTest sqrt scaling.RtsqrtR5iR6gńh戵ųä>N(RR7RR$R:R8(RR((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_sqrtƒscCsEd}ttddd|ƒ}t|tƒt|ddddƒdS( sTest power scaling.g@RtpowerR5iR6gńh戵ųä>N(RR7RR8(RR<R((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_powerˆscCsYttddƒ}tjdtdƒtjdƒ}t|tƒ|dddd ƒd S( sTest log10 scaling.Rtlogičgš?gH@R5iR6gńh戵ųä>N(RR7R$tlog10R8R(RRtref((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_logŽs$cCsed}ttddd|ƒ}tjt|ƒtjd|ƒ}t|tƒ|dddd ƒd S( sTest asinh scaling.gš™™™™™¹?Rtasinhtasinh_agš?R5iR6gńh戵ųä>N(RR7R$tarcsinhR8R(RtaRR@((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_asinh”s$cCsDttddddƒ}t|tƒdddgdddd ƒd S( sTest linear scaling.RR4tmin_cutgš?gR5iR6gńh戵ųä>N(RR7R(RR((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyttest_min›scCs~ttddddƒ}t|tƒtddddƒttddd d d d ƒ}t|tƒ|tƒddddƒd S(sTest percent keywords.RR4tpercentgĄX@R5iR6gńh戵ųä>t min_percentgą?t max_percentgąX@N(RR7RR8(RRR((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyt test_percent s  cCs*tjtƒttddƒWdQXdS(sTest invalid stretch keyword.RtinvalidN(RRt ValueErrorRR7(R((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyR©s( R1R2R9R;R=RARFRHRLR(((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyR3{s       (RtnumpyR$Rt numpy.testingRt mpl_normalizeRRRRRRt matplotlibRRR R#tlinspaceRtarangeR7R8tmarktskipifRtobjectRR3(((sDlib/python2.7/site-packages/astropy/visualization/tests/test_norm.pyts&       Z