a
    ߙfb                     @   sl   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g ddgdd Zdd ZdS )    NFlagCollectionc                   C   s   t dd d S )N         shaper    r
   r
   Hlib/python3.9/site-packages/astropy/nddata/tests/test_flag_collection.py	test_init
   s    r   c                  C   sH   t t} t  W d    n1 s&0    Y  | jjd dksDJ d S )Nr   z?FlagCollection should be initialized with the shape of the datapytestraises	Exceptionr   valueargsexcr
   r
   r   test_init_noshape   s    $r   c                  C   sL   t t} tdd W d    n1 s*0    Y  | jjd dksHJ 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   g       @g      @c                 C   sT   t dd}tt}| |d< W d    n1 s20    Y  |jjd dksPJ 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   test_setitem_invalid_type$   s    
&r$   c                  C   sr   t dd} tt}td| d< W d    n1 s80    Y  |jjd dsXJ |jjd 	dsnJ 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   
startswithendswith)r"   r   r
   r
   r   test_setitem_invalid_shape,   s
    
,r(   )r   Znumpyr   Zastropy.nddatar   r   r   r   r#   ZmarkZparametrizer$   r(   r
   r
   r
   r   <module>   s   
