
    IR-e                         d dl Zd dlZd dlmZ d Zd Zd Zd Zej	        
                    ddd	d
g ddg          d             Zd ZdS )    NFlagCollectionc                  &    t          d           d S )N         shaper        Ilib/python3.11/site-packages/astropy/nddata/tests/test_flag_collection.py	test_initr   
   s    ######r   c                      t          j        t                    5 } t                       d d d            n# 1 swxY w Y   | j        j        d         dk    sJ d S )Nr   z?FlagCollection should be initialized with the shape of the datapytestraises	Exceptionr   valueargsexcs    r   test_init_noshaper      s    	y	!	! S               		qL	M 	M 	M 	M 	M 	Ms   599c                      t          j        t                    5 } t          d           d d d            n# 1 swxY w Y   | j        j        d         dk    sJ d S )N      ?r
   r   z1FlagCollection shape should be an iterable objectr   r   s    r   test_init_notiterabler      s    	y	!	! "SS!!!!" " " " " " " " " " " " " " "9>! SSSSSSSs   7;;c                     t          d          } t          j        d                              t                    | d<   t          j        d                              t
                    | d<   t          j        d                              t                    | d<   t          j        d                              t                    | d<   d S )Nr   r
   abcd)r   nponesastypefloatintboolstr)fs    r   test_setitemr*      s    Y'''AWY&&u--AcFWY&&s++AcFWY&&t,,AcFWY&&s++AcFFFr   r   r   r   spamr   )r   g       @g      @c                     t          d          }t          j        t                    5 }| |d<   d d d            n# 1 swxY w Y   |j        j        d         dk    sJ d S )Nr   r
   r   r   z&flags should be given as a Numpy array)r   r   r   r   r   r   )r   r)   r   s      r   test_setitem_invalid_typer-   %   s    Y'''A	y	!	! S#              9>! HHHHHHHs   <A A c                  R   t          d          } t          j        t                    5 }t	          j        d          | d<   d d d            n# 1 swxY w Y   |j        j        d                             d          sJ |j        j        d         	                    d          sJ 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   s     r   test_setitem_invalid_shaper2   -   s    Y'''A	z	"	" $c###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $9>!''(;<<<<<9>!%%&KLLLLLLLs   AAA)numpyr"   r   astropy.nddatar   r   r   r   r*   markparametrizer-   r2   r   r   r   <module>r7      s         ) ) ) ) ) )$ $ $  T T T, , , 1c6999o"NOOI I POIM M M M Mr   