B q\@s:ddlmZddlZddlmZdgZGdddeZdS)) OrderedDictN) isiterableFlagCollectionc@s eZdZdZddZddZdS)ra The purpose of this class is to provide a dictionary for containing arrays of flags for the `NDData` class. Flags should be stored in Numpy arrays that have the same dimensions as the parent data, so the `FlagCollection` class adds shape checking to an ordered dictionary class. The `FlagCollection` should be initialized like an `~collections.OrderedDict`, but with the addition of a ``shape=`` keyword argument used to pass the NDData shape. cOsFd|kr(|d|_t|js0tdntdtj|f||dS)Nshapez1FlagCollection shape should be an iterable objectz?FlagCollection should be initialized with the shape of the data)poprr ValueError Exceptionr__init__)selfargskwargsr =lib/python3.7/site-packages/astropy/nddata/flag_collection.pyr s    zFlagCollection.__init__cKsNt|tjrB|j|jkr,tj|||f|qJtd|j|jntddS)Nz3flags array shape {0} does not match data shape {1}z&flags should be given as a Numpy array) isinstancenpZndarrayrr __setitem__rformat TypeError)r itemvaluer r r rr's   zFlagCollection.__setitem__N)__name__ __module__ __qualname____doc__r rr r r rr s  ) collectionsrZnumpyrZastropy.utils.miscr__all__rr r r rs