[c@`s/dZddlmZmZmZmZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZmZddlmZyddlZeZWnek reZnXeZdd Zd Zd d Zd ZdZdZdZ de!fdYZ"dS(u Tests for RGB Images i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(t assert_equali(tconvolvetGaussian2DKerneli(t lupton_rgbcC`sLddlj}|j|dddd|r>|j|n|j|S(u<Display an rgb image using matplotlib (useful for debugging)iNt interpolationunearesttoriginulower(tmatplotlib.pyplottpyplottimshowttitletshow(trgbR tplt((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyt display_rgbs  cC`s)|j}||k}tj||<|S(u Return image with all points above satValue set to NaN. Simulates saturation on an image, so we can test 'replace_saturated_pixels' (tcopytnptnan(timagetsatValuetresultt saturated((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pytsaturate(s   idcC`s#tjtjjddd|S(Ni idtdtype(Rtarraytrandom(RtN((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyt random_array4scC`sGttj}tj|}|j|jks6tt||dS(N(RRtfloat64Rtcompute_intensityRtAssertionErrorR(timage_rt intensity((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_compute_intensity_1_float8scC`sGttj}tj|}|j|jks6tt||dS(N(RRtuint8RR RR!R(R"R#((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_compute_intensity_1_uint?scC`swttj}ttj}ttj}tj|||}|j|jksZtt||||ddS(Ng@(RRRRR RR!R(R"timage_gtimage_bR#((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_compute_intensity_3_floatFs cC`swttj}ttj}ttj}tj|||}|j|jksZtt||||ddS(Ni(RRR%RR RR!R(R"R'R(R#((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_compute_intensity_3_uintOs t TestLuptonRgbcB`seZdZdZdZdZdZdZdZe j j ddZ d Z d Zd Zd Zd ZRS(uA test case for RgbcC`stjjdd\|_|_|_d\}}||_||_||f}tj|}tj|}tj|}ddgdd gd d gd dgg}dd d d g} ddddg} ddddg} xt || | | D]|\} } }}| t dd||| d| df<| t dd||| d| df<| || d| dft||dddt}tjjdd|j}||S(Ntboundaryuextendtnormalize_kernelii(RtTrueRRtnormaltshape(RtpsftconvolvedImaget randomImage((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pytconvolve_with_noisevs(iii(iUiK(RRtseedtmin_tstretch_tQtwidththeighttzerostziptpowRR"R'R((tselftmethodR9R:R0R"R'R(tpointstvaluestg_rtr_itptvtgrtriR4R1((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyt setup_method[s,    *())  cC`setj|j|j|j}|j|j|j|j}t rat |dt j j jndS(u1Test creating an RGB image using an asinh stretchR N(Rt AsinhMappingR6R7R8tmake_rgb_imageR"R'R(tdisplayRtsyst _getframetf_codetco_name(R>tasinhMaptrgbImage((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyt test_AsinhscC`setj|j|j|j}|j|j|j|j}trat|dtj j j ndS(uHTest creating an RGB image using an asinh stretch estimated using zscaleR N( RtAsinhZScaleMappingR"R'R(RJRKRRLRMRNRO(R>tmapRQ((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_AsinhZscalescC`setj|j|j|j}|j|j|j|j}trat|dtj j j ndS(uYTest creating an RGB image using an asinh stretch estimated using zscale on the intensityR N( RRSR"R'R(RJRKRRLRMRNRO(R>RTRQ((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_AsinhZscaleIntensityscC`sdddg}|j|d7_|j|d7_|j|d7_tj|j|j|jd|}|j|j|j|j}trt|dtj j j nd S( uTest creating an RGB image using an asinh stretch estimated using zscale on the intensity where the images each have a pedestal addedidiipiiitpedestalR N( R"R'R(RRSRJRKRRLRMRNRO(R>RWRTRQ((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyt!test_AsinhZscaleIntensityPedestals$cC`sYtj|j}|j|j|j|j}trUt|dtjjj ndS(ulTest creating a black-and-white image using an asinh stretch estimated using zscale on the intensityR N( RRSR"RJRKRRLRMRNRO(R>RTRQ((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_AsinhZscaleIntensityBWsunot HAS_MATPLOTLIBc C`sd}tjdd}t|j|}t|j|}t|j|}tj||||j|j |j d|t j j |jstWdQXdS(u"Test the function that does it allg@@tsuffixu.pngtfilenameN(ttempfiletNamedTemporaryFileRR"R'R(Rtmake_lupton_rgbR6R7R8tostpathtexiststnameR!(R>Rttemptredtgreentblue((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyt test_make_rgbs+cC`stjdd}tjddt}t|j|}t|j|}t|j|}tj ||||j |j |j ddddd |WdQXdS( Nu(saturation correction is not implementedg@@RZu.pngtsaturated_border_widthitsaturated_pixel_valueiR[( tpytesttskipR\R]RR"R'R(RR^R6R7R8(R>RRcRdReRf((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_make_rgb_saturated_fixs $ cC`sYtjdd}|j|j|j|j}trUt|dtj j j ndS(u%Test using a specified linear stretchgfffff gzG*@R N( Rt LinearMappingRJR"R'R(RKRRLRMRNRO(R>RTRQ((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyt test_linearscC`s\tjd|j}|j|j|j|j}trXt|dtj j j ndS(u=Test using a min/max linear stretch determined from one imageRR N( RRmR(RJR"R'RKRRLRMRNRO(R>RTRQ((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_linear_min_maxscC`stjdd}t|j||_t|j||_t|j||_tj|j|j|jddtj |jj j j st tj |jj j j st tj |jj j j st |jj |_|jj |_|jj |_tj|j|j|j}|j|j|j|j}trt|dtjjjndS(u*Test interpolationolating saturated pixelsu8replaceSaturatedPixels is not implemented in astropy yetg@@iiR N(RjRkRR"R'R(RtreplaceSaturatedPixelsRtisfinitetgetImagetgetArraytallR!timagesRtimagesGtimagesBRIR6R7R8RJRKRRLRMRNRO(R>RRPRQ((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_saturateds  "***cC`sktjt;}|jj|j|j}tj||j |j WdQXdt |j ksgt dS(Nushapes must match(Rjtraisest ValueErrorR"treshapeR:R9RR^R'R(tstrtvalueR!(R>texcinfoR"((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyttest_different_shapes_assertss(t__name__t __module__t__doc__RHRRRURVRXRYRjtmarktskipifRgRlRnRoRxR(((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyR+Xs %        (#Rt __future__RRRRRLR_R\RjtnumpyRt numpy.testingRt convolutionRRtRt matplotlibR.tHAS_MATPLOTLIBt ImportErrortFalseRKtNoneRRRR$R&R)R*tobjectR+(((sJlib/python2.7/site-packages/astropy/visualization/tests/test_lupton_rgb.pyts."