a ß™fb~ã@s:ddlmZddlZddlmZdgZGdd„deƒZdS)é)Ú OrderedDictN)Ú isiterableÚFlagCollectionc@s eZdZdZdd„Zdd„ZdS)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. cOsLd|vr(| d¡|_t|jƒs0tdƒ‚ntdƒ‚tj|g|¢Ri|¤ŽdS)NÚshapez1FlagCollection shape should be an iterable objectz?FlagCollection should be initialized with the shape of the data)ÚpoprrÚ ValueErrorÚ ExceptionrÚ__init__)ÚselfÚargsÚkwargs©r ú=lib/python3.9/site-packages/astropy/nddata/flag_collection.pyr s    zFlagCollection.__init__cKsRt|tjƒrF|j|jkr0tj|||fi|¤ŽqNtd |j|j¡ƒ‚ntdƒ‚dS)Nz1flags array shape {} does not match data shape {}z&flags should be given as a Numpy array) Ú isinstanceÚnpZndarrayrrÚ __setitem__rÚformatÚ TypeError)r ÚitemÚvaluer r r rr's  ÿzFlagCollection.__setitem__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rr r r rr s  )Ú collectionsrZnumpyrZastropy.utils.miscrÚ__all__rr r r rÚs