a jDf@s8UddlmZddlmZmZddlZddlmZddl m Z Gddde Z Gdd d ej e ZGd d d eZGd d d ej ZiZeeefed<ddZeejddedddddZeejddedddddZeejejdddZeejeedfdddddZGdddZdS) )Iterable)AnyCallableN)utils)ExplicitlyIndexedc@s eZdZdS)UnexpectedDataAccessN)__name__ __module__ __qualname__r r `/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/xarray/tests/arrays.pyrsrc@s<eZdZdZddZddZd ejjddd Z d d Z dS) InaccessibleArrayzDisallows any loading.cCs ||_dSNarrayselfrr r r __init__szInaccessibleArray.__init__cCs tddSNzTried accessing datarrr r r get_duck_arraysz InaccessibleArray.get_duck_arrayNdtypecCs tddSrrrrr r r __array__szInaccessibleArray.__array__cCs tddS)NTried accessing data.rrkeyr r r __getitem__szInaccessibleArray.__getitem__)N) rr r __doc__rrnptyping DTypeLikerrr r r r r s r c@seZdZddZdS)FirstElementAccessibleArraycCs$|j}t|dkrtd|j|S)Nz&Tried accessing more than one element.)tuplelenrr)rrZ tuple_idxrr r r r#s z'FirstElementAccessibleArray.__getitem__N)rr r rr r r r r$"sr$c@sDeZdZdZejdddZddZd ejj dd d Z d d Z dS)DuckArrayWrapperzBArray-like that prevents casting to array. Modeled after cupy.rcCs ||_dSrrrr r r r.szDuckArrayWrapper.__init__cCst||j|Sr)typerrr r r r1szDuckArrayWrapper.__getitem__NrcCs tddSrrrr r r r4szDuckArrayWrapper.__array__cCsdS)z&Present to satisfy is_duck_array test.Nr rr r r __array_namespace__7sz$DuckArrayWrapper.__array_namespace__)N) rr r r r!Zndarrayrrr"r#rr*r r r r r(*s r(+CONCATENATABLEARRAY_HANDLED_ARRAY_FUNCTIONScsfdd}|S)zNRegister an __array_function__ implementation for ConcatenatableArray objects.cs |t<|Sr)r+)funcnumpy_functionr r decoratorAszimplements..decoratorr )r.r/r r-r implements>s r0axisConcatenatableArray)arraysreturncCs6tdd|Drttjdd|D|d}t|S)Ncss|]}t|t VqdSr isinstancer3.0arrr r r Lzconcatenate..cSsg|] }|jqSr _arrayr8r r r Or<zconcatenate..r1)any TypeErrorr! concatenater3r4r2resultr r r rBHsrBcCs6tdd|Drttjdd|D|d}t|S)Ncss|]}t|t VqdSrr6r8r r r r;Wr<zstack..cSsg|] }|jqSr r=r8r r r r?Zr<zstack..r1)r@rAr!stackr3rCr r r rESsrEr5cGs0dd|D^}}|D]}||krtdq|S)zGCalled by xarray to ensure all arguments to concat have the same dtype.css|]}t|VqdSr)r!r)r9objr r r r;ar<zresult_type..zdtypes not all consistent) ValueError)Zarrays_and_dtypesZ first_dtypeZ other_dtypesZ other_dtyper r r result_type^s  rI.)shapexr5cCs&t|tsttj|j|d}t|S)zx Broadcasts an array to a specified shape, by either manipulating chunk keys or copying chunk manifest entries. )rJ)r7r3rAr! broadcast_tor>)rKrJrDr r r rLhs rLc@seZdZdZddZeeejdddZeee e dfddd Z eee dd d Z ee dd d ZddZd ejjdddZddddZedddZedddZddejeddddZdS)!r3zTDisallows loading or coercing to an index but does support concatenation / stacking.cCs ||_dSrr=rr r r ryszConcatenatableArray.__init__)rr5cCs|jjSr)r>rrr r r r}szConcatenatableArray.dtype.cCs|jjSr)r>rJrr r r rJszConcatenatableArray.shapecCs|jjSr)r>ndimrr r r rMszConcatenatableArray.ndimcCst|jd|jdS)Nz(array=))r)rr>rr r r __repr__szConcatenatableArray.__repr__cCs tddSrrrr r r rsz"ConcatenatableArray.get_duck_arrayNrcCs tddSrrrr r r rszConcatenatableArray.__array__rFcCsH|j}t|D]0\}}|dur,t||}q|tur6qtdqt|S)zNSome cases of concat require supporting expanding dims by dimensions of size 1Nr)r> enumerater!Z expand_dimsEllipsisrr3)rrr:r2Z indexer_1dr r r rs zConcatenatableArray.__getitem__cCs4|tvr tStdd|Ds"tSt||i|S)Ncss|]}t|tVqdSr) issubclassr3)r9tr r r r;r<z9ConcatenatableArray.__array_function__..)r+NotImplementedall)rr,typesargskwargsr r r __array_function__s z&ConcatenatableArray.__array_function__cOstS)z|We have to define this in order to convince xarray that this class is a duckarray, even though we will never support ufuncs.)rT)rZufuncmethodinputsrXr r r __array_ufunc__sz#ConcatenatableArray.__array_ufunc__T)copy)rr]r5cCs||jkrtn|SdS)z;Needed because xarray will call this even when it's a no-opN)rNotImplementedError)rrr]r r r astypes zConcatenatableArray.astype)N)rr r r rpropertyrr!rr&intrJrMstrrOrr"r#rrrYr\boolr_r r r r r3vs  )collections.abcrr"rrnumpyr!Z xarray.corerZxarray.core.indexingr ExceptionrZ NDArrayMixinr r$r(r+dictrb__annotations__r0rBrErIrrLr&rar3r r r r s: