U e{j"@sddlZddlZddlmZddlmZddlZddlZddlm Z m Z m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZdd lm Z m!Z!m"Z"ddl#m$Z$ddl%m&Z&ee&j'Z(ddZ)GdddZ*GdddZ+ddZ,ej-.dddZ/ddZ0ddZ1dd Z2d!d"Z3d#d$Z4d%d&Z5d'd(Z6d)d*Z7d+d,Z8d-d.Z9d/d0Z:d1d2Z;d3d4Zd9d:Z?ej-.d;dd?ZAd@dAZBdBdCZCddIdJZDdKdLZEdMdNZFdOdPZGdQdRZHej-.d;dSdTZIdUdVZJdWdXZKdYdZZLd[d\ZMd]d^ZNd_d`ZOdadbZPdcddZQdedfZRdgdhZSej-jTe&j'ddikdjdkdldmZUej-jTe&j'ddikdjdkdndoZVdpdqZWdrdsZXdtduZYdvdwZZdxdyZ[dzd{Z\d|d}Z]d~dZ^ddZ_ddZ`ddZaddZbddZcddZdddZeddZfddZgej-.dddZhddZiGdddZjddZkddZlddZmddZnej-jTe(edkddkddZoej-jTe(edkddkddZpddZqdS)N) nullcontext)datetime)assert_allcloseassert_array_almost_equalassert_array_almost_equal_nulpassert_array_equal)Version)units)wcs)SkyCoord)fits)Cutout2D)assert_quantity_allclose)get_pkg_data_contentsget_pkg_data_filenameget_pkg_data_filenames)AstropyDeprecationWarningAstropyUserWarningAstropyWarning)NumpyRNGContext)_wcscCs4ttdkr*ttdkr*tjtjdd}nt}|S)N7.17.3zF'datfix' made the change 'Set DATE-REF to '1858-11-17' from MJD-REF'\.match) _WCSLIB_VERrpytestwarnsr FITSFixedWarningr)ctxr Y/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/wcs/tests/test_wcs.pyctx_for_v71_dateref_warnings)sr"c@s$eZdZddZddZddZdS)TestMapscCsttddd|_dS)Nz data/maps*.hdrpatternlistr _file_listselfr r r! setup_method8szTestMaps.setup_methodcCs0d}t|j|ks,tdt|jd|dS)N0test_spectra has wrong number data files: found , expected lenr)AssertionErrorr+Z n_data_filesr r r!test_consistency<szTestMaps.test_consistencycCs|jD]~}tj|}ttjdd|dd}t|}|ddggd}t |ddggdd | ddggd}t |ddggd d qdS) Ndatamapsbinaryencodingagq@gP)decimalr) r)ospathbasenamerjoinr WCS wcs_pix2worldr wcs_world2pix)r+filenameheaderwcsobjZworldZpixr r r! test_mapsIs   zTestMaps.test_mapsN)__name__ __module__ __qualname__r,r4rGr r r r!r#7s r#c@s$eZdZddZddZddZdS) TestSpectracCsttddd|_dS)Nz data/spectrar$r%r'r*r r r!r,[szTestSpectra.setup_methodcCs0d}t|j|ks,tdt|jd|dS)Nr.r/r0r3r r r!r4^szTestSpectra.test_consistencyc Cs|jD]v}tj|}ttjdd|dd}ttdkrLtj t j dd}nt }|t |}W5QRXt|dkstqdS) Nr5Zspectrar7r87.4zG'datfix' made the change 'Set MJD-OBS to 53925\.853472 from DATE-OBS'\.r )r)r=r>r?rr@rrrrr rr find_all_wcsr1r2)r+rDrErZall_wcsr r r! test_spectraks   zTestSpectra.test_spectraN)rHrIrJr,r4rPr r r r!rKZs rKc Cstddd}ttj,ttj}tj|ddW5QRXW5QRXtdt krftdkrnn(t |dkszt d t | jkst nt |d kst t |d j}d |krd |krd|kst dt |djkrdt |djkst dS)z From github issue #36 data/nonstandard_units.hdrr7r8Zdhs)Ztranslate_unitsrM7.6z#'datfix' made the change 'Success'.rZunitfixHzzM/Sz plane angler;m/sN)rrraisesr InvalidTransformErrorrrrArrr1r2strpopmessage)rEwZ first_wmsgr r r! test_fixess " r]z ignore:PV2_2cCstddd}t|}tt|ddggds8ttt|ddggdsZttt|ddggdr|td S) z From github issue #107 zdata/outside_sky.hdrr7r8gY@g@@rgi@g@@N) rr rAnpallisnanrBr2anyrEr\r r r!test_outside_skys   ""rcc Cstd}ttj}t|}W5QRXtdtkrFtdkr\nnt|dkslt nt|dkslt d}t |t d|fj }|j|ddd }|j|d |d ddd t d d gddgg}t j|jj|ddst t ddgddgddgg}t j||dddst dS)z! From github issue #1463 zdata/sip2.fitsrMrRrTr;rSrT)Z ra_dec_order.r.r;g@'G^0?g$+.?gxƠ3.?g\a0:0yE>)atolt||d|d|d d}t |dksjtW5QRXd S) rrSnaxisrrr;)rrSrdre).rTN) r rArr^rrrBshaper2r1)r\r5rpr r r!test_3d_shapess   rcCstjdd}tjd}tjd}|||d\}}|jdksDt|jdksRt|||d\}}|jdksrt|jdkstdSNrTrrTrSr;)r rAr^rrBrr2rC)r\xyxwywrrr r r!test_preserve_shape!s   rcCsNtjdd}tjd}d}|||d\}}|jdkss       rc Csxd}tjdd}tjd}tjt|d||dW5QRXtjd}tjt|d||dW5QRXdS) z Issue #1395zCWhen providing two arguments, the array must be of shape [(]N, 2[)]rTr)rTrSrr;)rTr;N)r rAr^rrrWrrB)rr\Zxyr r r!test_invalid_shapeXs   rc Cstddd}tdtkr(tdkr2nnd}nd}tdD]Z}ttj}t|W5QRXt ||kstt |ddD]}d t |j kst qq>dS) Ndata/defunct_keywords.hdrr7r8rMrRrrTZPCi_ja) rrrrangerrr rrAr1r2rYr[)rEZn_warn_r\itemr r r!#test_warning_about_defunct_keywordsgs   rc Cs2tddd}ttjt|W5QRXdS)Nrr7r8)rrrr rrArEr r r!-test_warning_about_defunct_keywords_exceptionys rcCsd}ttdkr|d7}nttdkr.|d7}|d7}d|}t}tj| }d|krj|d=d|krx|d=tj| }t |t |kst dS) N)zPWCSAXES = 2 / Number of coordinate axes zPCRPIX1 = 0.0 / Pixel coordinate of reference point zPCRPIX2 = 0.0 / Pixel coordinate of reference point zPCDELT1 = 1.0 / Coordinate increment at reference point zPCDELT2 = 1.0 / Coordinate increment at reference point zPCRVAL1 = 0.0 / Coordinate value at reference point zPCRVAL2 = 0.0 / Coordinate value at reference point zPLATPOLE = 90.0 / [deg] Native latitude of celestial pole r)zPMJDREF = 0.0 / [d] MJD of fiducial time r)zPDATEREF = '1858-11-17' / ISO-8601 fiducial time zPMJDREFI = 0.0 / [d] MJD of fiducial time, integer part zPMJDREFF = 0.0 / [d] MJD of fiducial time, fractional part )ENDCOMMENT) rrr@r rAr Header fromstringto_header_stringstripdictr2)Zhdrstr header_stringr\Zh0h1r r r!test_to_header_strings    rcCsttdkrdnd}ttdkr&d}nttdkr8d}nd}t}|}|}t|tjsdt t|dtj sxt ||dj | dkst dS)Nr rrNr) rrr rA to_headerto_fits isinstancer HDUListr2 PrimaryHDUrE)Znrecr\rwfitsr r r! test_to_fitss  rc Cs`td}ttjt|}W5QRXtjtdd}|W5QRXt|dks\t dS)NrrZA_ORDERrr;) rrrr rrArrr1r2)rsrr\r r r!test_to_header_warnings rcCst}|}|jj|ks td|jj|ks8td|ksDtd}|j|d}||ks`td|ksltd|jj|kstdS)NrPkey)r rArZaltr2r)r\rEZwkeyr r r!test_no_comments_in_headers    rc Csbttd}|}W5QRXttj,ttjtj |ddW5QRXW5QRXdS)zD Causes a double free without a recent fix in wcslib_wrap.C zdata/too_many_pv.hdrFZfixN) openrreadrrWr rXrrrO)fdrEr r r!test_find_all_wcs_crashsrignorec Csttd}tddt|D}ttdkr:d}nLttdkrLd}n:ttdkr^d }n(ttd krpd }nttd krd }nd}tt|}| }W5QRXtdd|D|kst dS)Ndata/validate.fitscSsh|] }|qSr r.0rr r r! sz test_validate..rRzdata/validate.7.6.txtrMzdata/validate.7.4.txtz6.0zdata/validate.6.txtz5.13zdata/validate.5.13.txtz5.0zdata/validate.5.0.txtzdata/validate.txtcSsh|] }|qSr rrr r r!rs) r validatersortedrepr splitlinesrrr readlinesr2)results results_txtrDrlinesr r r! test_validates      rcCsBttd}tddt|D}|dddddgks>tdS) Nzdata/tab-time-last-axis.fitscSsh|] }|qSr rrr r r!rsz(test_validate_wcs_tab..rzHDU 0 (PRIMARY):zHDU 1 (WCS-TABLE):z No issues.z WCS key ' ':)r rrrrrr2)rrr r r!test_validate_wcs_tabsrc Cs8ttttd}W5QRXdt|ks4tdS)Nzdata/2wcses.hdrz WCS key 'A':)rrrr rrrYr2)rr r r!test_validate_with_2_wcsess rc Csptjdd}ddg|j_ddg|j_ddg|j_tddgd d gg|j_d d g|_d d g|j_ tdddddgdddddgdddddgdddddgdddddgg}tdddddgddd d!dgd"d#d$ddgd%d&dddgd'ddddgg}t ||dd|jj|_ |j t tj|}t|j|jjd*|jjdd(|d)t|j|jjd+|jjdd(|d)dS),NrTrgzGio@g ףp=L@r;igr%>g]/M?gZQf?g=zhqܾiz RA---TAN-SIPz DEC--TAN-SIPrgc)Şl>gA=gNqLg>gǢ%=g{p8b=rzg 2g=gaR"=gW.2LgdV>OY!c=gT SgbYza>g@U]g= ri=g ! g&Y=g$-g o+>guM/g< CHg>䔍Y=g3--”/=g3;6dgư>rjrg)r;)r;)r rAcrvalcdeltcrpixr^rlrn_naxisctypeSipsipsetsqrtutilsproj_plane_pixel_arearrB all_pix2world)ZtwcsabZpscaler r r!test_crpix_maps_to_crval sL                   r-C6?aFTc Cs\|dkrtd}d}tj|s2td|ddt|}t||j |} | ~| jj } | j d} t d| t} t d | t} td d tjtt| | Dd}td tj|| }W5QRXd | dd}| d||d|}tj||dd}| ||}z,t}| j||||||d}t}Wntjjk r}zt}d}|jdk r|jj d}td|dtd|jtd|j |jdtdt!tj"j#|j |jddtdt!tj"j#|j$|jdddntdd}|j%dk rn|j%j d}td|dtd|j%td|j |j%dntdtd|j j d||d td!|j td"|j$dtd#|j&dd$||d|W5d}~XYnXt'tj(t)||d dd}t!|}t*|}td%|d&d'|d&d(||d|d)|ksXt+dS)*z6Test all_world2pix, iterative inverse of all_pix2worldNzdata/j94f05bgq_flt.fits)ZSCIr;z Input file 'sz$' to 'test_all_world2pix' not found.rg?g?cSsg|] }|qSr )flattenrir r r! \sz&test_all_world2pix..rrT?r?r;)Zaxis) toleranceadaptivemaxiterdetect_divergencez There are z diverging solutions.z Indices of diverging solutions: zDiverging solutions:  z(Mean radius of the diverging solutions: z*Mean accuracy of the diverging solutions: z!There are no diverging solutions.z slowly converging solutions.z(Indices of slowly converging solutions: zSlowly converging solutions: z*There are no slowly converging solutions. z converged solutions.zBest solutions (all points): z Accuracy: z; Finished running 'test_all_world2pix' with errors. ERROR: z Run time: z5 Finished running 'test_all_world2pix'. Mean error = ez (Max error = z ) Run time: ru),rr=r>isfileOSErrorr rr rArEcloserrr(Zastypeintr^ZdstackZmeshgridmaprrrrappendrrnow all_world2pixZ NoConvergenceZ divergentprintZ best_solutionZmeanZlinalgZnormZaccuracyZ slow_convargsrsumpowerZamaxr2)fnameextrrZ random_nptsrrrhr\rZncoordZnaxesi_lZnaxesi_uZimg_pixZrnd_pixZmwidthZtest_pixZ all_worldZ runtime_beginZall_pixZ runtime_endrZndivZnslowerrorsZmeanerrZmaxerrr r r!test_all_world2pix:s        "$     rc Cs@tddd}ttjt|}W5QRX|ddddS)z` Test parsing of WCS parameters with redundant SIP and SCAMP distortion parameters. rr7r8rN)rrrr rrArrbr r r!$test_scamp_sip_distortion_parameterss r c Cs6tddd}ttjtj|ddW5QRXdS)z! From github issue #1854 rQr7r8FrN)rrrWr rXrArr r r! test_fixes2s r cCs.tddd}t|}|jjddks*tdS)z! From github issue #1918 z data/unit.hdrr7r8rTrVN)rr rAcunitr2rbr r r!test_unit_normalizations  r c Cs@dddddddddd d d d }t|}|d }||t|}|}W5QRXt|dksft|ddksvtd|dkst|j|dddt|}|}W5QRXt|dkst|ddkstd|dksttt |j|ddW5QRX|d=|d=t|}t t ||W5QRXdS)z! From github issue #1912 zRA---ZPNryggy\tgCs@zDEC--ZPNgW[ʧ@g(D!TE@rgk@i) r}ZCRUNIT1rrr~ZCRUNIT2rrPV2_1ZPV2_3NAXIS1NAXIS2ztest.txtrrTzICRS z color=greenrSZFK5red)coordsyscolorzFK5 z color=redZFOO)rrrN) r rAZfootprint_to_filerrr1r2rrWrrr)tmp_pathrr\Ztestfilefrr r r!test_footprint_to_filesD       rcCsDt}d|d<d|d<tjdgg|d}t|g}t|dS)z! From github issue #2053 ZICRSZRADESYSArrrrN)r rrrr r)rZhduhdulistr r r!test_validate_faulty_wcss  rc Cs`tddd}ttj>ttj&tj|dd}|ddggdW5QRXW5QRXdS) Nzdata/invalid_header.hdrr7r8F)Z_do_setg@g@r) rrrWr rXrrrArrbr r r!test_error_messages  rcCsvtddd}t|}|ddd\}}t|s6tt|sDt|ddd\}}t|rdtt|rrtdS)Nzdata/zpn-hole.hdrr7r8nr)rr rArBr^r`r2)rEr\radecr r r!test_out_of_boundss  rc Csjtd}ttjLt|}d}tddgddgddgd d gg}|j|d }t ||W5QRXdS) NrrigLi@g,kG@gOhWi@gү yG@g>H^i@g9֤_KG@gWi@g)$ӀG@)axes rrrr rrAr^rlcalc_footprintrr r\r ref footprintr r r!test_calc_footprint_1,s  r&c Csltd}ttjNt|}d}tddgddgddgd d gg}|j|d d }t ||W5QRXd S)z'Test calc_footprint without distortion.rrrrhrigԧWi@gLm@uG@g ci@g9}BaG@gKYXi@gȅڡ'G@F)r  undistortNr!r#r r r!test_calc_footprint_2>s r(cCspt}ddg|j_ddg|j_ddg|j_d}tddgdd gd d gd dgg}|j|d d d }t||d S)z-Test calc_footprint with corner of the pixel.rwrxg?g@r{r|)rTrvgrgfffff~v@F)r r'centerN) r rArrrr^rlr"r)r\r r$r%r r r!test_calc_footprint_3Qs   "r*cCsntddd}t|}|ddd\}}td|dtd|d|||d\}}td|dtd|ddS)Ndata/irac_sip.hdrr7r8rHMbP?)rr rAZ sip_pix2focrZ sip_foc2pix)rEr\Zx0Zy0x1y1r r r!test_sip^s     r1cCsVtddd}tj|}d|d<|jdd|dd tj|d d }|jd ksRt dS) Nr+r7r8rSZNAXISZNAXIS3@r)afterrTr) rr rrrindexr rArr2rbr r r!test_sub_3d_with_sipns   r5c Cstddd}ttj.t|}||}d|jksBt W5QRXtddd}t|}||}d|jkst dS)z% Just make sure that it runs zdata/spectra/orion-freq-1.hdrr7r8z WCS Keywordszdata/3d_cd.hdrN) rrrr rrAZprintwcsZ readouterroutr2)Zcapsysrr\Zcapturedr r r! test_printwcsxs    r7cCsRd}t|}tj|}t|}|ddd\}}t |rJt |sNt dS)Na SIMPLE = T / conforms to FITS standard BITPIX = 8 / array data type WCSAXES = 2 / no comment CTYPE1 = 'RA---TAN' / TAN (gnomic) projection CTYPE2 = 'DEC--TAN' / TAN (gnomic) projection EQUINOX = 2000.0 / Equatorial coordinates definition (yr) LONPOLE = 180.0 / no comment LATPOLE = 0.0 / no comment CRVAL1 = 16.0531567459 / RA of reference point CRVAL2 = 23.1148929108 / DEC of reference point CRPIX1 = 2129 / X reference pixel CRPIX2 = 1417 / Y reference pixel CUNIT1 = 'deg ' / X pixel scale units CUNIT2 = 'deg ' / Y pixel scale units CD1_1 = -0.00912247310646 / Transformation matrix CD1_2 = -0.00250608809647 / no comment CD2_1 = 0.00250608809647 / no comment CD2_2 = -0.00912247310646 / no comment IMAGEW = 4256 / Image width, in pixels. IMAGEH = 2832 / Image height, in pixels. ir) ioStringIOr r fromtextfiler rArCr^r`r2)rErr\rrr r r!test_invalid_sphericals    r<c Csd}tjdd}tjt|ddt|W5QRXGdddtj}|dd}tjt|ddt|W5QRXdS) zRegression test for #3066z'{}' object is not iterablerTrrArc@s eZdZdS)z!test_no_iteration..NewWCSN)rHrIrJr r r r!NewWCSsr=N)r rArrWrformatiter)rr\r=r r r!test_no_iterations  r@5z'TPV only works with wcslib 5.x or later)reasonc Csttjdddd}ttjdddd}ttjt|}t|}t | |jj gd| |jj gdt| }t| }t | |jj gd| |jj gdt | |jj gd| |jj gdt | |jj gd| |jj gdW5QRXdS)Nr5z siponly.hdrr7r8 tpvonly.hdrr;) rr=r>r@rrr rrArrrr)Z sip_header tpv_headerZw_sipw_tpvZw_sip2Zw_tpv2r r r!test_sip_tpv_agreements:    rFc Csttjdddd}ttjvt|}| dddgdddgd\}}|d|d krl|d |d kspt |d|d kr|d |d kst W5QRXdS) Nr5rCr7r8rrr,r;rT) rr=r>r@rrr rrArBr2)rDrErrr r r! test_tpv_copys  $rHc Cs^td}t|@}t|dj|}t|ddgtj t| dtj tj | dddgdd dgd| dddgdd dgd|j |_ |j|_|j|_|j|_|j|_|j j|j _|j j|j _|j j|j _|j j|j _|jjd kst|jjd kst|jjdkst|jjdks,tt|jjd d gt|dj|W5QRXdS) Ndata/dist_lookup.fits.gzr;g|g >go:O>g>=rrr,rGrg@g@)rr rr rArErproj_plane_pixel_scalesuryrZ p4_pix2focZdet2imZcpdis1Zcpdis2Zdet2im1Zdet2im2rrrrr5Za_orderr2Zb_orderZap_orderZbp_orderr)r>rr\r r r! test_hst_wcss8     rLc Cstd}t|}t|dj|}|dj}|t|dj t|dj }dd|D}dd d d d d ddddg }t |t |kst |D]\}}} |||| fkst qdS)NrIr;rzCPDIS*zDP*cSsi|]\}}}|||fqSr r rkvcr r r! %sz'test_cpdis_comments..)ZCPDIS1LOOKUPPrior distortion function type)z DP1.EXTVERr$Version number of WCSDVARR extension)z DP1.NAXESru1Number of independent variables in CPDIS function)z DP1.AXIS.1r3Axis number of the 1st variable in a CPDIS function)z DP1.AXIS.2ru3Axis number of the 2nd variable in a CPDIS function)ZCPDIS2rRrS)z DP2.EXTVERrurT)z DP2.NAXESrurU)z DP2.AXIS.1rrV)z DP2.AXIS.2rurW) rr rr rArErrr(cardsr1r2) r>rr\rwcscardswcsdictrefcardsrNrOrPr r r!test_cpdis_commentss*  r\c Cstd}t|}ttjt|dj|}W5QRX| t | djdj }dd|D}dddd d d d d ddg }t |t |kst|D]\}}}||||fkstqdS)Nzdata/ie6d07ujq_wcs.fitsrzD2IM*cSsi|]\}}}|||fqSr r rMr r r!rQBsz&test_d2im_comments..)ZD2IMDIS1rR!Detector to image correction type)z D2IM1.EXTVERrrT)z D2IM1.NAXESru0Number of independent variables in D2IM function)z D2IM1.AXIS.1r2Axis number of the 1st variable in a D2IM function)z D2IM1.AXIS.2ru2Axis number of the 2nd variable in a D2IM function)ZD2IMDIS2rRr])z D2IM2.EXTVERrurT)z D2IM2.NAXESrur^)z D2IM2.AXIS.1rr_)z D2IM2.AXIS.2rur`)rr rrrr rrArErr(rrXr1r2) r>rr\rYrZr[rNrOrPr r r!test_d2im_comments:s* racCstd}t|dS)Ndata/sip-broken.hdr)rr rA)rr r r!test_sip_brokenZsrccCstjdd}dddg|j_dddg|j_ddd g|j_|j|}tdD]H}|d |d |jj|ksvt|d |d |jj|ksPtqPd S)L Regression test for https://github.com/astropy/astropy/issues/4612 rSr2XdHBr.חARA---TANDEC--TANFREQCRVALr;CDELTN) r rArrrrrrr2r\rEiir r r!test_no_truncate_crvalbs   "rocCstjdd}dddg|j_dddg|j_ddd g|j_d d d g|j_d d d g|j_d |j_|j| }t dD]H}|d|d |jj|kst |d|d |jj|kstt qtdS)rdrSrrerfgh㈵>gj@zRA---SINzDEC--SINrjryrUr;gR=KBrkrlN) r rArrrr rZrestfrqrrrr2rmr r r!test_no_truncate_crval_try2rs   "rpcCstjdd}ddg|j_ddg|j_ddg|j_|j|}|d|jjdksZt|d |jjd ksrt|jtjd }|d|jjdkst|d |jjd kstd S) zL Regression test for https://github.com/astropy/astropy/issues/5162 rTrg{nI@r.rhrirrrr;relaxN) r rArrrrrr2Z WCSHDO_P17)r\rEr r r!test_no_truncate_crval_p17s     rscCs`tjdd}dddg|j_dddg|j_dd d g|j_|jt|}|j|jd S) z Regression test for https://github.com/astropy/astropy/issues/4612 This one uses WCS.wcs.compare and some slightly different values rSrgiJn@rerfr.rgrhrirjN)r rArrrrrcompare)r\w2r r r!test_no_truncate_using_compares  rvc Cstd}t|~}ttjft|d}t|dj}|j |jsRt t|d}t|dj}|j |jst W5QRXW5QRXdS)zj Passing ImageHDU or PrimaryHDU and comparing it with wcs initialized from header. For #4493. rrr;N) rr rrrr rrArErtr2)r>rZwcs_hduZ wcs_headerr r r!test_passing_ImageHDUs rwc CsVtd}t}|t|}W5QRXtt|jdd}W5QRX|t|}W5QRXtdd|jj Ds~t |jdd}d|kst |t|}W5QRXtdd|jj Dst tt|jd d }W5QRXd|kst |tj|d d }W5QRXtd d|jj Ds2t tt|jd d }W5QRX|t|}W5QRXtd d|jj Dst |jdd}|t|}W5QRXtdd|jj Dst d|kst |j dk st |t|}W5QRXddg|j_ |jdd}|t|}W5QRXtdd|jj DsRt dS)z Test for #4814 rbNrqcss|]}|d VqdSz-SIPNendswithrZctypr r r! sz(test_inconsistent_sip..FZA_0_2css|]}|d VqdSrxryr{r r r!r|sCrcss|]}|d VqdSrxryr{r r r!r|s css|]}|d VqdSrxryr{r r r!r|sTcss|]}|dVqdSrxryr{r r r!r|srhricss|]}|dVqdSrxryr{r r r!r|s) rr"r rArrrrr_rr2r)rrr\ZnewhdrZwnewr r r!test_inconsistent_sipsL         rcCsttjdd}ddg|j_ddg|j_ddg|j_ddg|j_|jd d |d d d \}}t|d t|d dS)zTest for #4957rTrzRA---CARzDEC--CARrviZr;Fi,riLiN) r rArrrrZ bounds_checkrBr)r\rrr r r!test_bounds_checks      rcCstjdd}ddg|j_ddg|j_ddg|j_ddg|_|jdksHt|jdksVtd |_|jd d gksntd |_|jd dgkst|jdkstd|_|j dkstdS)NrTrr;r|r)rr)rr)c;rr)-rr)rr) r rArrrrZ pixel_shaper2Z array_shapeZ pixel_boundsr\r r r! test_naxiss     rc Csttd.}ttjt|dj}W5QRXW5QRX|j ddd}|j dd}d|d <d |d <| |t tj|dd }W5QRX|jj t dd gkstd S)z Test that when creating a WCS object using a key, CTYPE with that key is looked at and not the primary CTYPE. fix for #5443. rrrTA)rrrFrqz RA---SIN-SIPZCTYPE1Az DEC--SIN-SIPZCTYPE2ArN)r rrrrr rrArErupdater"rr^rlr_r2)rr\rZh2r r r!test_sip_with_altkeys$  rc Csltd}ttt|}W5QRX|}t|tj s@t t|dtj sTt t|dtj sht dS)z5 Test to_fits() with LookupTable distortion. zdata/dist.fitsrr;N) rrrrr rArrr rr2rZImageHDU)rsr\rr r r!test_to_fits_1"s rcCsltd}tj|}|d=|d=tj|dd}t|jtjs@t |jj ddksTt |jj dd ksht d S) z* Test sip reading with extra key. rbrrr)rErrr r;iN) rr rfromfiler rArrrr2r)Zhdr_namerEr\r r r! test_keyedsip/s rc Csttd.}ttjt|dj}W5QRXW5QRXt d}t || |dt || |dgdgf}| gdgd}t |d|dt |d|d| gdgd}t |d|dt |d|ddS)NrrrrrTr;)r rrrrr rrArEr^zerosrrr)rr\Zinprpr r r!test_zero_size_input>s$  rcCsttjdd}|dd}t|tdg|djdks.zCTYPE*csg|]}j|qSr rrr*r r!rszCRVAL*csg|]}j|qSr rrr*r r!rszCRPIX*csg|]}j|qSr rrr*r r!rszCDELT*csg|]}j|qSr rrr*r r!rszCUNIT*z7.9iriii r)Zdtyper;rPCrrZtimesysZtrefposZtrefdirZplephemZtimeunitZdaterefZdateobsZdatebegZdateavgZdateendrZmjdrefZmjdobsZmjdbegZmjdendZjepochZbepochZtstartZtstopZxposureZtimsyerZtimrderZtimedelZtimepixrZtimeoffsZtelapseZczphsZcperiZMJDREFZMJDREFIAZMJDREFFAN)rEr(r\r rr2rrZ axis_typesrrrrr rr^rZfloat64rgetrngetattruppernan)r+rrrrr Ztime_axis_coderrnrjZ char_keysrZnum_keysZhdrvr r*r!test_keywods2wcsprmsv .4 &z#TestWcsWithTime.test_keywods2wcsprmcCs&t|jj|jjjd|jjjdS)Nr;)r;)rr\rr rrr*r r r!test_transformsszTestWcsWithTime.test_transformsN)rHrIrJr,rrr r r r!rs HrcCstjdd}d|j_d|j_d|j_d|j_dddg}dd dg}dd dg}||||d \}}}t|d d d gt|tj d dgt|tj ddgdS)NrSr)ZVELO_LSRrwrx)irr)r;i)g?gg?irrvrirLgXf@g0f@g= 7.1 < 7.4r.rTz cutout.fitsN)rr^rlr rrrr rrArEr r5rrrrZxfailZpixel_to_worldZ separationrKZmasr2Zwritetor) rr>ZcenZsizrr\cutZw0rZcutfilerur r r!test_distortion_header.sH       *& rc Cstd}tj|}ttjtj|ddgddgd}W5QRX|j dksRt t |jj dd gksjt t |jjd d gst t |jjd d gd d ggst t |jjddgst t |jjdst dS)zS Test selection of a specific pixel list WCS using ``colsel``. See #11412. zdata/chandra-pixlist-wcs.hdrimageZpixelr )ZkeyselZcolselrTrhrig6-l@g}gMr;rg>B!g>B!?gf@N)rr rr;rrr rrArr2r(rr^rmrrnrZlonpole)Zhdr_filerr\r r r!test_pixlist_wcs_colselds " rz7.8z8TIME axis extraction only works with wcslib 7.8 or latercCstjdd}dddg|j_|jt|tjgjjdgksDt|dddggdd |tjgdggdd ks~tdS NrSrrhriZTIMEr;rTrr)rr) r rArrr(sub WCSSUB_TIMEr2rBrr r r!test_time_axis_selectionts   rcCsxtjdd}dddg|j_|j|js.t|tjgjsBt| dddggdd |j dggdd ksttdSr) r rArrZ has_temporalr2rrZ is_temporalrBZtemporalrr r r! test_temporals   rcCstjdd}dddg|j_dddg|j_d d dg|j_d d d gddd gd d dgg|j_dddg|j_|jdddg}tdd|D}| |}t dddg}||}| |gdd}| |gdd}t j |||dddstt j ||gdd|dddstdS)NrSrrhrirjg@@@g0@rgQ@g33333Rg>g'eRC>rzgTqs*gh㈵>rTr;css|]}|dVqdS)r;Nr rr r r!r|sz3test_swapaxes_same_val_roundtrip..rrrfr)r rArrrrnrrr(rr^rlrBrmr2rC)r\Z axes_orderZ axes_order0wsZimcoordZimcoordsZval_refZ val_swappedr r r! test_swapaxes_same_val_roundtrips      r)NrrrrFrT)rr9r= contextlibrrnumpyr^rZ numpy.testingrrrrZpackaging.versionrZastropyr rKr Zastropy.coordinatesr Z astropy.ior Zastropy.nddatar Zastropy.tests.helperrZastropy.utils.datarrrZastropy.utils.exceptionsrrrZastropy.utils.miscrZ astropy.wcsr __version__rr"r#rKr]markfilterwarningsrcrqrtrrrrrrrrrrrrrrrrrrrr r r rrrrr&r(r*r1r5r7r<r@ZskipifrFrHrLr\rarcrorprsrvrwrrrrrrrrrrrrrrrrrrr r r r!s            #+  +/   ,      0 u   5        "  * 7    ,$V 6