a ߙfbC:@sddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZdd lmZGd d d eZGd ddeZddZddZddZddZddZddZdS)N) assert_equal)fits)SUBTRACTIVE_DITHER_1DITHER_SEED_CHECKSUM)AstropyUserWarning)get_pkg_data_filename) HAS_SCIPY)comparerecords) FitsTestCasec@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zejd-d.ejd/d0d1d2Zd3d4Zd5d6Zejd7d8d9Z d:d;Z!dd?Z#d@dAZ$dBdCZ%dDdEZ&dFdGZ'dHdIZ(dJdKZ)dLdMZ*dNdOZ+dPdQZ,dRS)STestImageFunctionscCst}|jdksJd|jvs$Jd|_|jdks8J|jddksJJtjdd}|jdksdJ|jddksvJt}d|d<tj|dd}|jdksJ|jddksJdS)z/Like the test of the same name in test_table.pyEXTNAMEFOO)nameZEVENTS)headerrN)rImageHDUrrHeader)selfhduhdrr?lib/python3.9/site-packages/astropy/io/fits/tests/test_image.pytest_constructor_name_args z,TestImageFunctions.test_constructor_name_argcCsdd}t}|jdksJd|jvs,Jd|_||dtjdd}||dt}d|d<tj|dd }||dt}d |d<tj|dd }||d tj|d }||d dS) NcSs$|j|ksJ|jd|ks JdS)NEXTVER)verr)rZ reference_verrrr assert_ver_is.szBTestImageFunctions.test_constructor_ver_arg..assert_ver_isr r)r)rrr)rrrrr)rrrrrrrtest_constructor_ver_arg-s$      z+TestImageFunctions.test_constructor_ver_argcCsXtt}|dj}d|d<tj|d}t|}d|djd<|ddksTJdS)z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/153 Ensure that a header from one HDU is copied when used to initialize new HDU. rzlabq01i3q_rawtag.fitsZFILENAMEr!zlabq01i3q_flt.fitsN)rHDUList PrimaryHDUr)rZifdZphdr primary_hduofdrrrtest_constructor_copies_headerKs   z1TestImageFunctions.test_constructor_copies_headercCsrt|d}|tt*}|djddddfWdn1sR0Yt|jdksnJdS)N test0.fitsr rsHDU not found, possibly because the index is out of range, or because the file was closed before all HDUs were read) ropendataclosepytestraises IndexErrorstrvaluerrexc_inforrr test_open^s  8zTestImageFunctions.test_opencCsZt|d}dgddtddD}z |jdd|ks@JW|n |0dS) Nr()rPRIMARYr r$rr r c Ss g|]}|d|dddddfqS)SCIr=)(r:int16r r.0xrrr xsz2TestImageFunctions.test_open_2..r r Foutput)rr*r+rangeinfor,)rr3rCrrr test_open_2tszTestImageFunctions.test_open_2cCsPt|d}|tt}|dWdn1s@0Yt|jdks\Jtj|ddd}|t |dtj sJt |dksJtt}|dWdn1s0Yt|jdksJtj j sJdtj _ zLt|d}|t |dtj s$Jt |dks6JWd tj _ n d tj _ 0dS) Nr(r r)F)lazy_load_hdusr zlist index out of rangeT)rr*r+r,r-r.r/r0r1 isinstancerlenZconfrEr2rrr test_open_3s( & & zTestImageFunctions.test_open_3cCs:tddd}t|}|d}|d}tj|d|tj|d|tj t ||tj t |||d}|d}t |d d "}tj|d|Wdn1s0Yt |d d "}tj|d|Wdn1s0Ytj t ||tj t ||dS) Nrz a_str.fitsz b_str.fitsr+za_fileobj.fitszb_fileobj.fitswbmode nparangereshapeZasfortranarraytemprr$writetotestingassert_array_equalgetdatar*rabZafitsZbfitsZaafitsZbbfitsfdrrrtest_fortran_arrays      02z%TestImageFunctions.test_fortran_arraycCstdddd}t|}|d}|d}tj|ddddddfd|tj|ddddddfd|tj t ||ddddddftj t ||ddddddf|d }|d }t |d d 6}tj|ddddddfd|Wdn1s,0Yt |d d 6}tj|ddddddfd|Wdn1s0Ytj t ||ddddddftj t ||ddddddfdS) Nirr rKza_str_slice.fitszb_str_slice.fitsrrLza_fileobj_slice.fitszb_fileobj_slice.fitsrMrNrPrYrrr!test_fortran_array_non_contiguouss    &&((  FF(z4TestImageFunctions.test_fortran_array_non_contiguouscCstddg}ttj|dg}d|djvs4J|djdksFJ|dj|djdksbJdg}|jdd |ks|J|d |dusJ|d |d usJd |d_|djdd ksJ||d t |d "}|djd ksJWdn1s0YdS)aXRegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/151 Tests that the EXTNAME keyword works with Primary HDUs as well, and interacts properly with the .name attribute. For convenience hdulist['PRIMARY'] will still refer to the first HDU even if it has an EXTNAME not equal to 'PRIMARY'. )rXPRIMARY)rr r!rrr`)rr`r r$r rr r Fr@r6)r`r Z XPRIMARY2 test.fitsN) rrr#r$rrrCrUrTr*)rZprihdrhdulrCrrrtest_primary_with_extnames  z,TestImageFunctions.test_primary_with_extnamec Cst|d}|djddks*J|djddks@J|djddksVJd|djd <d d d |djjd dDdksJ|djddtt |djjddtt |djjdd|djddt |djddddft j gdgdgdgt j ds*Jtj|dddZ}||d||d||djddks|Jd|djd<Wdn1s0Y~t|d|dtj|ddd6}|djddksJ||dWdn1s0Y~t|d|d tj|d d!d}|djd"d#kshJd$|djd"<|djddksJd|djd<||d=|||d||dWdn1s0Y~Wdn1s0Yt|d0}||djdd%ksDJWdn1sZ0Yt jd&t jd}tj|d'd(}t |jt j gd)gd)gd)gt jdsJtj|djt j ddgd*dd+}d d,d |jjdd-Dd.ksJdS)/Nr(primaryZnaxisr)Zscir Zdetectorr gZ0l!KZxxx css|]}t|VqdSNr0r<rrr z:TestImageFunctions.test_io_manipulation..zEXPFLAG = 'NORMAL ' / Exposure interruption indicator FILENAME= 'vtest3.fits' / File name XXX = 1.234E+56 filenamefnamehistorysimplerr)]roi\)roro[)rpi^rodtype test_new.fitsappendrNr r rocztest_append.fitsztest_update.fitsupdateZrootnameZ U2EQ0201Tabci;)rr r8r+r)?rzrzrzrzint32)rr+css|]}t|VqdSrfrgr<rrrrhorir aCBITPIX = 32 / array data type NAXIS = 1 / number of array dimensions NAXIS1 = 2 PCOUNT = 0 / number of parameters )rr*r+rjoinZcardsZrename_keywordr-r. ValueErrorrQ array_equalarrayr;rTrtflushosrenamerUreadallZonesfloat32r)rr3nrZur+rhdu2rrrtest_io_manipulations"  . .  0"8   z'TestImageFunctions.test_io_manipulationcCs tj|ddd}|dS)Nr(r )Zmemmap)rr*r+r,)rf1rrrtest_memory_mappingusz&TestImageFunctions.test_memory_mappingcCst}t|}tjtdd}|Wdn1s>0Yt|dksXJtjtdd"}|| ddWdn1s0Yt|dksJdS)Nz,HDUList's 0th element is not a primary HDU\.matchrzQHDUList's 0th element is not a primary HDU\. Fixed by inserting one as 0th HDU\.ztest_new2.fitsZfix) rrr#r-warnsrverifyrHrUrT)rr>rwrrrtest_verification_on_outputzs &0z.TestImageFunctions.test_verification_on_outputc Cs`t|d}t|djdds*Jt|djddddftgdsXJt|djddddftgd sJt|djdddd ftgd sJt|djddd d ftgd sJt|djdddddftgdgdgdgs$Jt|djdddddfddddftgdgdgdgsvJ|dj}t|djddddd f|ddddd fsJt|djddddf|ddddfsJt|djddddddfddddddftgdgdgdggdgdgdggdgdgdggsvJt|djddddddfddddddftddgddggddgd d!ggd"d#gd$d%gggsJt|djdddddf|dddddfs&Jt|djddddddf|ddddddfsjJt|djdddddf|dddddfsJt|djdddd&ddf|dddd&ddfsJt|djdddddf|dddddfs&Jt|djddgd'df|ddgd'dfsbJtgd(}t|djdd|ddf|dd|ddfsJt|djdddddd)f|dddddd)fsJt|djd)dddf|d)dddfs Jt|djd)gd'df|d)gd'dfsTJ|dS)*Nz arange.fitsr)rrr err) `abcdrfghijr)rrrrrrr)rrrrrrrri)rrrrrr ) ikiliminioipiqirisitiu) iviwixiyizi{i|i}i~ii)iJiKiL)iUiViW)rrrrF)iii)iii)iii)i&i'i()i1i2i3)i<i=i>r noyzrK)r rr) TFTTFFTTFT.)rr*r+rQr~sectionrr,)rZfsdatZ bool_indexrrr test_sections      .   <48  8  <D<D<  2 4 zTestImageFunctions.test_sectioncCstdg}t|}||dt|d}|dj}|dj}t |d|dsbJt |d|dszJt |d|dsJt |d|dsJ| dS)Nr rsr.).r)r.) rQrrr$rUrTr*rr+r~r,)rrZrrbZsecrrrrtest_section_data_singles    z+TestImageFunctions.test_section_data_singlecCsvtddd}t|}||dt|d}|d}|dj}|j ddddf|ddddfk sJ|j dddf|dddfk sJ|j dddf|dddfk sJ|j dddf|dddfk sJ|j dddf|dddfk s*J|j d|dk sFJ|j d|dk sbJ|j d|dk s~J|j d |d k sJ|j ddddf|ddddfk sJ|j ddddf|ddddfk sJ|j ddddf|ddddfk s6J|j ddddf|ddddfk sjJ| dS) Nrrrsrr rrrr r rru rQrRrSrr$rUrTr*r+rallr,rrZrrbdrrrrtest_section_data_squares(  2***,4444z+TestImageFunctions.test_section_data_squarec Cs0tdddd}t|}||dt|d}|d}|dj}|j dd|ddk srJ|j ddddf|ddddfk sJt dddfD]T}t ddddfD]>}t ddddfD](}|||f} |j | || k sJqqqt ddt ddt ddt ddt ddt ddfD]4}t dt d|f} |j | || k s4$D<D<Dz)TestImageFunctions.test_section_data_fourcCst|d}|d}|dj}|jddddf|ddddfksTJ|jdddf|dddfks~J|jdddf|dddfksJ|jdddf|dddfksJ|jdddf|dddfksJ|jd|dksJ|jd|dks4J|jd|dksPJ|jd|dkslJ|jddddf|ddddfksJ|jdd ddf|dd ddfksJ|jdddd f|dddd fksJ|jdd dd f|dd dd fksi2r)rr*r+rrrQrrbrrrtest_do_not_scale_image_dataVs8z/TestImageFunctions.test_do_not_scale_image_datacCstj|dtjgdddtddgd}tj|d|dtj|ddd$}|d j j dkspJWd n1s0Yd S) zRegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/56 (BZERO and BSCALE added in the wrong location when appending scaled data) rsuint8rqrLduint16Tuintr N) rrUrTrQrzerosastypertr*r+rr)rrfrrrtest_append_uint_data]s z(TestImageFunctions.test_append_uint_datacCs(ttjdd}|jdddddS)zT Regression test for https://github.com/astropy/astropy/issues/6399 rrr r)typebscalebzeroN)rrrQrandomZrandscale)rrrrr%test_scale_with_explicit_bzero_bscaleksz8TestImageFunctions.test_scale_with_explicit_bzero_bscalec CsxdD]l}d|d}|dkr(t|}d|}tjd|d}|||tjd|d8}tj|d}t|j|ks|J|jj j d|ksJd |j vsJ|j d d|dksJd|d }| | |tj| |d d t}|d }t|j|ks J|jj j d|ks&Jd |j vs6J|j d d|dksRJWdq1sh0YqdS)z Regression test for https://github.com/astropy/astropy/issues/2305 This ensures that an HDU containing unsigned integer data always has the appropriate BZERO value in its header. ) @rr rrrrqrLBZEROz.fitsTrrN)rQZuint64emptyZfillrRrr$rr+rrrrrUrTr*) rZint_sizeZmax_uintrrarruint_hdurkrbZ new_uint_hdurrrtest_uint_header_consistencyts*       z/TestImageFunctions.test_uint_header_consistency from_file)FT do_not_scale)Fc Cs^tjddd}|rtt|}d}|||tj|||d}|d}|j}Wdq1sh0Yntj||d}d|jvsJd |jvsJ|jdd ksJ|jd d ksJ|jd |_|jd dksJd|jvsJd |jvsJd}|||t||(} | djdk s:JWdn1sP0YdS)a< Regression test for https://github.com/astropy/astropy/issues/4974 BZERO/BSCALE should be removed if data is converted to a floating point type. Currently excluding the case where do_not_scale_image_data=True because it is not clear what the expectation should be. rrrqzunsigned_int.fitsrrNBSCALErr irzBITPIXztest_uint_to_float.fits) rQrrr$rUrTr*r+rr) rrrrZtmp_uintrkrr_rbrrr5test_uint_header_keywords_removed_after_bitpix_changes4  & zHTestImageFunctions.test_uint_header_keywords_removed_after_bitpix_changecCstjdtjd}d|d<tj|d}d|jd<||dt|d,}t |dj d slJWdn1s0YdS) zhTest image data with blank spots in it (which should show up as NaNs in the data array.  rrqir rLBLANKrsN) rQrr{rrrrUrTr*isnanr+rrrrrbrrr test_blankss  zTestImageFunctions.test_blanksc Cstjdtjd}tj|d}d|jd<tjtdd }| | dWd n1sZ0Yt |d kstJtjtddX}t | d(}t ||d jksJWd n1s0YWd n1s0Yt |d ksJd S) z Test that invalid use of the BLANK keyword leads to an appropriate warning, and that the BLANK keyword is ignored when returning the HDU data. Regression test for https://github.com/astropy/astropy/issues/3865 r rqrLrr!Invalid 'BLANK' keyword in headerrrsNr r)rQrRfloat64rr$rr-rrrUrTrHr*rr+)rrrrhrrrtest_invalid_blankss   .Tz&TestImageFunctions.test_invalid_blanksz(ignore:Invalid 'BLANK' keyword in headerc Cstjdtjd}tj|d}|jddd|d}d|jd <d|jd <| |t |n}|d j}t |d szJt j tjjd d  | |d Wdn1s0YWdn1s0Yt |d @}d |d jvsJ|d j}t |d sJWdn1s20Ytj |ddd.}|d j}t |d slJWdn1s0Ytj |ddX}|d jd dksJ|d jddksJ|d jd dksJWdn1s0YdS)a Test that when auto-rescaling integer data with "blank" values (where the blanks are replaced by NaN in the float data), that the "BLANK" keyword is removed from the header. Further, test that when using the ``scale_back=True`` option the blank values are restored properly. Regression test for https://github.com/astropy/astropy/issues/3865 r rqrLr;gGz?rrai'rrrr test2.fitsNTrw) scale_backrOrr)rQrRr{rr$rrTr+rrUr*rr-rr VerifyWarning) rrrrkrbr+hdul2Zhdul3Zhdul4rrrtest_scale_back_with_blankss>        L 4 4z.TestImageFunctions.test_scale_back_with_blankscCstdd}tj|d}d|jd<||dt|d.}|d7}|dj|k sdJWdn1sx0YdS)zUTest use of the BZERO keyword in an image HDU containing float data. rr rLrzrrsN) rQrrrrrUrTr*r+rrrrrtest_bzero_with_floats5s  z)TestImageFunctions.test_bzero_with_floatscCsbt|d}|dj}|j|ddd|t|d}|dj|ks\J|t|d}|j|ddd|t|d}|dj|ksJ|tj|ddd}|j|dddd|t|d}|dj}|tj|dd d }|t|d}|dj|ksVJ|d S) zRegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/84 and https://aeon.stsci.edu/ssb/trac/pyfits/ticket/101 fixed-1890.fitsrrsT overwriter silentfixrZ output_verifyrwrNN)rr*r+rUrTr,r)rrb orig_datarrr!test_rewriting_large_scaled_imageCs8   z4TestImageFunctions.test_rewriting_large_scaled_imagecCs|dtj|ddd<}|dj}|dj}|d=||d_Wdn1s`0Yt|d&}||djksJWdn1s0YdS)z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/105 Replacing the original header to an image HDU and saving should update the NAXISn keywords appropriately and save the image data correctly. r(rwrNr zNAXIS*N) copy_filerr*rTr+copyrr)rrbrZhdr_copyrrrtest_image_update_headeris (z+TestImageFunctions.test_image_update_headercCs|dt|dj}tdtj|ddd |t|djksXJtdt|dd}|dj }| |t|djksJtj|ddd}|dj j t dksJ|dj ddksJd |dj vsJd |dj vs J||dj ks"Jd |dj _| t|d}|djd ksZJ|dj j t dksvJ|dj ddksJd |dj vsJd |dj vsJ| d S) al Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/119 (Don't update scaled image data if the data is not read) This ensures that merely opening and closing a file containing scaled image data does not cause any change to the data (or the header). Changes should only occur if the data is accessed. rr rwrNrz>f4rrr*rN)rrstatrTst_mtimetimesleeprr*r,r+rrrQrrshape)rmtimerbrrrrtest_open_scaled_in_update_mode|s2     z2TestImageFunctions.test_open_scaled_in_update_modecCs|dtj|ddddV}|djd}|djd}|djd}|dj}d|djd<Wd n1s|0Ytj|ddd }|djd|ksJ|djd|ksJ|djd|ksJtt | |}|djd|k sJWd n1s$0Yt|d8}|djd d |d d k shJWd n1s~0Yd S) z|A simple test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/120 The scale_back feature for image HDUs. rrwTrOrrrrrNrr ) rrr*rTrr+rintmathfloorr)rrb orig_bitpix orig_bzero orig_bscaler zero_pointrrrtest_scale_backs* , <z"TestImageFunctions.test_scale_backcCst|d8}|djd|d_|d|dWdn1sN0Yt|d\}|djdus|J|djddksJd|djvsJd|djvsJWdn1s0YdS) zX Regression test for https://github.com/spacetelescope/PyFITS/issues/27 r(r NraNAXISrNAXIS1NAXIS2)rr*r+rUrTr)rrrrrtest_image_nones  2z"TestImageFunctions.test_image_nonec Cs$tjdtjd}t|}d|jd<tjtjj dd | | dWdn1s\0Ytt X}t | d(}t|d j|ksJWdn1s0YWdn1s0Yt|d ksJd }|t|d jvsJd }|t|d jvs JdS)z Regression test for https://github.com/astropy/astropy/issues/2711 If the BLANK keyword contains an invalid value it should be ignored for any calculations (though a warning should be issued). rrqnanrz2Invalid value for 'BLANK' keyword in header: 'nan'rraNrrz+Invalid value for 'BLANK' keyword in headerzInvalid 'BLANK' keywordr )rQrRrrr$rr-rrrrUrTrr*rr+rHr0message)rr+rrrbmsgrrrtest_invalid_blanks  . Tz%TestImageFunctions.test_invalid_blankcCstjdtjd}tj|d}|jdd||dt |dd}| }Wdn1sj0Ytj |}t |d j|sJdS) zT Regression test for https://github.com/astropy/astropy/issues/2710 rrqrLg?rrarbNr)rQrRrrr$rrrUrTr*readr#Z fromstringallcloser+)rrZrrZ file_datarbrrrtest_scaled_image_fromfiles & z-TestImageFunctions.test_scaled_image_fromfilecCst}td}||_dS)z4 Test data assignment - issue #5087 rN)rrrQrRr+)rZimZarrrr test_set_datas z TestImageFunctions.test_set_datacCsftjddtjd}tj|d}tj|d}|jddd|jdddt|j|jsbJd S) zT Regression test for https://github.com/astropy/astropy/issues/4600 rrqrLr;X@rrvN) rQrRr;rr$rrrr+rrZhdu1rrrrtest_scale_bzero_with_int_datas z1TestImageFunctions.test_scale_bzero_with_int_datacCstjddtjd}t||dtj|dddd2\}d|jd d <t |jdsbJWd n1sv0Yd S) z Extend fix for #4600 to assignment to data Suggested by: https://github.com/astropy/astropy/pull/4602#issuecomment-208713748 rrrqrarwTrrN) rQrRrrr$rUrTr*r+r)rrZrrrrtest_scale_back_uint_assignmentsz2TestImageFunctions.test_scale_back_uint_assignmentcCs|d}tjt|dtjddWdn1s40Ytjt|dtjddWdn1sn0YdS)Nz9data object array\(1\) should have at least one dimensionrr rL)r-r. TypeErrorrrr$)rrrrrtest_hdu_creation_with_scalar-s *z0TestImageFunctions.test_hdu_creation_with_scalarN)-__name__ __module__ __qualname__rr"r'r5rDrIr]r_rcrrrrrrrrrrrrrr-mark parametrizerrrfilterwarningsrrrrrrr rrrrrrrrrrr sT " C&- $   8  8&4 r c @seZdZddZejdejdej dddfejdej ddd fejdej dd d fed d d dfedddfgejdddgddZ ej dddZ ddZ ddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Zdd?Z d@dAZ!dBdCZ"dDdEZ#dFdGZ$dHdIZ%dJdKZ&ejdLdMej'ej fdMej(ej fdMej)ej*fdNej'ej fdNej(ej fdNej)ej*fgdOdPZ+ejdQej'ej(ej,ej)ej-fdRdSZ.dTdUZ/dVdWZ0dXS)YTestCompressedImagecCs&t}|jdusJ||dtj|ddd\}t|dksLJt|dtjs`J|djdusrJtj dtj d|d_Wdn1s0Yt|dZ}t|dksJt|dtjsJt |djtj dtj dksJWdn1s0YdS) zT Regression test for https://github.com/astropy/astropy/issues/2595 NrarwrNrr rrq) r CompImageHDUr+rUrTr*rHrGrQrRr{rrrrbrrr test_empty6s4zTestCompressedImage.test_emptyr+compression_typequantize_levelrrrrqRICE_1rGZIP_1g{GzZGZIP_2rrr HCOMPRESS_1rZPLIO_1 byte_order<>c Cs||}t}t|}tj|d||d}|||j|ddd|t |d}|dj |k s|J|dj d|j dksJ|dj d|j dksJ|dj d |j d ksJ|dj d |j d ksJWdn1s0YdS) Nr8)rr)r*rsTrr rr r r) Z newbyteorderrr$r#r%rtrUrTr,r*r+rr) rr+r)r*r0r%r&chdur\rrrtest_comp_imageMs   z#TestCompressedImage.test_comp_imagez not HAS_SCIPYcCsHddl}tjd|jtjddd}t| | dtj |ddd d d  | d tj |ddd d d  | dt | d}t | d }t | d}t ||rJtjt||dddsJtjt||dddsJtjt||ddds$Jtjt||dddsDJdS)z Regression test for https://github.com/astropy/astropy/issues/5969 Test that quantize_level is used. rNrirzim1.fitsr,r r )r)quantize_methodr* dither_seedzim2.fitsizim3.fitsggMbP?)Zatolg?g?2)Z scipy.miscrQrseedZmiscZascentZrandnrrrUrTr%rXr~Ziscloseminmax)rZscipyr+Zim1Zim2Zim3rrrtest_comp_image_quantize_levelgs,     z2TestCompressedImage.test_comp_image_quantize_levelc Cs.tjttjtjdtjddddgdddS)z Tests compression with the HCOMPRESS_1 algorithm with data that is not 2D and has a non-2D tile size. r+rqr8r/r)rr)r* tile_sizeN)r-r.r}rr%rQrr)rrrr+test_comp_image_hcompression_1_invalid_datas  z?TestCompressedImage.test_comp_image_hcompression_1_invalid_datacCstjdtjdddd}tj|dddgdd }||d t|d 0}t |dj | d kstJWd n1s0Yd S) a Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/171 Tests that data containing more than two dimensions can be compressed with HCOMPRESS_1 so long as the user-supplied tile size can be flattened to two dimensions. i,rqrrr8r/r)r r r )r+rr)r*r>rag?N) rQrRrrSrr%rUrTr*absr+r<)rZcuberrbrrr%test_comp_image_hcompress_image_stacks z9TestCompressedImage.test_comp_image_hcompress_image_stackcCstddd}|dddd}tj|ttd}| | dt | d}t |dtjsrJd |dj vsJ|dj d d ksJd |dj vsJ|dj d |ksJt|dj|ksJWd n1s0Yd S) aF Regression test for https://github.com/spacetelescope/PyFITS/issues/32 Ensure that when floating point data is compressed with the SUBTRACTIVE_DITHER_1 quantization method that the correct ZDITHER0 seed is added to the header, and that the data can be correctly decompressed. gY@rrri'r )r+r6r7raZZQUANTIZrZZDITHER0N)rQrRrSZviewsumrr%rrrUrTr*rG_headerrr+)rrZcsumrrbrrrtest_subtractive_dither_seeds z0TestCompressedImage.test_subtractive_dither_seedcCstj|ddd8}t|dtjs*Jt|dtjr>JWdn1sR0Yt|d$}t|dtjsJWdn1s0YdS)N comp.fitsTZdisable_image_compressionr )rr*r+rGZ BinTableHDUr%rrrrtest_disable_image_compressions 2z2TestCompressedImage.test_disable_image_compressioncCs\|dt|dj}tdtj|ddd |t|djksXJdS)z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/167 Similar to test_open_scaled_in_update_mode(), but specifically for compressed images. rEr rwrNN) rrrrTrrrrr*r,)rrrrr#test_open_comp_image_in_update_modes  z7TestCompressedImage.test_open_comp_image_in_update_modecCstj|ddd:}tj|dj|djd}||dWdn1sT0Yt|dj }t dtj|ddd  |t|dj ksJt dt|dd}|dj| |t|dj ksJtj|ddd }|djj t d ks$J|djd d ks  .    z>TestCompressedImage.test_open_scaled_in_update_mode_compressedc Cst|d$}|d|dWdn1s:0Yt|dx}t|dH}t|dj|djksJt|dj|djsJWdn1s0YWdn1s0YdS)NrEr ra) rr*r+rUrTrQrr compressed_datarrbhdul1rrrr(test_write_comp_hdu_direct_from_existings2 z  ., <(z.TestCompressedImage.test_scale_back_compressedcCstjd}|jdd}tj|dd}||dt|d0}t ||dj  dkshJWd n1s|0Y~tj|ddd }|j|dd d t|d&}||dj k sJWd n1s0Yd S) zERegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/198r)rQsizer-)r+r)rar gNr(Tr) rQrZ default_rngnormalrr%rUrTr*r@r+r<r)rrngZnoiseZchdu1rZchdu2rrrtest_lossless_gzip_compressionts  >z2TestCompressedImage.test_lossless_gzip_compressioncCstjddtjddddtjftdd}tjdtjjdd }||d|jd d|jdf<tj |d d d }| | dtj | dddD}t d|djdsJt d|djdsJWdn1s0YdS)zERegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/199r rrqrNrKi9)rRrr,)rFrK)r)r>raTrFz ^1PB\(\d+\)$ZTFORM1ZTFORM2)rQrRrZnewaxisrr:Zuniformrrr%rUrTr*rerr)rZdata2Zdata1r3rrrrtest_compression_column_tformss "    z2TestCompressedImage.test_compression_column_tformscCs|dtj|ddd@}t|dtjs4Jd|djd<d|djd<Wdn1sd0Yt|d`}d|djvsJ|djddksJd|djvsJ|djddksJWdn1s0Ytj|ddd(}|dj}d ||d <Wdn1s20Yt|d(}|djd d ksfJWdn1s|0Ytj|ddd}d |djd <Wdn1s0Yt|d4}t |djd  d d gksJWdn1s0Ytj|ddd4}|dj}|d }d |||d<Wdn1sp0Yt|d4}t |djd  d d gksJWdn1s0Ytj|ddd}d|djd<Wdn1s 0Yt|dH}|djdddksBJ|djdddks^JWdn1st0Ytj|ddd8}|dj}|d=|d }|||d=Wdn1s0Yt|d}d|djvsJd|djvsJd |djvs*Jd |djvs>Jd|djvsRJd|djvsfJWdn1s|0YdS)zWRegression test for https://github.com/spacetelescope/PyFITS/issues/23 rErwrNr testZtest1Ztest2NZfooZTEST1quxzTEST*barrzI am fire. I am death!)COMMENTr r^ZTEST2) rrr*rTrGr%rrCindexlistvalues)rrbridxrrrtest_compression_update_headersT ,4 .8.D  0D.<  .z2TestCompressedImage.test_compression_update_headercCsvdd}t|dJ}|dj}||dd||dd||d d ||d dWd n1sh0Yd S) z Ensure that setting reserved keywords related to the table data structure on CompImageHDU image headers fails. cSsrtt}|||<Wdn1s(0Yt|dksBJt|djd|dsbJ||vsnJdS)Nr rzKeyword z is reserved)r-r UserWarningrHr0r  startswith)rkeywordr1rrrrtest_set_keywords & zZTestCompressedImage.test_compression_update_header_with_reserved..test_set_keywordrEr TFIELDSrZTTYPE1ZFooZZCMPTYPEZASDFZZVAL1N)rr*r+r)rrgrbrrrr,test_compression_update_header_with_reserveds    z@TestCompressedImage.test_compression_update_header_with_reservedc Cs8t|d}|dj}|dj}tjtdd}|dWdn1sV0Yt |dkspJd|vs|J|jdddd |vsJ|d d ksJd |vsJ|d d ksJ|d d |vsJ|d dksJd |vsJd|vsJ|ddksJWdn1s*0YdS)NrEr Keyword 'TFIELDS' is reservedrrh)rr]r\T)endrr5r])CHECKSUMabcd1234rlrmZZHECKSUM) rr*r+rrCr-rrdrtrHrrbimghdrtblhdrrrrrtest_compression_header_appends$  (      z2TestCompressedImage.test_compression_header_appendcCsrt|dN}|dj}t|dkr2|q|d|d|dWdn1sd0YdS)zZ Regression test for issue https://github.com/astropy/astropy/issues/5827 rEr )ZQ1_OSAVGr z[adu] quadrant 1 overscan mean)ZQ1_OSSTDr z [adu] quadrant 1 overscan stddev)ZQ1_OSMEDr z [adu] quadrant 1 overscan medianN)rr*r+rrHrt)rrbrrrrtest_compression_header_append2 s     z3TestCompressedImage.test_compression_header_append2c Cstt|dL}|dj}|dj}tjtdd}|ddWdn1sX0Yt |dksrJd|vs~J| ddksJ|ddd |vsJ| d | ddksJd |vsJ| d | ddksJ| d }|d d d |vsJ| d |ks$Jd |vs2J| d | d dksPJWdn1sf0YdS) NrEr rjrrrrhZTELESCOP)OBSERVERz Phil Plaitrt)rr) rr*r+rrCr-rrdinsertrHcountr_)rrbrorprrbrrrtest_compression_header_inserts&  *      z2TestCompressedImage.test_compression_header_insertc Cst|d}|dj}|dj}tjtdd"}|jddddd Wdn1s^0Yt |dksxJd|vsJ| ddksJ|ddksJ|jd d d d | d | d dksJ|d d ksJ| d| ddksJ|dd ksJ| d dks$J| d dks8J|d dksJJ|jd dd d | d | d dksxJ|d dksJ| d| ddksJ|ddksJ| d dksJ| d dksJ|d dksJWdn1s 0YdS)NrEr zKeyword 'ZBITPIX' is reserved rZBITPIXMZasdfZXTENSION)ZafterZGCOUNTrvZPCOUNT)ZbeforeZZGCOUNTZZPCOUNTr rFr) rr*r+rrCr-rrdsetrHrvr_rnrrr(test_compression_header_set_before_after2s2  0 zTestCompressedImage.test_duplicate_compression_header_keywordscCsftjddtjd}tj|d}tj|d}|jddd|jdddt|j|jsbJd S) z Regression test for https://github.com/astropy/astropy/issues/4600 and https://github.com/astropy/astropy/issues/4588 Identical to test_scale_bzero_with_int_data() but uses a compressed image. rrrqrLr;rrrvN) rQrRr;rr%rrrr+rrrr)test_scale_bzero_with_compressed_int_datas z=TestCompressedImage.test_scale_bzero_with_compressed_int_datacCstjddtjd}t||dtj|dddd8}d|d jd d <t |d jdshJWd n1s|0Yd S) z Extend fix for #4600 to assignment to data Identical to test_scale_back_uint_assignment() but uses a compressed image. Suggested by: https://github.com/astropy/astropy/pull/4602#issuecomment-208713748 rrrqrarwTrrr N) rQrRrrr%rUrTr*r+r)rrZrbrrr*test_scale_back_compressed_uint_assignments z>TestCompressedImage.test_scale_back_compressed_uint_assignmentcCstjddtjd}t|}|jdtt |j Wdn1sN0Yt|}|jdtt |j Wdn1s0YdS)NrrrqZZNAXISrx) rQrRrrr%rCpopr-r.KeyErrorrK)rrZr~rrr%test_compressed_header_missing_znaxiss   $   z9TestCompressedImage.test_compressed_header_missing_znaxisc Cst|d}|d}|jjd}t|dks8J|jdksFJd|jvsTJ|j|jdkshJ|jjd}t|dksJd}||_|j|ksJ|jd|ksJ|jd|ksJ|jd|ksJd|jd<|jdksJ|j | dd d t| dB}|d}t|jjddks2J|jdksBJWd n1sX0Y|jd=d |_|jd ksJt t d |_Wd n1s0YWd n1s0Yd S)z^Test that a double EXTNAME with one default value does not mask the non-default value.zdouble_ext.fitsr rrZccd00ZNEW_NAMEZNEW2ztmp.fitsTrNzRE-ADDEDr)rr*r+rC_keyword_indicesrHrrZ _image_headerrUrTr-r.r)rrbrindicesnew_namerMrrrr%test_compressed_header_double_extnames6   0 z9TestCompressedImage.test_compressed_header_double_extnamecCsd}tjtd|d}|jd|ks*J|jd|ksZx0Zx1Zhdusrrrtest_comphdu_bscaleUs        ( rcCs(ttjdgtjd}|jdddS)Nr rqg?r)rrrQrr{r)rrrrtest_scale_implicit_castingtsrcCsDtd}t|}|d}|jWdn1s60YdS)Nz data/compressed_float_bzero.fitsr )rrr*r+)rkrbrrrr&test_bzero_implicit_casting_compressed}s  rcCsj|dj}ttd}d|jd<|j|ddt|}| Wdn1s\0YdS)Nzfloatimg_with_bzero.fitsrrrTr) r|rrrrQrrrUr*rC)rrkrrbrrrtest_bzero_mishandled_infos    rcCstgd}|jddtj|d}||dj}||t |$}t |dj gdWdn1sx0Ytgd}|jddtj |d}|dj}||t |$}t |dj gdWdn1s0YdS) N)r rrF)writerLrar )rzg@g@r) rQrZsetflagsrrZ add_datasumr|rrUr*rr+r%)rr>Zghdurkrrrrtest_image_write_readonlys     2     rcCstjddtjddd}t|}||dt|dx}|djddksXJ|djd d ksnJ|djd d ksJt |dj ||dj j |j ksJWd n1s0Yd S)zFTest for int8 support, https://github.com/astropy/astropy/issues/11995r8r9rqrz int8.fitsrrrrirrzN) rQrRZint8rSrr$rUr*rrr+rr)Ztmp_pathZimgrrbrrr test_int8s r) rrrYrr-ZnumpyrQZ numpy.testingrZ astropy.iorZastropy.io.fits.hdu.compressedrrZastropy.utils.exceptionsrZastropy.utils.datarZ"astropy.utils.compat.optional_depsrZ test_tabler r r r r$rrrrrrrrrrsF       '&