a ߙfbE@s<ddlZddlmZddlZddlZddlZddlmZ ddl m Z ddl m ZddlmZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZmZmZdd l m!Z!dd l"m#Z#eej$j%Z&e&edkZ'e(d Z)eej*j+Z,ddZ-ddZ.ddZ/ej0j1dddZ2ej34dddgddZ5ddZ6d d!Z7d"d#Z8ej0j1d$ddZ9d%d&Z:d'd(Z;d)d*Zd/d0Z?d1d2Z@d3d4ZAd5d6ZBd7d8ZCd9DZEd:DZFd;d<ZGejHd=d>ZId?d@ZJej3KdAdBdCZLej34dDeegdEdFZMej34dGdHdIgdJdKZNdLdMZOdNdOZPdS)PN)Version)QuadContourSet)units)WCS)fits)SkyCoord)get_pkg_data_filename)SlicedLowLevelWCSHighLevelWCSWrapper)WCSAxes)EllipticalFrameRectangularFrameRectangularFrame1D)get_coord_meta)CurvedTransformz2.6.1TcCstddS)Nall)pltclose)ZfunctionrLlib/python3.9/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyteardown_function srcCs,tjdddtjdd}t|gddS)NZaxesT)gridrZfigsize皙?r皙?r)rrcfigurer )ignore_matplotlibrcfigrrrtest_grid_regression$s r"cCstjdd}t|gd}|||dddks8J|jdddksPJ|jdddkshJ||dj|ddd ksJ|jddd ksJ|jddd ksJdS) Nrrr rZnothingz10.0 10.0 (world)z10.0) rrr add_axesZ format_coordcoordssavefigjoinstrpath)r tmpdirr!axrrrtest_format_coord_regression,s  r-a NAXIS = 2 NAXIS1 = 200 NAXIS2 = 100 CTYPE1 = 'RA---MOL' CRPIX1 = 500 CRVAL1 = 180.0 CDELT1 = -0.4 CUNIT1 = 'deg ' CTYPE2 = 'DEC--MOL' CRPIX2 = 400 CRVAL2 = 0.0 CDELT2 = 0.4 CUNIT2 = 'deg ' COORDSYS= 'icrs '  )sep grid_typelinescontourscCst}|jdddttd}|td|jj d|dt Zt j dddt j dd dt j dd dt j dd d| |d jWdn1s0YdS) Nr% projection)dZwhitecolorr0ignorez&.*converting a masked element to nan.*messagez6.*No contour levels were found within the data range.*z6.*np\.asscalar\(a\) is deprecated since NumPy v1\.16.*z%.*PY_SSIZE_T_CLEAN will be required.*test.png)rr add_subplotr TARGET_HEADERimshownpzerosr'rwarningscatch_warningsfilterwarningsr(r)r*)r r+r0r!r,rrrtest_no_numpy_warningsMs rEcCstjdddttd}tt}|dWdn1s@0Y|jj ddks^Jtt}t dWdn1s0Y|jj ddksJdS)Nr%r3ZbananarzFrame banana not foundzUnknown frame: banana) rsubplotrr>pytestraises ValueErrorZget_coords_overlayvalueargsr)r r,excrrrtest_invalid_frame_overlaybs ( &rMcCstd}tj|}tjdd}|jgdt|dd}|dd| ddt d t j d t j }t t&|j|d |d d Wdn1s0YdS)Nzdata/2MASS_k_header)rNr)333333?rOrrZequal)r4Zaspectg@gS *|v@gS>f3?oZgalactic transform)rrHeaderZ fromtextfilerrr&rset_xlimset_ylimrudegrGrH TypeErrorZ plot_coord get_transform)r ZtwoMASS_k_headerr!r,crrrtest_plot_coord_transformos      r\cCstjdddttd}|jdddd|jddd d|jd jdksNJ|jd j d dkshJ|jd j dksJ|jdjdksJ|jdj d dksJ|jdj d ksJ| dksJ| dksJdS) Nr%r3z Test x labelZred)Zlabelpadr8z Test y labelrZgreenrbl) rrFrr> set_xlabel set_ylabelr'Z axislabelsZget_textZ get_minpadZ get_color get_xlabel get_ylabel)r r,rrrtest_set_label_propertiessrda{ SIMPLE = T / conforms to FITS standard BITPIX = -32 / array data type NAXIS = 3 / number of array dimensions NAXIS1 = 31 NAXIS2 = 2881 NAXIS3 = 480 EXTEND = T CTYPE1 = 'DISTMOD ' CRVAL1 = 3.5 CDELT1 = 0.5 CRPIX1 = 1.0 CTYPE2 = 'GLON-CAR' CRVAL2 = 180.0 CDELT2 = -0.125 CRPIX2 = 1.0 CTYPE3 = 'GLAT-CAR' CRVAL3 = 0.0 CDELT3 = 0.125 CRPIX3 = 241.0 cCstdd}gd|j_gd|j_gd|j_gd|j_gd|j_tDtj dd d t j d d d |d d t | djWdn1s0Ytt}tLtj dd d t t j d d d |dd t | djWdn1s0YdS)NrZnaxisxyzrXrXzkm/sg4@gĊ@iM@rmrpg?r9z.*PY_SSIZE_T_CLEAN.*r:r%rgrhr%r4Zslicesr<)rgrhr])rwcsctypecunitcrpixcdeltcrvalrBrCrDrrFr(r)r* GAL_HEADERclf)r r+wcs3drrrtest_slicing_warningss        0 r|cCs8tjtdtdtdt|djdS)Nr3zGalactic LongitudezGalactic Latituder<)rrFrZxlabelZylabelr(r)r*)r+rrrtest_plt_xlabel_ylabels  r}cCstGdddt}|}dtjtjfddd}t}t|gd||d}|||jd d | | d j dS) Nc@seZdZddZdS)z:test_grid_type_contours_transform..CustomTransformcSs|dS)Ng?r)selfvaluesrrrrSszDtest_grid_type_contours_transform..CustomTransform.transformN)__name__ __module__ __qualname__rSrrrrCustomTransformsr)scalarr)NNrgrh)typeZunitZwrapnamer)rS coord_metar2r0r<) rrWmsrrr r&rr(r)r*)r+rrSrr!r,rrr!test_grid_type_contours_transforms    rcCsBtjtd}ttd|dks.J|dks>JdSNr3)r]r])rP?rrFrr?r@ZonesZget_xlimZget_ylimr,rrrtest_plt_imshow_originsrcCsBtjtd}|td|dks.J|dks>JdSrrrrrrtest_ax_imshow_originsrcCs|dj}tjtd}|dd|dd|jdjgt j d|jdj ddt ||jdj ddt |dS) Nr<r3rPrr)rr2r) r)r*rrFrrUrVr' set_ticksrWZonerr()r+filenamer,rrrtest_grid_contour_large_spacings    rcCst}t|gd}|||jtddd|dd}t |t sPJ|j tddd|dd}t |t sJdS)NrworldrR) rrr r&contourr@ZarangeZreshaperZ isinstancerZcontourf)r!r,Zcsetrrrtest_contour_returns ""rcCslt}t|gd}||tjtdd*|jt d| ddWdn1s^0YdS)Nrz2No contour levels were found within the data range)match)rrrrR) rrr r&rGZwarns UserWarningrr@rArZ)r!r,rrrtest_contour_empty,s  rcCsjtdd}gd|j_gd|j_gd|j_gd|j_gd|j_tjddd|d d }|j \}}}dS) Nrrerfrjrkrlror%rqrr) rrsrtrurvrwrxrrFr')r r+r{r,rgrhrirrrtest_iterate_coords8s      rcCsBtdd}gd|j_tjddd|ddtt"}tjddd|dWdn1s\0Y|jj dd kszJtt$}tjddd|d dWdn1s0Y|jj dd ksJtd d}d dg|j_t tjddd|d}|j t us Jt tjddd|dd}|j t us8Jt tjddd|dd}|j t usdJt tjddd|d dgd}|j t usJt tjddd|dd}|j t usJtdd}d g|j_t tjddd|d}|j t usJtt$tjddd|ddWdn1s40YdS)Nrrerfr%rqrrr3rz?WCS has more than 2 pixel dimensions, so 'slices' should be set)rgrhr%r]zO'slices' should have as many elements as WCS has pixel dimensions (should be 3)r]rgrhr)rhrg)r%rg)r%rh)rrsrtrrFrGrHrIrJrKrz frame_classr r)r r{rLZwcs2dr,Zwcs1drrrtest_invalid_slices_errorsHsB   0 2     ra a cCs`tt}tjddd|dd}t|jtks.Jttjddd|dd}t|jtks\JdS)Nr%)r%rgrhrrrq) rryrrFreprr'EXPECTED_REPR_1rzEXPECTED_REPR_2)r r{r,rrr test_reprs rcCs$tdd}ddg|j_|j|S)Nr]reZFREQZTIME)rrsrtset)rsrrrtime_spectral_wcs_2ds   rcCstj|ddS)Nr3)rrF)rrrr test_time_wcssrTEX_UNAVAILABLEc Cstjddddddddd d d tdtjdtdtjd d dd }t|}tjtt|dd\}}| d|dd| d|dd|j dj dd|j dj dd|j dj dtjd|j dj dtjd|||ddS)zCRegression test for https://github.com/astropy/astropy/issues/8004.textT)Zusetexr]hf@V@f@rpzRA---MOLzDEC--MOLZICRS) NAXISNAXIS1NAXIS2CRPIX1CRPIX2CRVAL1CRVAL2CDELT1CDELT2CTYPE1CTYPE2ZRADESYSrr4Z subplot_kwrPrg?rr)Zexclude_overlappingr%-)Zspacingplot.pngN)rrr@sqrtpirsubplotsdictr rUrVr'Z set_ticklabelrrWrXrr()r r+headerrsr!r,rrrtest_simplify_labels_usetexs4 rrc Csddg}ddddddd d tdtjdtdtjd d d }t|}tjt||dd\}}||d||d| |dksJ| |dksJt dD]}|j | ||ksJqdS)zTest if ``axis.set_xlabel()`` calls the correct ``coords[i]_set_axislabel()`` in a WCS plot. Regression test for https://github.com/astropy/astropy/issues/10435. ZRAZ Declinationr]rrrrrrprzRA---AITzDEC--AIT) rrrrrrrrrrrrrrr%N)r@rrrrrrr`rarbrcranger'Z get_axislabel)r rlabelsrrsr!r,irrrtest_set_labels_with_coordss.   ratolg?g:0yE>cCstgd}t}t|gd}|||j|jj}||}|dkrZtsZt dt j |j ||dspJdS)N)grq&@g @g@g{@rrzAExact BoundingBox dimensions are only ensured with FreeType 2.6.1)r)rrr r&ZcanvasZdrawrendererZ get_tightbbox FREETYPE_261rGZxfailr@Zallcloseextents)rrr!r,rZax_bboxrrrtest_bbox_sizes     rcCsVtt}t|tjddddf}tjddd|d}||t|}||dS)Nr%r3) rr>r r@Zs_rrFrZr )rsZ sliced_wcsr,Zhigh_wcsrrr"test_wcs_type_transform_regression s  rcCsJt}|jdddtd}|jddd||d||ddS)Nr%r3Zblackr2r7r)rrr=rrr()r+r!r,rrr!test_multiple_draws_grid_contourss r)QrBZpackaging.versionrrGZnumpyr@Z matplotlibZmatplotlib.pyplotZpyplotrZmatplotlib.contourrZastropyrrWZ astropy.wcsrZ astropy.iorZastropy.coordinatesrZastropy.utils.datarZastropy.wcs.wcsapir r Z"astropy.visualization.wcsaxes.corer Z#astropy.visualization.wcsaxes.framer r rZ#astropy.visualization.wcsaxes.utilsrZ(astropy.visualization.wcsaxes.transformsrZft2fontZ__freetype_version__Z ft_versionrZcheckdep_usetexr __version__Z is_devreleaseZMATPLOTLIB_DEVrr"r-rTZ fromstringr>ZmarkZ parametrizerErMr\rdryr|r}rrrrrrrrstriprrrZfixturerrZskipifrrrrrrrrrst                      3