a ߙfbFS@s,ddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZmZddlmZddlmZdd lmZdd l mZgd Zgd Zed deddeddeddeddeddgZ dZ!gdZ"ej#ej#fej$ej$fdej#fej#dfdej$ fej$ dfgZ%ddZ&d d!Z'ej()d"e"d#d$Z*d%d&Z+d'd(Z,d)d*Z-ej()d+e%d,d-Z.d.d/Z/d0d1Z0d2d3Z1d4d5Z2d6d7Z3d8d9Z4d:d;Z5dgd?d@dAZ7dBdCZ8dDdEZ9dFdGZ:dHdIZ;dJdKZGdOdPdPZ?dS)Q)VersionN)assert_allcloseassert_array_equal)assert_quantity_allclose) extract_array add_arraysubpixel_indicesoverlap_slicesNoOverlapErrorPartialOverlapErrorCutout2D)WCSSip)proj_plane_pixel_area)SkyCoord)units)CCDData)) ףp= %@(\@){Gz@q= ף)@)GzT?@Q?@)q= ףp?Gz?)33333s4@= ףp=(@)QE@Qk8@))rr))r"r!)r"rrr"rrrrrrrrrrrr)))r)r!)r)r)r#r!rcCsLtt}tdddWdn1s,0Ydt|jvsHJdS)z>Overlap from arrays with different number of dim is undefined.)r"r%r)r#r!rrNthe same number of dimensionspytestraises ValueErrorr strvalueer5>lib/python3.9/site-packages/astropy/nddata/tests/test_utils.pytest_slices_different_dim%s *r7cCsLtt}tdddWdn1s,0Ydt|jvsHJdS)z&Position must have same dim as arrays.)r"r%r*)rrrNr,r-r3r5r5r6test_slices_pos_different_dim,s *r8poscCs:tttdd|Wdn1s,0YdS)zAIf there is no overlap between arrays, an error should be raised.r%r%r!r!N)r.r/r r )r9r5r5r6test_slices_no_overlap3s r<c Cstddd}|tdddftdddffks0Jtdddd d }|tdddftdddffksdJd D]R}tt"}tdd|fd d }Wdn1s0Yd t|jvshJqhdS)z1Compute a slice for partially overlapping arrays.r%rrrr!Nr#rpartialmode)rr"strictzArrays overlap only partially.)r slicer.r/r r1r2)Ztempr9r4r5r5r6test_slices_partial_overlap:s $$ 0rEcCstddddd\}}|dj|djkr4dks:nJ|dj|djkrZdks`nJ|dj|djkrdksnJ|dj|djkrdksnJtddd dd\}}|dj|djkrd ksnJ|dj|djkr d ksnJ|dj|djkr4dks:nJ|dj|djkr^dksdnJtdd d \}}|dj|djkrdksnJ|dj|djkrdksnJ|dj|djkrdksnJ|dj|djkrdksnJtdd d \}}|dj|djkrPdksVnJ|dj|djkrzdksnJ|dj|djkrdksnJ|dj|djkrdksnJdS)z: Test overlap_slices when extracting along edges.  rGrrr#r#rCrArr#r)rJrGr+r:r%N)r startstop)Zslc_lgZslc_smr5r5r6test_slices_edgesHs(&&&&&**********rNcCsPtt }tdddddWdn1s00Ydt|jvsLJdS)z+Call overlap_slices with non-existing mode.r=r>r?fullrANzMode can be onlyr-r3r5r5r6test_slices_overlap_wrong_modegs .rPpositioncCs:tttdd|Wdn1s,0YdS)zT A ValueError should be raised if position contains a non-finite value. )rKrKrHN)r.r/r0r )rQr5r5r6test_slices_nonfinite_positionns rRc Cstd}d}gd}|D]$\}}t|||fdd}t||qd}d}d}|gd |g} t|| D]&\}}t|||fdd d }t||qfd S) b Test overlap_slices (via extract_array) for rounding with an even-shaped extraction. rGr!) )ףp= ?r*)?r*)jt?r*)ףp= ?r*)@r*)Gz@r!r)Q@r[)@r[)5^I @r[)Q@r[)@r[)Gz@r$r@rAgGzgRQgg\(\߿g{Gzr)r)rr#r)rcrB fill_valueNnparangerrzip datashapeZpositions_expectedr9ZexpoutZ positionsZexp1Zexp2expectedr5r5r6&test_extract_array_even_shape_roundingys    roc Cstd}d}gd}|D]$\}}t|||fdd}t||qd}d}d}|gd |gd } t|| D]&\}}t|||fdd d }t||qjd S)rSrGr>) )rUrr#r!)rVrp)rWr#r!r)rXrq)rYrq)rZrq)r\rq)r]rq)r^r!rr")r_rr)r`rr)rarrr@rArbrcrcrrcrr#rr"rcrdNrfrjr5r5r6%test_extract_array_odd_shape_roundings    rucCsVtt&}ttdddddWdn1s60Ydt|jksRJdS)z*Call extract_array with non-existing mode.r"rTr?rOrANz0Valid modes are 'partial', 'trim', and 'strict'.)r.r/r0rrgrhr1r2r3r5r5r6test_extract_array_wrong_modes 4rvcCstttdddddtddgks.JdD]4}tttdd|ft|d|gks2Jq2tttd dd tjdtd tjgksJd S) [Extract 1 d arrays. All dimensions are treated the same, so we can test in 1 dim. r"rTr?rcrerrqr#@r"rN)rgallrrharrayinfir5r5r6test_extract_array_1d_evens  rc Cstttdddddtgdks.Jtttdddddtgdks\Jd D]:}tttdd|ft|d ||d gks`Jq`tttdddddtgd ksJtd }t|dd }|ddksJt|d sJ|j|jksJdS)aExtract 1 d arrays. All dimensions are treated the same, so we can test in 1 dim. The first few lines test the most error-prone part: Extraction of an array on the boundaries. Additional tests (e.g. dtype of return array) are done for the last case only. r"r>r&rcrxrsr?rtr*r#)r!rrcryrzrrN)rgr{rrhr|Zisnandtype)rZarrayin extractedr5r5r6test_extract_array_1d_odds0       rcCs`tttdddddtgdks.Jtttdddddtgdks\JdS) z(In 1d, shape can be int instead of tupler"rrrcrxrsr&Nrgr{rrhr|r5r5r5r6test_extract_array_1ds  rcCsBtdddD].}tttdd|tgdksJqdS)zinteger is at bin centergGz@gQ @皙?r%rrrN)rgrhr{rr|)ar5r5r6test_extract_Array_floats rc Cstttdddddtdgks,JdD]8}tttdd|fddt|d|gks0Jq0tttd dd ddtd gksJd S) rwr"rTr?trimrArrqr#ryrzrNrr~r5r5r6test_extract_array_1d_trims    rrB)r@rrCcCsNtd}td}||ddddf<t|dd|d}t||ksJJdS)zq Test extract_array utility function. Test by extracting an array of ones out of an array of zeros.  rr:rrJrAN)rgzerosonesrr{)rBlarge_test_arraysmall_test_arrayZextracted_arrayr5r5r6test_extract_array_easys  rcCstjdtd}tddD]&}t|d|ddd\}}|d ksJqtgd gd D].\}}t|d |fd dd\}}||fksTJqTttdddD].\}}t|d|fddd\}}||fksJqdS)zCheck that the return position is calculated correctly. The result will differ by mode. All test here are done in 1d because it's easier to construct correct test cases. r%rr&r)rr@T)rBZreturn_position)r#)rUg)\(?r)g\(\?gRQ?r#rTrC)r&rr#r#r#r#r#r>rN)rgrhfloatrri)rrrZnew_posrnr5r5r6test_extract_array_return_poss"     rcCsdttjtdkr`d}tjt|d,ttjdtdddtj dWdn1sV0YdS) Nz1.20z9fill_value cannot be set to np.nan if the input array has)matchrFrr:rIrx) rrg __version__r.r/r0rrintnan)msgr5r5r6 test_extract_array_nan_fillvalue's rcCsZtd}td}|}|ddddf|7<t||d}t||ksVJdS)zi Test add_array utility function. Test by adding an array of ones out of an array of zeros. rr:rrJNrgrrcopyrr{rrZlarge_test_array_refZ added_arrayr5r5r6test_add_array_odd_shape/s    rcCsjtd}td}|}|ddddf|ddddf7<t||d}t||ksfJdS)l Test add_array_2D utility function. Test by adding an array of ones out of an array of zeros. r)r"r"rr!r"r+Nrrr5r5r6test_add_array_even_shape>s   , rcCsFtd}td}|}||7}t||d}t||ksBJdS)rrr:Nrrr5r5r6test_add_array_equal_shapeMs    rrQZsubpixel_indexcCstt|t|ksJdS)z Test subpixel_indices utility function. Test by asserting that the function returns correct results for given test values. N)rgr{r subsamplingrr5r5r6test_subpixel_indices\s rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%S)& TestCutout2DcCsHtddd|_tddd|_tdd}tjd }d }|t|| t |g|t ||t|gg|j _ d d g|j _ |jj tj|jjtjg|j _d d g|j _||_ |}ddg|j _ tgdgdgdgdgdg}tgdgdgdgdgdg}t||dd|j j|_|j ||_dS)Ng4@r%r"z13h11m29.96s -01d19m18.7sZicrs)framer!)Znaxisr`g2.Z >zRA---TANzDEC--TANrz RA---TAN-SIPz DEC--TAN-SIP)rrgc)Şl>gA=gNqL)rg>gǢ%=g{p8b=)g 2g=gaR"=rr)gW.2LgdV>OY!c=rrr)gT Srrrr)rrgbYza>g@U]g= ri=)rg ! g&Y=g$-r)g o+>guM/g< CHrr)g>䔍Y=g3--”/=rrr)g3;6drrrr)rgrhZreshaperkrrQr ZpiZcosZsinwcsZcdZctyperaZto_valueuZdegdeccrvalcrpixdeepcopyr|rZsipsetsipwcs)selfrZrhoZscalerrbr5r5r6 setup_classisF      zTestCutout2D.setup_classcCs<ddtjddtjdtjfddtjfdg}|D]}d}t|j||}|jjdks\J|jddksnJ|jdks|J|jdksJ|j|ksJt |j d |j d ksJ|j d ksJ|j d ksJ|jd ksJ|jd ksJ|jd ksJ|jtddtddfksJ|jtddtddfks4Jq4dS)NrrHr`)g333333@gffffff @)g@gffffff?rIrGr+)g?g?)rYrY)?r)r#rr)r r r#r"r)rpixelZpixr rkrlZorigin_originalZ origin_cutoutZinput_position_originalrZinput_position_cutoutZposition_originalZposition_cutoutcenter_original center_cutoutZ bbox_originalZ bbox_cutoutslices_originalrD slices_cutout)rZsizessizerQcr5r5r6 test_cutouts(*   zTestCutout2D.test_cutoutcCs<ttt|jddWdn1s.0YdS)Nr;)r#r#r#)r.r/r0r rkrr5r5r6test_size_lengths zTestCutout2D.test_size_lengthc CsZdtjddtjffD]<}ttt|jd|Wdq1sJ0YqdS)Nrr;)rcmKr.r/r0r rk)rrr5r5r6test_size_unitss zTestCutout2D.test_size_unitscCsdtjdtjtj}t|jd|}|jjdks8J|jddksJJ|jtddtddfkshJ|jtd d td d fksJd S) z4 Check size in derived pixel units. g333333?rr;rHr+r%r#r"rrN) rarcsecrr rkrlrrDr)rrrr5r5r6test_size_pixels zTestCutout2D.test_size_pixelcCs|t|jddtj|jd}|jjdks*J|jddkstt t|j|jdWdn1s00YdS)NrH)r.r/r0r rkrQrr5r5r6test_skycoord_without_wcss z&TestCutout2D.test_skycoord_without_wcscCsrt|j|jd|jd}|j|jd|jd|j}|j|jd|jd|j}t|j|jt|j |j dS)NrHrr#r r rkrQr from_pixelrrrrrrrskycoord_originalskycoord_cutoutr5r5r6 test_skycoords zTestCutout2D.test_skycoordcCstt|j|jd|jdd}|j|jd|jd|j}|j|jd|jd|j}t|j|jt|j |j dS)NrHr@rrBr#rrrr5r5r6test_skycoord_partials z"TestCutout2D.test_skycoord_partialcCs:d}d}t|j|j||f|jd}|jj||fks6JdS)Nr!rr)r rkrQrZ array_shape)rZxsizeZysizerr5r5r6test_naxis_updateszTestCutout2D.test_naxis_updatecCst|jdd|jddj}tt|}t|jg|jj dR|jj dd|dt|j g|jj dR|jj dd|ddS) Nr+rHr@rr#rgư>)ZrtolZatol) r rkrrrgZsqrtrrZ wcs_pix2worldrrZ all_pix2world)rwZpscaler5r5r6test_crpix_maps_to_crval sz%TestCutout2D.test_crpix_maps_to_crvalcCs|t|j|jdd}t||jd}|j|jd|jd|j}|j|jd|jd|j}t|j |j t|j |j dS)NZadu)rkrZunitrHr#r) rrkrr rQrrrrrr)rZccdrrrr5r5r6 test_cutout_with_nddata_as_inputs z-TestCutout2D.test_cutout_with_nddata_as_inputN)__name__ __module__ __qualname__rrrrrrrrrrrrrrrrrrr5r5r5r6rhs$#    r)@Zpackaging.versionrr.ZnumpyrgZ numpy.testingrrZastropy.tests.helperrZastropy.nddatarrrr r r r Z astropy.wcsr rZastropy.wcs.utilsrZastropy.coordinatesrZastropyrrrZtest_positionsZtest_position_indicesrDZ test_slicesrZ test_pos_badrr}Ztest_nonfinite_positionsr7r8ZmarkZ parametrizer<rErNrPrRrorurvrrrrrrrrrrrrirrr5r5r5r6sd  $