ó šßÈ[c@`sÿddlmZmZmZmZddlZddlZddlZddl m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZmZdd lmZdd lmZdd lmZmZmZddlm Z ddl!m"Z"idd6dd6dd6dd6Z#dZ$dZ%dZ&d„Z'ej(d„ƒZ)d„Z*d„Z+d„Z,d„Z-ej.j/dƒd„ƒZ0d „Z1ej.j/dƒd!„ƒZ2d"„Z3d#„Z4d$„Z5d%„Z6d&„Z7d'„Z8d(„Z9d)„Z:d*„Z;d+„Z<d,„Z=d-„Z>d.„Z?d/„Z@d0„ZAd1„ZBd2„ZCd3„ZDd4„ZEd5„ZFej.jGd6d7eHfd8eHfgƒej.jGd9d:dejIdejJejKgƒej.jGd;eHeLgƒej.jMejKƒd<„ƒƒƒƒZNej.jGd6d=eLfd>eLfgƒej.jGd?d:ejOfdejIejOfdejKeLfgƒej.jGd;eHeLgƒej.jMejKƒd@„ƒƒƒƒZPdA„ZQdB„ZRdC„ZSdD„ZTdE„ZUdF„ZVdG„ZWdH„ZXdI„ZYdJ„ZZdK„Z[dL„Z\dM„Z]dN„Z^dO„Z_dP„Z`dQ„ZadR„Zbej.jGdSd7d8d=d>gƒdT„ƒZcdU„Zdej.jGdSd7d8d=d>gƒdV„ƒZedW„ZfdX„ZgdY„ZhdZ„Zid[„Zjd\„Zkd]„Zld^„Zmd_„ZndS(`i(tabsolute_importtdivisiontprint_functiontunicode_literalsNi(tsix(tfitsi(tStdDevUncertaintytMissingDataAssociationException(tunits(tlog(tWCStFITSFixedWarning(tcatch_warnings(tNumpyRNGContext(tget_pkg_data_filenametget_pkg_data_filenamestget_pkg_data_contents(tCCDData(tTablei{useedidu data_sizegð?u data_scalegu data_meancC`s:y|j|jd}Wntk r5t|}nX|S(Ni(tkeywordstargstKeyErrortDEFAULTS(tkeytrequesttval((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pytvalue_from_markers's  c C`s¡td|ƒ}td|ƒ}td|ƒ}ttdƒ,tjjd|d||gd|ƒ}WdQXid d 6d d 6}t|d tjƒ}||_ |S(uR Return a CCDData object with units of ADU. The size of the data array is 100x100 but can be changed using the marker @pytest.mark.data_size(N) on the test function, where N should be the desired dimension. Data values are initialized to random numbers drawn from a normal distribution with mean of 0 and scale 1. The scale can be changed with the marker @pytest.marker.scale(s) on the test function, where s is the desired scale. The mean can be changed with the marker @pytest.marker.scale(m) on the test function, where m is the desired mean. u data_sizeu data_scaleu data_meanuseedtloctsizetscaleNi*umy_keyunot 42uyour_keytunit( RR RtnptrandomtnormalRtutadutheader(RRRtmeantdatat fake_metatccd((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pytccd_data/s- cC`s!tjtƒ tƒWdQXdS(N(tpytesttraisest TypeErrorR(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_ccddata_emptyNscC`s3tjtƒttjddgƒƒWdQXdS(Nid(R*R+t ValueErrorRRtzeros(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_ccddata_must_have_unitSscC`s#tjtƒd|_WdQXdS(N(R*R+R,tNoneR(R)((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt'test_ccddata_unit_cannot_be_set_to_noneXscC`sgtjtƒR}tdddgdddidd6didd6ƒdt|ƒks]t‚WdQXdS( NiiiRutmetaR$u can't have both header and meta.(R*R+R.RtstrtAssertionError(texc((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt!test_ccddata_meta_header_conflict]s3i cC`sL|jdkst‚|jdks*t‚|jtjtƒksHt‚dS(Ni id(i i (tshapeR5RtdtypeRtfloat(R)((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_ccddata_simplecscC`s7ttjdƒddƒ}|jtjks3t‚dS(Ni Ruelectron(i i (RRR/RR"telectronR5(R(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt+test_ccddata_init_with_string_electron_unitjscC`sàtj|ƒ}tj|gƒ}|jdƒj}|j|ƒtj|dtj ƒ}|j dksmt ‚|j dks‚t ‚t j|jjt jƒs£t ‚x6|jjƒD]%\}}|j||ks³t ‚q³WdS(Nu afile.fitsRi id(i i (Rt PrimaryHDUtHDUListtjointstrpathtwritetoRtreadR"R<R8R5RRt issubdtypeR&R9tfloatingR$titemsR3(R)ttmpdirthduthdulisttfilenametcdtktv((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_initialize_from_FITSos !cC`s±tjjddƒ}tj|ƒ}tjjƒ|jd<|jdƒj }|j |ƒt j |ƒ}|j tjks€t‚t j |ddƒ}|j tjks­t‚dS(NRidubunitu afile.fitsRuphoton(idid(RR RR>R"R#t to_stringR$R@RARBRRCRR5tphoton(RGtfake_imgRHRJR(tccd2((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt-test_initialize_from_fits_with_unit_in_header}s cC`s{tjjddƒ}tj|ƒ}d|jd<|jdƒj}|j|ƒtj |ƒ}|j t j kswt ‚dS(NRiduADUubunitu afile.fits(idid(RR RR>R$R@RARBRRCRR"R#R5(RGRQRHRJR(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt,test_initialize_from_fits_with_ADU_in_headerŒs  cC`sktjtjdƒƒ}d|jd<|jdƒj}|j|ƒtj t ƒt j |ƒWdQXdS(Niudefinetely-not-a-unitubunitu afile.fits(ii( RR>RtonesR$R@RARBR*R+R.RRC(RGRHRJ((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt5test_initialize_from_fits_with_invalid_unit_in_header—s   cC`sßtjjddƒ}tjƒ}tj|ƒ}tj||gƒ}|jdƒj}|j|ƒt t ƒ}t j |ddƒ}WdQXt |ƒdks¦t‚tjj|j|ƒ|j|j|jksÛt‚dS(NRidu afile.fitsRuadui(idid(RR RR>tImageHDUR?R@RARBR R RRCtlenR5ttestingtassert_array_equalR&R$(RGRQthdu1thdu2thdusRJtwR(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt:test_initialize_from_fits_with_data_in_different_extension s  c C`sÀtjjddƒ}tjjdd ƒ}tjƒ}tj|ƒ}tj|ƒ}tj|||gƒ}|jdƒj}|j|ƒt j |ddddƒ}tj j |j |ƒdS( NRidu afile.fitsRHiRuadu(idid(idid(RR RR>RWR?R@RARBRRCRYRZR&( RGt fake_img1t fake_img2thdu0R[R\R]RJR(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt(test_initialize_from_fits_with_extension°s  cC`sU|j}|jƒ}d|djks.t‚|djd|jƒksQt‚dS(Nubuniti(Rtto_hduR$R5RO(R)RGtccd_unitRI((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_write_unit_to_hdu¾s  c C`s‡|jdƒj}|j|ƒtjtƒ!tj|d|jdt ƒWdQXtjtƒ!tj|d|jdt ƒWdQXdS(Nu test.fitsRtdo_not_scale_image_datat scale_back( R@RAtwriteR*R+R,RRCRtTrue(R)RGRJ((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt2test_initialize_from_FITS_bad_keyword_raises_errorÅs  cC`sT|jdƒj}|j|ƒtj|d|jƒ}tjj|j |j ƒdS(Nu test.fitsR( R@RARiRRCRRRYRZR&(R)RGRJtccd_disk((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_ccddata_writerÒs cC`sbd}d|j|<|jƒ|jks.t‚|jƒ|jksIt‚||jks^t‚dS(NuSoMeKEYi (R3tlowerR5tupper(R)R((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt#test_ccddata_meta_is_case_sensitiveÚs  cC`s0idd6|_t|jtjƒ s,t‚dS(Nu Edwin HubbleuOBSERVER(R3t isinstanceRtHeaderR5(R)((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt$test_ccddata_meta_is_not_fits_headerâscC`sÐtj|ƒ}tjd|jƒ}tj|ƒ}|j|ƒ|jdƒj}|j|ƒtj |dt j ƒ}t j j|j|jƒtj |dddt j ƒ}t j j|jd|jƒdS(Niu afile.fitsRRHi(RR>R&R?tappendR@RARBRRCR"R<RRYRZ(R)RGRHR\RIRJRK((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt test_fromMEFçs  cC`s}tjjƒ}d|dR$RXR5R@RARBRRC(RGRQRHttmp_filetccd_read((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt1test_history_preserved_if_metadata_is_fits_header0s    cC`s||jdƒ}|j|jƒtjdƒd}tjƒ8}tj|jd|ƒ}||djksrt ‚WdQXdS(Nu temp.fitsuINFOuphotonRi( R@RiRAR tsetLevelt log_to_listRRCtmessageR5(R)RGttmpfiletexplicit_unit_nametlog_listt ccd_from_disk((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt(test_infol_logged_if_unit_in_fits_header>s c C`sQ|jdƒ}tddƒ}tj|jƒd|j_tjddgƒ|j_ddg|j_dd g|j_ |jj dgƒ|j ƒdj |_ |j j |jƒd tƒ|j|jƒt|j ƒ}tj|jƒ}|jdk st‚|jj|jkst‚|j ƒd}t|j ƒ|ksJt‚t|jd |ddƒ}|j ƒd}|jƒ} xX| jƒD]J} | dkr¦qŽn| |j ks»t‚|j | | | ksŽt‚qŽW|jjjd9_|j ƒd} | j d|jjjdks't‚| j d|jjjd ksMt‚dS(u$ Check that WCS attribute gets added to header, and that if a CCDData object is created from a FITS file with a header, and the WCS attribute is modified, then the CCDData object is turned back into an hdu, the WCS object overwrites the old WCS information in the header. u temp.fitstnaxisigNñ¸¨±¿gNñ¸¨±?ii¦ÿÿÿuRA---AIRuDEC--AIRig€F@t useblanksR¡RuaduuuCOMMENTuHISTORYuCDELT1uCDELT2N(iig€F@(uuCOMMENTuHISTORY(R@R RR‚R8R¡tcrpixtcdelttcrvaltctypetset_pvRdR$textendt to_headertFalseRiRARXRRCR1R5R&tkeys( R)RGRÀR¡toriginal_header_lengthtccd_newt ccd_new_hdutccd_wcs_not_in_headerRHt wcs_headerRLtccd_new_hdu_mod_wcs((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_wcs_attributeHs8  !&cC`sddlm}t|ƒ}tdƒ}tj|ƒ}|jjƒ}t|ƒt|jƒ|@ sjt ‚tdƒ}tj|ddƒ}dS(ut Test, for the file included with the nddata tests, that WCS keywords are properly removed from header. i(t_KEEP_THESE_KEYWORDS_IN_HEADERudata/sip-wcs.fitsu+../../io/fits/tests/data/o4sp040b0_raw.fitsRucountN( R²R×tsetRRRCR¡RÍR3R5(R×tkeeperst data_fileR(RÔt data_file1((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt%test_wcs_keywords_removed_from_header„s  $ c C`sWddlm}ddlm}t|ƒ}tdddƒ}x|D] }d|ksEd|ksEd |kruqEnt|ƒ}t|ƒ}|jd tƒ}||ƒ\}} t|ƒt| jd tƒƒ|@ sÞt ‚| jd tƒ} x\| j ƒD]N\} } t | t j ƒr4|| | ksKt ‚qýtjj|| | ƒqýWqEWd S( u± Test, for the WCS test files, that keyword removall works as expected. Those cover a much broader range of WCS types than test_wcs_keywords_removed_from_header i(t_generate_wcs_and_update_header(R×u../../wcs/tests/datatpatternu*.hdruinvalidu nonstandardusegfaulttrelaxN(R²RÝR×RØRRR RÍRjR5RFRqRt string_typesRRYtassert_almost_equal( RÝR×RÙt wcs_headersRvt header_stringR¡R$t new_headertnew_wcstnew_wcs_headerRLRM((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt+test_wcs_keyword_removal_for_wcs_test_files–s(    $   cC`s±tjdƒ}ttƒtjj|ddƒ}WdQXtjtjt j d ƒd|ƒgƒ}|j dƒj }|j |ƒtj|dd ƒ}|jdks­t‚dS( Nu< SIMPLE = T / Fits standard BITPIX = 16 / Bits per pixel NAXIS = 2 / Number of axes NAXIS1 = 1104 / Axis length NAXIS2 = 4241 / Axis length CRVAL1 = 164.98110962 / Physical value of the reference pixel X CRVAL2 = 44.34089279 / Physical value of the reference pixel Y CRPIX1 = -34.0 / Reference pixel in X (pixel) CRPIX2 = 2041.0 / Reference pixel in Y (pixel) CDELT1 = 0.10380000 / X Scale projected on detector (#/pix) CDELT2 = 0.10380000 / Y Scale projected on detector (#/pix) CTYPE1 = 'RA---TAN' / Pixel coordinate system CTYPE2 = 'WAVELENGTH' / Pixel coordinate system CUNIT1 = 'degree ' / Units used in both CRVAL1 and CDELT1 CUNIT2 = 'nm ' / Units used in both CRVAL2 and CDELT2 CD1_1 = 0.20760000 / Pixel Coordinate translation matrix CD1_2 = 0.00000000 / Pixel Coordinate translation matrix CD2_1 = 0.00000000 / Pixel Coordinate translation matrix CD2_2 = 0.10380000 / Pixel Coordinate translation matrix C2YPE1 = 'RA---TAN' / Pixel coordinate system C2YPE2 = 'DEC--TAN' / Pixel coordinate system C2NIT1 = 'degree ' / Units used in both C2VAL1 and C2ELT1 C2NIT2 = 'degree ' / Units used in both C2VAL2 and C2ELT2 RADECSYS= 'FK5 ' / The equatorial coordinate system tsepu i‘iPR$u afile.fitsRuadu(i‘iP(ttextwraptdedentR R RRrt fromstringR?R>RRUR@RARBRRCR¡R1R5(RGthdr_txt_example_WCSRvthdulRJR(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_read_wcs_not_creatable¸s  * cC`s8idd6}t|d|ƒ}|j|ks4t‚dS(NuHubbleuObserverR$(RR3R5(R)R°R(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt test_headerßs cC`s1d|_|jdƒ}|jdks-t‚dS(Nigð?(R¡R›R5(R)R•((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_wcs_arithmeticås u operationcC`sa|jƒ}d|_|j|ƒ}||ƒ}|j|jksHt‚|jdks]t‚dS(Ni(RR¡R‹R5R1(R)R‘t ccd_data2R“R•((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_wcs_arithmetic_ccdës    cC`stdƒ}d„}tj|ƒ}tj|ƒ}||djƒ}WdQXt|ƒsat‚|jƒ}||djƒ}t|ƒs’t‚|jdt ƒ}||djƒ}t |ƒ sÊt‚|djddksçt‚|djdd kst‚dS( uc Check whether the ctypes RA---TAN-SIP and DEC--TAN-SIP survive a roundtrip unchanged. udata/sip-wcs.fitscS`sAidd6dd6}g|jƒD]\}}|||k^q!S(Nu RA---TAN-SIPuCTYPE1u DEC--TAN-SIPuCTYPE2(RF(R$texpected_wcs_ctypesRLRM((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pytcheck_wcs_ctypesýs iNt wcs_relaxuCTYPE1uRA---TANuCTYPE2uDEC--TAN( RRRCRtopenR$tallR5RdRÎtany(RÚRôt ccd_originaltrawt good_ctypeRÑt ccd_no_relax((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_wcs_sip_handlingös   cC`sV|jƒ}|jdk|_|j|ƒ}||ƒ}tjj|j|jƒdS(Ni(RR&tmaskR‹RRYt assert_equal(R)R‘RñR“R•((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_mask_arithmetic_ccds   cC`sr|jdk|_|jdƒj}|j|ƒtj|ƒ}|jdk sUt‚t j j |j|jƒdS(Ni u afile.fits( R&RþR@RARiRRCR1R5RRYRZ(R)RGRJt ccd_after((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt/test_write_read_multiextensionfits_mask_default%s  cC`s|t|jdƒ|_|jdƒj}|j|ƒtj|ƒ}|jdk sYt ‚t j j |jj |jj ƒdS(Ni u afile.fits(RR&R}R@RARiRRCR1R5RRYRZR‚(R)RGRJR((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt6test_write_read_multiextensionfits_uncertainty_default/s cC`s|jdk|_t|jdƒ|_|jdƒj}|j|ddddƒtj |ƒ}|jdkswt ‚|jdksŒt ‚dS(Ni u afile.fitsthdu_maskthdu_uncertainty( R&RþRR}R@RARiR1RRCR5(R)RGRJR((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt&test_write_read_multiextensionfits_not:scC`s |jdk|_t|jdƒ|_|jdƒj}|j|ddddƒtj|ƒ}|jdkswt ‚|jdksŒt ‚tj|ddddƒ}|jdk s¼t ‚|jdk sÑt ‚t j j |j|jƒt j j |jj|jjƒdS(Ni u afile.fitsRuFunRuNoFun(R&RþRR}R@RARiRRCR1R5RRYRZR‚(R)RGRJR((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt3test_write_read_multiextensionfits_custom_ext_namesEscC`s"d|_|jdkst‚dS(Ni(R¡R5(R)((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_wcsZs cC`s|dddg}xf|D]^}|jdj|ƒƒ}|j|jƒtj|jƒ}|j|jkjƒst‚qWdS(Nufitufitsuftsutest.{}( R@tformatRiRARRCR&R÷R5(R)RGtsupported_extensionstexttpatht from_disk((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt+test_recognized_fits_formats_for_read_write_s  cC`s0tjtƒttjdƒƒjWdQXdS(Ni (i i (R*R+RRRRUt parent_nddata(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt2test_stddevuncertainty_compat_descriptor_no_parentjscC`s[ttjdƒddƒ}ttjdƒƒ}||_|j|ksNt‚d|_dS(Ni Ru(i i (i i (RRRURt_parent_nddataRR5R1(R(tuncert((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyt3test_stddevuncertainty_compat_descriptor_no_weakrefos  cC`s¹ttjdƒjddƒƒ}tjdƒ}tjdtjƒtj|jƒƒtj |ƒgƒ}|j dƒj }|j |ƒt j|ddƒ}|jjd ksµt‚dS( Ni iiR]utable_image.fitsRuadu(ii(ii(RRRUtreshapeRR?R>tTableHDUtas_arrayRWR@RARBRRCR&R8R5(RGttbltimgRíRJR(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pyttest_read_returns_image{s!$ (ot __future__RRRRRétnumpyRR*texternRtioRt nduncertaintyRRtRR"R R¡R R t tests.helperR tutilsR t utils.dataRRRR²Rt astropy.tableRRt DEFAULT_SEEDtDEFAULT_DATA_SIZEtDEFAULT_DATA_SCALERtfixtureR)R-R0R2R7tmarkt data_sizeR;R=RNRSRTRVR_RcRfRkRmRpRsRuRxRzR{R|R~R„R…R‡R‰t parametrizeRjtdimensionless_unscaledRPR#RÎt data_unitR˜t UnitsErrorRšRŸR¢R§R©R¬R´R¶R·R¸R¹R¼RÄRÖRÜRçRîRïRðRòRýRRRRRRRRRR(((s@lib/python2.7/site-packages/astropy/nddata/tests/test_ccddata.pytsÒ"                              ''      ')    $     <  " '    %