U
    e»{jœ  ã                	   @   sn   d dl Zd dlZd dlmZ dd„ Zdd„ Zdd„ Zd	d
„ Zej	 
dddddddgdg¡dd„ ƒZdd„ ZdS )é    N©ÚFlagCollectionc                   C   s   t dd� d S )N©é   é   é   ©Úshaper   © r
   r
   úh/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/nddata/tests/test_flag_collection.pyÚ	test_init
   s    r   c               	   C   s4   t  t¡�} tƒ  W 5 Q R X | jjd dks0t‚d S )Nr   z?FlagCollection should be initialized with the shape of the data©ÚpytestÚraisesÚ	Exceptionr   ÚvalueÚargsÚAssertionError©Úexcr
   r
   r   Útest_init_noshape   s    
ÿÿr   c               	   C   s8   t  t¡�} tdd� W 5 Q R X | jjd dks4t‚d S )Nç      ð?r   r   z1FlagCollection shape should be an iterable objectr   r   r
   r
   r   Útest_init_notiterable   s    r   c                  C   s^   t dd�} t d¡ t¡| d< t d¡ t¡| d< t d¡ t¡| d< t d¡ t¡| d< d S )Nr   r   ÚaÚbÚcÚd)r   ÚnpÚonesZastypeÚfloatÚintÚboolÚstr)Úfr
   r
   r   Útest_setitem   s
    
r$   r   r   r   Zspamr   r   )r   g       @g      @c              	   C   s@   t dd�}t t¡�}| |d< W 5 Q R X |jjd dks<t‚d S )Nr   r   r   r   z&flags should be given as a Numpy array)r   r   r   r   r   r   r   )r   r#   r   r
   r
   r   Útest_setitem_invalid_type%   s    
r%   c               	   C   s^   t dd�} t t¡�}t d¡| d< W 5 Q R X |jjd  d¡sDt	‚|jjd  
d¡sZt	‚d S )Nr   r   )r   r   r   r   r   zflags array shapez#does not match data shape (1, 2, 3))r   r   r   Ú
ValueErrorr   r   r   r   Ú
startswithr   Úendswith)r#   r   r
   r
   r   Útest_setitem_invalid_shape-   s
    
r)   )Únumpyr   r   Zastropy.nddatar   r   r   r   r$   ÚmarkZparametrizer%   r)   r
   r
   r
   r   Ú<module>   s   	
