a ߙfb&@sPddlZddlZddlZddlmZddlmZddlm Z Gddde Z dS)N)assert_array_equal)fits) FitsTestCasec@sPeZdZeddZejdgdddZejddd d Z d d Z d S)TestUintFunctionscCs@d|_tjtjtjd|_tjtjtjd|_ dddd|_ dS)Nu2u4u8IJK) Zutypesnpuint16Zuint32Zuint64 utype_mapint16Zint32int64 itype_map format_map)clsr>lib/python3.9/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|rNttjgdtjd}d}nttjgdtjd}d}|j d|d d d |dd |j | d ddtj | d dd.}||j j|j|ksJ||j tjd |dd |d d |dddd dg|j|dks J| | dtj | ddd}||j }||j } || kslJ|s||jdd} | jjd|ksJ| |ddksJWdn1s0YWdn1s0YdS)Nrr64bit@)rrdtypeintdr!)bzero tempfile.fitsTZ overwriteuintr"ztempfile1.fits)rr,)r%platform architecturerZ CompImageHDUrarrayr PrimaryHDUZscalewritetotempopendatar$rallZsectionname) selfrrbitsZhduZ hdu_numberZhdulZhdul1Zd1Zd2Zsecrrr test_uints:     zTestUintFunctions.test_uintrrcCsdt|d}tddks*|dkr|j|d|d}|j|d}tj|d|j|d}d||}|dkr|||d<|d|d||d<|||j|}tj |||j ||d }tj |g} | j ||ksJ| j jj||ks Jt} t| | g} | j|d d d ~ tj|d d d b} | dj } | ||kslJ| |j|j|ksJ| j||ksJWdn1s0Y|j||j|fgd}tj|d|d d tj|dd d n}|dj }|j|| j jj|ks.J||| j |ksJJ|||ks`JWdn1sv0YdS)zTest basic functionality of tables with columns containing pseudo-unsigned integers. See https://github.com/astropy/astropy/pull/906 rrrrrr!r#?)r6r/formatr(r)Tr*r+Nztempfile2.fits)r%r-r.rrarangeZviewrrColumnr BinTableHDU from_columnsr4r5baser0ZHDUListr1r2r3r$)r7rr8r(ZoneZu0uuucoltableZhdu0hdulistZhdulist2ZhdudatavZhdulist3Zhdudata3rrrtest_uint_columns;sD    8  z#TestUintFunctions.test_uint_columnsc CsRtjdtjd}tjtjddtjdtjddtjdddd|d g}|| d t | d }|dj }Wd n1s0Yt |d||dd k}t ||d||||dj ||j ksJt | d }|dj }Wd n1s0Yt ||d||||dj ||j ksNJd 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 )r6r;r/bri)r6r;Zbscaler(r/ztest_scaled_slicing.fitsNr)rr<rrr>r?r=rr1r2r3r4rr$)r7ZdatarefZtbhdurEr4Zselrrrtest_uint_sliceqs( ( *z!TestUintFunctions.test_uint_sliceN) __name__ __module__ __qualname__ classmethodrpytestZmarkZ parametrizer9rGrJrrrrr s  !  5r) r-rOZnumpyrZ numpy.testingrZ astropy.iorr'rrrrrrs