U e»{j$žã @s<ddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddlmZddlmZmZmZmZddlmZdd lmZdd lmZmZmZdd lmZdd lm Z m!Z!d Z"edƒ&ej#j$e"e"gdZ%ej#j$ddZ&W5QRXej'dd„ƒZ(dd„Z)dd„Z*dd„Z+dd„Z,dd„Z-dd„Z.dd „Z/d!d"„Z0d#d$„Z1d%d&„Z2d'd(„Z3d)d*„Z4d+d,„Z5d-d.„Z6d/d0„Z7d1d2„Z8d3d4„Z9d5d6„Z:d7d8„Z;d9d:„Z„Z>d?d@„Z?dAdB„Z@dCdD„ZAdEdF„ZBdGdH„ZCdIdJ„ZDdKdL„ZEdMdN„ZFdOdP„ZGejH IdQdRdSg¡ejH IdTdUdVejJdVejKejLg¡ejH IdWdXdYg¡dZd[„ƒƒƒZMejH IdQd\d]g¡ejH Id^dUejNfdVejJejNfdVejLdYfg¡ejH IdWdXdYg¡d_d`„ƒƒƒZOdadb„ZPdcdd„ZQdedf„ZRdgdh„ZSdidj„ZTdkdl„ZUdmdn„ZVdodp„ZWdqdr„ZXdsdt„ZYdudv„ZZdwdx„Z[dydz„Z\d{d|„Z]d}d~„Z^ejH _d¡d€d„ƒZ`d‚dƒ„Zad„d…„Zbd†d‡„ZcejH Idˆd‰dŠd‹dŒg¡ddŽ„ƒZddd„ZeejH Idˆd‰dŠd‹dŒg¡d‘d’„ƒZfd“d”„ZgejH Id•eeeg¡d–d—„ƒZhejH Id•eeeg¡d˜d™„ƒZidšd›„Zjdœd„ZkdždŸ„Zld d¡„Zmd¢d£„Znd¤d¥„Zod¦d§„Zpd¨d©„Zqdªd«„Zrd¬d­„Zsd®d¯„Ztd°d±„Zud²d³„ZvdS)´éN)Úlog)Úunits)Úfits)Ú_testing)ÚCCDData)ÚInverseVarianceÚMissingDataAssociationExceptionÚStdDevUncertaintyÚVarianceUncertainty)ÚTable)ÚNumpyRNGContext)Úget_pkg_data_contentsÚget_pkg_data_filenameÚget_pkg_data_filenames)ÚAstropyWarning)ÚWCSÚFITSFixedWarningédé{)Úsize)ércCs$| dt|ƒ¡| dt|ƒ¡dS)zÆ Pytest fixture to run a test case with tilde-prefixed paths. In the tilde-path case, environment variables will be temporarily modified so that '~' resolves to the temp directory. ÚHOMEÚ USERPROFILEN)ZsetenvÚstr)Útmp_pathZ monkeypatchÚrequest©rú`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/nddata/tests/test_ccddata.pyÚhome_is_tmpdir%s rcCs*t ¡}dddœ}t|tjd}||_|S)zf Return a CCDData object of size DEFAULT_DATA_SIZE x DEFAULT_DATA_SIZE with units of ADU. é*znot 42)Zmy_keyZyour_key©Úunit)Ú _random_arrayÚcopyrÚuÚaduÚheader)ÚdataZ fake_metaÚccdrrrÚcreate_ccd_data3s  r)c Cs t t¡ tƒW5QRXdS©N)ÚpytestÚraisesÚ TypeErrorrrrrrÚtest_ccddata_empty?s r.c Cs,t t¡tt ddg¡ƒW5QRXdS)Né)r+r,Ú ValueErrorrÚnpÚzerosrrrrÚtest_ccddata_must_have_unitDs r3c Cs&tƒ}t t¡ d|_W5QRXdSr*)r)r+r,r-r!©Úccd_datarrrÚ'test_ccddata_unit_cannot_be_set_to_noneIs r6c Cs<tjtdd$tdddgdddiddidW5QRXdS)Nz#.*can't have both header and meta.*©Úmatchér/éÚ)r!Úmetar&)r+r,r0rrrrrÚ!test_ccddata_meta_header_conflictOsr=cCsBtƒ}|jttfkst‚|jttks*t‚|jt t¡ks>t‚dSr*)r)ÚshapeÚDEFAULT_DATA_SIZEÚAssertionErrorrÚdtyper1Úfloatr4rrrÚtest_ccddata_simpleTsrCcCs*tt ddg¡dd}|jtjks&t‚dS)Nr/Úelectronr )rr1r2r!r$rDr@©r(rrrÚ+test_ccddata_init_with_string_electron_unit[srFcCs¦tƒ}t |¡}t |g¡}t|dƒ}| |¡tj|tj d}|j t t fksTt ‚|j t t ksft ‚t |jjtj¡s|t ‚|j ¡D]\}}|j||ks†t ‚q†dS)Nú afile.fitsr )r)rÚ PrimaryHDUÚHDUListrÚwritetorÚreadr$rDr>r?r@rr1Z issubdtyper'rAZfloatingr&Úitemsr<)rr5ÚhduÚhdulistÚfilenameÚcdÚkÚvrrrÚtest_initialize_from_FITS`s    rScCszt ddg¡}t |¡}tj ¡|jd<t|dƒ}|  |¡t   |¡}|j tjksXt ‚t j |dd}|j tjksvt ‚dS)Nr/ÚbunitrGÚphotonr )r1r2rrHr$r%Ú to_stringr&rrJrrKr!r@rU)rÚfake_imgrMrOr(Úccd2rrrÚ-test_initialize_from_fits_with_unit_in_headerns    rYcCsVt ddg¡}t |¡}d|jd<t|dƒ}| |¡t |¡}|j t j ksRt ‚dS)Nr/ZADUrTrG) r1r2rrHr&rrJrrKr!r$r%r@)rrWrMrOr(rrrÚ,test_initialize_from_fits_with_ADU_in_header}s     rZc CsTt t d¡¡}d|jd<t|dƒ}| |¡t t ¡t   |¡W5QRXdS)N©r/r/zdefinetely-not-a-unitrTrG) rrHr1Úonesr&rrJr+r,r0rrK)rrMrOrrrÚ5test_initialize_from_fits_with_invalid_unit_in_headerˆs     r]cCsTt t d¡¡}d|jd<t|dƒ}| |¡t |¡}|j t j t j ksPt ‚dS)Nr[z ELECTRONS/SrTrG)rrHr1r\r&rrJrrKr!r$rDÚsr@)rrMrOr(rrrÚAtest_initialize_from_fits_with_technically_invalid_but_not_really‘s     r_cCs€t d¡ dd¡}t ¡}t |¡}t ||g¡}t|dƒ}| |¡t j |dd}tj   |j |¡|j|j|jks|t‚dS)Nér/rGr%r )r1ÚarangeÚreshaperrHÚImageHDUrIrrJrrKÚtestingÚassert_array_equalr'r&r@)rrWÚhdu1Úhdu2ÚhdusrOr(rrrÚ:test_initialize_from_fits_with_data_in_different_extensionšs   ric CsÒt ddg¡}t d¡ dd¡}t ¡}tj|ddd}tj|ddd}t |||g¡}t|dƒ}|  |¡t j |ddd }tj   |j|¡t j |ddd }tj   |j|¡t j |d dd }tj   |j|¡dS) Nr/r`Úfirstr9)ÚnameÚverÚsecondrGr%©rMr!)rmr9)r1r2rarbrrHrcrIrrJrrKrdrer') rZ fake_img1Z fake_img2Zhdu0rfrgrhrOr(rrrÚ(test_initialize_from_fits_with_extension¨s  rocCsDtƒ}|j}| ¡}d|djks&t‚|djd| ¡ks@t‚dS)NrTr)r)r!Úto_hdur&r@rV)r5Zccd_unitrNrrrÚtest_write_unit_to_hdu¼s rqc Csptƒ}t|dƒ}| |¡t t¡tj||jddW5QRXt t¡tj||jddW5QRXdS)Nú test.fitsT)r!Zdo_not_scale_image_data)r!Z scale_back) r)rÚwriter+r,r-rrKr!)rr5rOrrrÚ2test_initialize_from_FITS_bad_keyword_raises_errorÄs    rtcCsBtƒ}t|dƒ}| |¡tj||jd}tj |j |j ¡dS)Nrrr ) r)rrsrrKr!r1rdrer'©rr5rOÚccd_diskrrrÚtest_ccddata_writerÑs   rwc Csžtƒ}t|dƒ}|j|ddt |¡}t|ƒdkstƒ}d|_t t |j¡¡}| ¡|_tj |jj |¡dSr*) r)r–r1ÚsqrtÚabsr'r#rdreÚarray)r5Zfake_uncertaintyrrrÚ#test_setting_uncertainty_with_array/s  r›c Cs0tƒ}t t¡t ddg¡|_W5QRXdS)Nr:r`)r)r+r,r0r1r2r–r4rrrÚ1test_setting_uncertainty_wrong_shape_raises_error7s rœcCsltƒ}ddi|_| ¡}t|tjƒs(t‚|j ¡D]\}}|dj||ks2t‚q2t j   |dj |j ¡dS)Nrˆrƒr) r)r<rpr{rrIr@rLr&r1rdrer')r5Ú fits_hdulistrQrRrrrÚ test_to_hdu=s ržcCsJtƒ}|jdd}t|dtjƒs&t‚|jdd}t|dtjƒsFt‚dS)NFrxrT)r)rpr{rrHr@rc)r5rrrrÚtest_to_hdu_as_imagehduGs   rŸcCsDtƒ}| ¡}tj |j|j¡|j|jks0t‚|j|jks@t‚dSr*) r)r#r1rdrer'r!r@r<)r5Zccd_copyrrrÚ test_copyOs r zoperation,affects_uncertainty)ÚmultiplyT)ÚdivideTÚoperandg@r/Úwith_uncertaintyTFc Cs@tƒ}|rtt |¡ƒ|_t||ƒ}tt|ƒ}||ƒ}||k sBt‚t|tƒsPt‚|jdksjt|jtƒsjt‚z |j }Wnt k rŒ|}YnXtj   |j ||j |ƒ¡|rä|rÌtj   |jj||jj|ƒ¡qòtj   |jj|jj¡n|jdksòt‚t|tjƒr*|d|jd|jƒj} |j| ks.return_truer/©Znaxisr¶r;©r!r¸r·) Ú nd_testingZcreate_two_equal_wcsrr1r\Úassert_wcs_seem_equalr±r¸r²r¡r¢)r½Zwcs1Zwcs2Úccd1rXrrrÚ test_arithmetic_with_wcs_compareâs"ÿÿÿrÃc CsÈdd„}tt d¡dtƒd}tt d¡dtƒd}t t¡|j||dW5QRXt t¡|j||dW5QRXt t¡|j ||dW5QRXt t¡|j ||dW5QRXdS)NcSsdS)NFrrºrrrÚ return_falseösz;test_arithmetic_with_wcs_compare_fail..return_falser¶r;r¿r·) rr1r\rr+r,r0r±r²r¡r¢)rÄrÂrXrrrÚ%test_arithmetic_with_wcs_compare_failõs    rÅcCsŽtƒ}tt |¡ƒ|_| ¡}| |¡}t|jƒdks:t ‚tj   |j d|j ¡tj   |jjt d¡|jj¡| |¡}t|jƒdksŒt ‚tj   |j d|j ¡tj   |jjt d¡|jj¡| |¡}t|jƒdksÞt ‚tj   |j |j d¡t d¡t |j ¡|jj}tj  |jj|¡| |¡}t|jƒdksBt ‚tj   |j t |j ¡¡t d¡t |j ¡|jj}tj  |jj|¡dS)Nrr/)r)r r1r¥r–r#r±rzr<r@rdrer'Zassert_array_almost_equal_nulpršr˜r²r¡r™Zassert_allcloser¢)r5r£r®Zexpected_uncertaintyrrrÚ(test_arithmetic_overload_ccddata_operands: ÿ ÿ ÿ ÿrÆcCs*t dddg¡tj}t dddg¡tj}t|ƒ}| |¡}tj |j t ||¡j ¡|j t ||¡j kspt ‚|  |¡}tj |j t  ||¡j ¡|j t  ||¡j ks¬t ‚| |¡}tj |j t ||¡j ¡|j t ||¡j ksèt ‚| |¡}tj |j t ||¡j ¡|j t ||¡j ks&t ‚dS)Nr9r/r:)r1ršr$ÚmÚcmrr±rdZassert_array_almost_equalr'r§r!r@r²r¡r¢)ÚaÚbZccddataÚresrrrÚ(test_arithmetic_overload_differing_units(s    rÌc Csztt d¡dd}| t d¡¡}tj |jdddggd¡tt d¡dd}t  t ¡| t d¡¡W5QRXdS)N©r:r:r;r r:r9r/r%) rr1r\r±rardrer'r+r,r0©r(rËrrrÚtest_arithmetic_add_with_array?s  rÏc Csztt d¡dd}| t d¡¡}tj |jdddggd¡tt d¡dd}t  t ¡| t d¡¡W5QRXdS) NrÍr;r r:r9réÿÿÿÿr%) rr1r\r²rardrer'r+r,r0rÎrrrÚ#test_arithmetic_subtract_with_arrayIs  rÑcCs\tt d¡dtjd}| t d¡d¡}tj |jdddggd¡|j |j ksXt ‚dS)NrÍr:r r/é) rr1r\r$rÇr¡rdrer'r!r@rÎrrrÚ#test_arithmetic_multiply_with_arraySsrÓcCsXtt d¡tjd}| t d¡d¡}tj |jdddggd¡|j |j ksTt ‚dS)NrÍr r/gà?r:) rr1r\r$rÇr¢rdrer'r!r@rÎrrrÚ!test_arithmetic_divide_with_arrayZsrÔcCsŒt ddg¡}t |¡}d|jd<d|jd<d|jd<t|jdƒdksLt‚t|dƒ}| |¡t j |dd }|jd|jdksˆt‚dS) Nr/ÚoneÚhistoryÚtwoÚthreer:ú temp.fitsr%r ) r1r2rrHr&rzr@rrJrrK)rrWrMZtmp_fileZccd_readrrrÚ1test_history_preserved_if_metadata_is_fits_headeras      rÚc Csbtƒ}t|dƒ}| |¡t d¡d}t ¡&}tj||d}||djksTt ‚W5QRXdS)NrÙÚINFOrUr r) r)rrsrÚsetLevelZ log_to_listrrKÚmessager@)rr5ÚtmpfileZexplicit_unit_nameZlog_listr»rrrÚ(test_infol_logged_if_unit_in_fits_headeros    rßc Cs°tƒ}t|dƒ}tdd}t |j¡d|j_t ddg¡|j_ddg|j_ dd g|j_ |j  d g¡|  ¡dj |_ |j j| ¡d d | |¡t|j ƒ}t |¡}|jd k s¸t‚|jj|jksÊt‚|  ¡d}t|j ƒ|ksèt‚t|j|dd}|  ¡d}| ¡} |  ¡D]<} | dkr&q| |j ks6t‚|j | | | kst‚q|jjjd9_|  ¡d} | j d|jjjdksŽt‚| j d|jjjdks¬t‚d S)a$ 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. rÙr/r¾gNñ¸¨±¿gNñ¸¨±?ri¦ÿÿÿzRA---AIRzDEC--AIR)r/r9g€F@F)Z useblanksNr%©r¸r!)r;ÚCOMMENTZHISTORYZCDELT1ZCDELT2r9)r)rrr1ršr>r¸ÚcrpixZcdeltZcrvalÚctypeZset_pvrpr&ÚextendÚ to_headerrsrzrrKr@r'Úkeys) rr5rÞr¸Zoriginal_header_lengthÚccd_newZ ccd_new_hduZccd_wcs_not_in_headerrMÚ wcs_headerrQZccd_new_hdu_mod_wcsrrrÚtest_wcs_attributezs:            réc Cs˜ddlm}t|ƒ}tdƒ}t |¡}tjtdd|j   ¡}W5QRXt|ƒt|j ƒ|@rdt ‚tdƒ}tjt ddtj|dd }W5QRXd S) zt Test, for the file included with the nddata tests, that WCS keywords are properly removed from header. r)Ú_KEEP_THESE_KEYWORDS_IN_HEADERúdata/sip-wcs.fitsz,Some non-standard WCS keywords were excludedr7z+../../io/fits/tests/data/o4sp040b0_raw.fitsz'unitfix' made the changeÚcountr N)Úastropy.nddata.ccddatarêÚsetrrrKr+Zwarnsrr¸rår<r@r)rêÚkeepersÚ data_filer(rèZ data_file1rrrÚ%test_wcs_keywords_removed_from_header·s  ÿrñc Cshtdƒ}ddg}t |¡"}|D]}||djks t‚q W5QRXt |¡}|D]}||jksPt‚qPdS)NrëZA_0_0ZB_0_1r)rrryr&r@rrK)rðZ test_keysrMr€r(rrrÚ)test_wcs_SIP_coefficient_keywords_removedÎs  ròÚignorecCs2ddlm}m}m}m}t|ƒ}tddd}|D]ü}d|ks0d|ks0d|ks0d |krVq0t|ƒ}tj   |¡}t |ƒ} | j d d } ||ƒ\} } | j d d } t| ƒt| ƒ|@r®t ‚| | }t|ƒ}||@rÔ||@rèt ‚n||@rè||@rèt ‚|  ¡D]:\}}t|tƒr| ||ks*t ‚qðtj | ||¡qðq0d S) zÙ Test, for the WCS test files, that keyword removal works as expected. Those cover a much broader range of WCS types than test_wcs_keywords_removed_from_header. Includes regression test for #8597 r)rêÚ_CDsÚ_generate_wcs_and_update_headerÚ_PCsz../../wcs/tests/dataz*.hdr)ÚpatternÚinvalidZ nonstandardZsegfaultzchandra-pixlist-wcsT)ZrelaxN)rírêrôrõrörîrr rr„Ú fromstringrrår@rLr{rr1rdZassert_almost_equal)rêrôrõrörïZ wcs_headersrZ header_stringr&r¸Zheader_from_wcsZ new_headerZnew_wcsZnew_wcs_headerZ final_headerZfinal_header_setrQrRrrrÚ+test_wcs_keyword_removal_for_wcs_test_filesës<  ÿþýü      rúcCslt d¡}tjj|dd}t tjt d¡|dg¡}t |dƒ}|  |¡t j |dd}|j dksht‚dS) Na< 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 Ú )Úsep)i‘iP©r&rGr%r )ÚtextwrapÚdedentrr„rùrIrHr1r\rrJrrKr¸r@)rZhdr_txt_example_WCSrÚhdulrOr(rrrÚtest_read_wcs_not_creatable*sÿ  rcCs,tƒ}ddi}t||d}|j|ks(t‚dS)NZObserverZHubblerý)r)rr<r@)r5rÉr(rrrÚ test_headerRs rcCs2tƒ}tdd}||_| d¡}t |j|¡dS)Nr/r¾gð?)r)rr¸r¡rÀrÁ)r5r¸r®rrrÚtest_wcs_arithmeticYs   rrªr¡r¢r±r²cCsNtƒ}| ¡}tdd|_t||ƒ}||ƒ}t |j|j¡|jdksJt‚dS©Nr/r¾)r)r#rr¸r¦rÀrÁr@©rªr5Z ccd_data2r¬r®rrrÚtest_wcs_arithmetic_ccdas  rc CsÂtdƒ}dd„}t |¡}t |¡}||djƒ}W5QRXt|ƒsJt‚| ¡}||djƒ}t|ƒslt‚|jdd}||djƒ}t |ƒr’t‚|djddks¨t‚|djd d ks¾t‚d S) zc Check whether the ctypes RA---TAN-SIP and DEC--TAN-SIP survive a roundtrip unchanged. rëcs dddœ}‡fdd„| ¡DƒS)Nz RA---TAN-SIPz DEC--TAN-SIP)ÚCTYPE1ÚCTYPE2csg|]\}}ˆ||k‘qSrr)Ú.0rQrRrýrrÚ vszCtest_wcs_sip_handling..check_wcs_ctypes..)rL)r&Zexpected_wcs_ctypesrrýrÚcheck_wcs_ctypesss z/test_wcs_sip_handling..check_wcs_ctypesrF)Z wcs_relaxrzRA---TANrzDEC--TANN) rrrKrryr&Úallr@rpÚany)rðr Z ccd_originalÚrawZ good_ctyperçZ ccd_no_relaxrrrÚtest_wcs_sip_handlingls      rcCsBtƒ}| ¡}|jdk|_t||ƒ}||ƒ}tj |j|j¡dS)Nr)r)r#r'Úmaskr¦r1rdZ assert_equalrrrrÚtest_mask_arithmetic_ccdŽs   rcCsVtƒ}|jdk|_t|dƒ}| |¡t |¡}|jdk s@t‚tj   |j|j¡dS©Nr}rG) r)r'rrrsrrKr@r1rdre©rr5rOÚ ccd_afterrrrÚ/test_write_read_multiextensionfits_mask_default˜s    rÚuncertainty_typecCsptƒ}||jdƒ|_t|dƒ}| |¡t |¡}|jdk sDt‚t|jƒ|ksVt‚t j   |jj |jj ¡dSr© r)r'r–rrsrrKr@Útyper1rdrerš©rrr5rOrrrrÚ6test_write_read_multiextensionfits_uncertainty_default£s   ÿrcCsxtƒ}||jdƒ|_t|dƒ}|j|ddtj|dd}|jdk sLt‚t|jƒ|ks^t‚t j   |jj |jj ¡dS)Nr}rGZBlah)Zkey_uncertainty_typerrrrrÚHtest_write_read_multiextensionfits_uncertainty_different_uncertainty_key´s ÿrcCshtƒ}|jdk|_t|jdƒ|_t|dƒ}|j|dddt |¡}|jdksVt ‚|jdksdt ‚dS)Nr}rG©Zhdu_maskZhdu_uncertainty) r)r'rr r–rrsrrKr@rrrrÚ&test_write_read_multiextensionfits_notÇs   rcCs¼tƒ}|jdk|_t|jdƒ|_t|dƒ}|j|dddt |¡}|jdksVt ‚|jdksdt ‚tj|ddd}|jdk s‚t ‚|jdk st ‚t j   |j|j¡t j   |jj |jj ¡dS)Nr}rGZFunZNoFunr)r)r'rr r–rrsrrKr@r1rdreršrrrrÚ3test_write_read_multiextensionfits_custom_ext_namesÓs    ÿrcCs|d}t ||g¡}|dk}t |¡}t|||dd}| ¡}|djd=t|dƒ}| |¡t |¡}t |j t ƒsxt ‚dS)Nr`gÍÌÌÌÌÌì?r%)r'rr–r!r/ZUTYPEzold_ccddata_mef.fits) r1r2r˜rrpr&rrJrKr{r–r r@)rrr'rÚuncertr(rNÚ file_namerrrÚ&test_read_old_style_multiextensionfitsês      r!cCs(tƒ}tdd}||_|j|ks$t‚dSr)r)rr¸r@)r5r¸rrrÚtest_wcss r"cCsXtƒ}dddg}|D]>}t|d|›ƒ}| |¡t |¡}|j|jk ¡st‚qdS)NÚfitrZftsztest.)r)rrsrrKr'r r@)rr5Zsupported_extensionsÚextÚpathZ from_diskrrrÚ+test_recognized_fits_formats_for_read_write s   r&c Cs*t t¡tt d¡ƒjW5QRXdS)Nr¶)r+r,rr r1r\Ú parent_nddatarrrrÚ2test_stddevuncertainty_compat_descriptor_no_parents r(cCs>tt d¡dd}tt d¡ƒ}||_|j|ks4t‚d|_dS)Nr¶r;r )rr1r\r Z_parent_nddatar'r@)r(rrrrÚ3test_stddevuncertainty_compat_descriptor_no_weakrefs r)cCs~tt d¡ dd¡ƒ}t d¡}tjt ¡t | ¡¡t  |¡gd}t |dƒ}|  |¡t j |dd}|jjdkszt‚dS) Nr}rŒr/r‹)rhztable_image.fitsr%r )r r1r\rbrrIrHZTableHDUZas_arrayrcrrJrrKr'r>r@)rZtblZimgrrOr(rrrÚtest_read_returns_image*s ÿ  r*cCs‚tdd}d|j_tt d¡|dd}|dd…dd…f}| ¡}t|tj ƒsRt ‚|dj dd ksht ‚|dj d d ks~t ‚dS) Nr/r¾r¶ZpixelràéþÿÿÿrZCRPIX1éZCRPIX2) rr¸rârr1r\rpr{rrIr@r&)r¸r(ZtrimmedrrrrÚtest_sliced_ccdata_to_hdu:s r-cCsTtƒ}tj dd¡}| |¡tj||jd}tj   |j |j ¡tj  |¡rPt ‚dS)Nú~rrr )r)Úosr%ÚjoinrsrrKr!r1rdrer'Úexistsr@)rr5rOrvrrrÚtest_read_write_tilde_pathsEs  r2c Cs`t ¡}tt ¡tj|d}|j|k ¡s.t‚t j t ddtt ¡tjddW5QRXdS)N)r!ÚpsfúThe psf must be a numpy array.r7Z something© Ú _random_psfr#rr"r$r%r3r r@r+r,r-©r3r(rrrÚtest_ccddata_with_psfSs r8c CsVt ¡}tt ¡tjd}||_|j|k ¡s2t‚t j t dd d|_W5QRXdS)Nr r4r7rŒr5r7rrrÚtest_psf_setter]s r9cCsÂtƒ}t|_|d}| |¡t |¡}tj |j |j ¡tj |j|j¡|d}|j|ddt |¡}tj |j |j ¡|jdksŒt ‚tj|dd}tj |j |j ¡tj |j|j¡dS)zATest that we can round-trip a CCDData with an attached PSF image.ztest_write_read_psf.fitsztest_write_read_psf_hdu.fitsZPSFOTHER)Zhdu_psfN) r)r6r3rsrrKr1rdrer'r@rurrrÚtest_write_read_psfhs   r:)wr/rþÚnumpyr1r+Zastropyrrr$Z astropy.iorZastropy.nddatarrÀrírZastropy.nddata.nduncertaintyrrr r Z astropy.tabler Z astropy.utilsr Zastropy.utils.datar rrZastropy.utils.exceptionsrZ astropy.wcsrrr?ÚrandomÚnormalr"r6Zfixturerr)r.r3r6r=rCrFrSrYrZr]r_rirorqrtrwr|rr…r‡r’r“r”r•r—r›rœržrŸr ÚmarkZ parametrizeZdimensionless_unscaledrUr%r°Z UnitsErrorr³rµr¹rÃrÅrÆrÌrÏrÑrÓrÔrÚrßrérñròÚfilterwarningsrúrrrrrrrrrrrr!r"r&r(r)r*r-r2r8r9r:rrrrÚs                    þþýþ(þþ ýþ,#   =  >( "  ÿ ÿ