a
    ß™fbR  ã                   @   s4   d dl Zd dlmZ ddlmZ G dd„ deƒZdS )é    N)Úfitsé   )ÚFitsTestCasec                   @   s8   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ ZdS )ÚTestDivisionFunctionsz6Test code units that rely on correct integer division.c                 C   sf   t  |  d¡¡}|d j ¡ }W d   ƒ n1 s40    Y  tjj|t g d¢¡t|ƒd d d S )Nútb.fitsr   ))Zc1z>i4)Zc2z|S3)Zc3z>f4)Zc4z|i1é   )ÚdtypeÚshape)	r   ÚopenÚdataÚtobytesÚnpZrecÚarrayr   Úlen)ÚselfÚt1Ús© r   úBlib/python3.9/site-packages/astropy/io/fits/tests/test_division.pyÚtest_rec_from_string   s    ,
üz*TestDivisionFunctions.test_rec_from_stringc                 C   s   t  ¡ }d|jd< d S )NZŒabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgÚabc)r   Z
PrimaryHDUÚheader)r   Úhr   r   r   Útest_card_with_continue   s    z-TestDivisionFunctions.test_card_with_continuec                 C   sN   t  |  d¡¡*}t|d jƒtdƒu s,J ‚W d   ƒ n1 s@0    Y  d S )Nr   r   )r   r
   r   ÚtypeÚsize)r   r   r   r   r   Útest_valid_hdu_size   s    z)TestDivisionFunctions.test_valid_hdu_sizec                 C   s4   t  |  d¡¡}W d   ƒ n1 s&0    Y  d S )Nr   )r   r
   r   )r   Ú_r   r   r   Útest_hdu_get_size   s    z'TestDivisionFunctions.test_hdu_get_sizec                 C   sX   t  |  d¡¡4}t |d jd t dg¡k¡s6J ‚W d   ƒ n1 sJ0    Y  d S )Nzarange.fitsr   )é   é   é   ie  )r   r
   r   r   ÚallZsectionr   )r   ZcapsysZfsr   r   r   Útest_section!   s    z"TestDivisionFunctions.test_sectionN)	Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r#   r   r   r   r   r   	   s   	r   )Znumpyr   Z
astropy.ior   Ú r   r   r   r   r   r   Ú<module>   s   