U e{j@sPddlZddlZddlZddlmZddlmZddlm Z Gddde Z dS)N)assert_array_equal)fits) FitsTestCasec@sVeZdZeddZejddddddgd d Zejd d d dZ ddZ dS)TestUintFunctionscCs@d|_tjtjtjd|_tjtjtjd|_ dddd|_ dS)Nu2u4u8IJK) Zutypesnpuint16Zuint32Zuint64 utype_mapint16Zint32int64 itype_map format_map)clsr^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/fits/tests/test_uint.py setup_classszTestUintFunctions.setup_class)utype compressed)rF)r F)r F)rT)r Tc Csdt|d}tddks*|dkr|rXttjdddddd d gtjd }d}n(ttjdddddd d gtjd }d}|j d |d dd |dd|j | dddtj | ddd}||j j|j|kst||j tjd |d d |d d |dddd d g|j|d ks4t| | dtj | dddv}||j }||j } || kst|s||jdd} | jjd|kst| |ddkstW5QRXW5QRXdS)Nrr64bit@dtypeintd)bzero tempfile.fitsT overwriteuintztempfile1.fits)rr-)r%platform architecturerZ CompImageHDUrarrayr PrimaryHDUscalewritetotempopendatar$rAssertionErrorallsectionname) selfrrbitsZhduZ hdu_numberZhdulZhdul1Zd1Zd2secrrr test_uints> (  zTestUintFunctions.test_uintrrc CsXdt|d}tddks*|dkrT|j|d|d}|j|d}tj|d|j|d}d||}|dkr|||d<|d|d||d<|||j|}tj |||j ||d }tj |g} | j ||kst| j jj||ks tt} t| | g} | j|d d d ~ tj|d d d X} | dj } | ||kslt| |j|j|kst| j||kstW5QRX|j||j|fgd}tj|d|d d tj|dd d d}|dj }|j|| j jj|kst||| j |ks4t|||ksJtW5QRXdS)zTest basic functionality of tables with columns containing pseudo-unsigned integers. See https://github.com/astropy/astropy/pull/906 rrrrrr!r#?)r:r0formatr(r)Tr*r,ztempfile2.fitsN)r%r.r/rrarangeviewrrColumnr BinTableHDU from_columnsr6r8r7baser1ZHDUListr3r4r5r$)r;rr<r(oneZu0uuucoltableZhdu0hdulistZhdulist2ZhdudatavZhdulist3Zhdudata3rrrtest_uint_columnsCsF   " "z#TestUintFunctions.test_uint_columnsc Cs(tjdtjd}tjtjddtjdtjddtjdddd|d g}|| d t | d }|dj }W5QRXt |d||dd k}t ||d||||dj ||j kstt | d }|dj }W5QRXt ||d||||dj ||j ks$td S) z Fix for https://github.com/astropy/astropy/issues/5490 if data is sliced first, make sure the data is still converted as uint ir#ar )r:r@r0bri)r:r@Zbscaler(r0ztest_scaled_slicing.fitsrN)rrArrrDrErCrr3r4r5r6rr$r7)r;ZdatarefZtbhdurLr6selrrrtest_uint_sliceys6  z!TestUintFunctions.test_uint_sliceN) __name__ __module__ __qualname__ classmethodrpytestmarkZ parametrizer>rNrRrrrrrs   '  5r) r.numpyrrWZ numpy.testingrZ astropy.iorZconftestrrrrrrs