U e{j"@sdddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z Gddde Z dS) N)fits)_NOT_OVERWRITING_MSG_MATCH) FitsTestCase)comparerecordsc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)TestGroupsFunctionsc Csvt|dZ}t|dtjs(td}dddddg}dd d d d |d dfg}|jdd|ksdt|d}|j|ksztt|jj j |dgkstt|jtj st|j dt |jkst|jjjt |jf|dddkst|jj|kstt|jdtjstt |jdt |d ks0t|jdjj|dddksRt|jdj|kshtW5QRXdS)Nrandom_groups.fitsr)rrrUUVVWWBASELINEDATEZPRIMARYr GroupsHDUfloat32z3 Groups 5 ParametersF)outputDATAGCOUNT)ropendata isinstancerAssertionErrorinfoparnameslistdtypenames GroupDataheaderlenshapeGroup)selfhdulZnaxes parametersrghdur)`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/fits/tests/test_groups.py test_opens4 ( "zTestGroupsFunctions.test_opencCs^|dt|dj}tdtj|dddd |t|djksZt dS)a Test that opening a file containing a groups HDU in update mode and then immediately closing it does not result in any unnecessary file modifications. Similar to test_image.TestImageFunctions.test_open_scaled_in_update_mode(). rrupdateF)modeZmemmapN) copy_fileosstattempst_mtimetimesleeprrcloser)r%mtimer)r)r*test_open_groups_in_update_mode3s  z3TestGroupsFunctions.test_open_groups_in_update_modec Csx|dtj|ddd}d|djd<W5QRXtj|ddd"}t|djddksjtW5QRXdS)z Regression test for https://github.com/astropy/astropy/issues/3730 and for https://github.com/spacetelescope/PyFITS/issues/102 rr,r-gzG?rr N)r.rrr1rnpallr)r%hr)r)r*test_random_groups_data_updateIs  z2TestGroupsFunctions.test_random_groups_data_updatec Cs|ddddddg}tj|ddd }|d j|ks@t|W5QRXt|d(}|d j|ksvt||d W5QRXt|d }|d j|kstW5QRXd S) z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/130 Ensures that opening a random groups file in update mode or writing it to a new file does not cause any change to the parameter names. rr r r rrr,r8r test.fitsN)r.rrr1rrflushwriteto)r%r'r;r)r)r*test_parnames_round_tripVs z,TestGroupsFunctions.test_parnames_round_tripc Csht|dN}|djdd}t|tjs4tt|dksDt|djj|jksZtW5QRXdS)a A simple test to ensure that slicing GroupData returns a new, smaller GroupData object, as is the case with a normal FITS_rec. This is a regression test for an as-of-yet unreported issue where slicing GroupData returned a single Group record. rrrN)rrrrr rr"r)r%r&sr)r)r*test_groupdata_sliceps z(TestGroupsFunctions.test_groupdata_slicec Cst|d}|djd}|dd}t|dkss