wdXddlZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZddlmZmZgdZe e e efZdZd Zd Zd Zd Z d Z!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dS)N)collectdshape)remove) DataShapeFixedVarEllipsisRecordTupleUnitdate_ datetime_TypeVarto_numpy_dtypeMapOption Categorical)floatingboolean) isdimension ishomogeneous istabularisfixedisscalarisrecord iscollection isnumeric isboolean isdatelikeisrealct|trt|}t|trt |dkr|d}tt |d|t tfS)z Is this dshape a single dtype? >>> isscalar('int') True >>> isscalar('?int') True >>> isscalar('{name: string, amount: int}') False rrty) isinstancestrrrlengetattrr rdss 4lib/python3.11/site-packages/datashape/predicates.pyrrsn"c BZZ"i  SWW\ U gb$++dK-@ A AAct|trt|}t|trt |dkr|d}tt |d|t S)z Is this dshape a record type? >>> isrecord('{name: string, amount: int}') True >>> isrecord('int') False >>> isrecord('?{name: string, amount: int}') True rrr#)r$r%rrr&r'r r(s r*rr$si"c BZZ"i  SWW\ U gb$++V 4 44r+c,t|tS)z Is a component a dimension? >>> from datashape import int32 >>> isdimension(Fixed(10)) True >>> isdimension(Var()) True >>> isdimension(int32) False )r$dimension_typesr(s r*rr5s b/ * **r+c t|}ttttt t |dkS)z Does datashape contain only one dtype? >>> from datashape import int32 >>> ishomogeneous(int32) True >>> ishomogeneous('var * 3 * string') True >>> ishomogeneous('var * {name: string, amount: int}') False r)rr&setrrrrr(s r*rrCs> B s6+wx'<'<==>> ? ?1 DDr+cDtt|jS)z&Number of dimensions of datashape )r&rshaper(s r* _dimensionsr3Rs vbzz  r+ct|}t|dtrdSt|dtrdSt|dtr-t t t|djSt|dkr"t| dSdS)z Contains no variable dimensions >>> isfixed('10 * int') True >>> isfixed('var * int') False >>> isfixed('10 * {name: string, amount: int}') True >>> isfixed('10 * {name: string, amounts: var * int}') False rNFrT) rr$rr r allmaprtypesr&subarrayr(s r*rrXs B"Q%!!t"Q%u"Q%  .3w1 ,,--- 2ww{'r{{1~~&&& 4r+cnt|}t|dkot|jS)z A collection of records >>> istabular('var * {name: string, amount: int}') True >>> istabular('var * 10 * 3 * int') False >>> istabular('10 * var * int') False >>> istabular('var * (int64, string, ?float64)') False r)rr3rmeasurer(s r*rrps/ B r??a  8HRZ$8$88r+ctt|trt|}t|dS)z Is a collection of items, has dimension >>> iscollection('5 * int32') True >>> iscollection('int32') False r)r$r%rrr(s r*rrs3"c BZZ r!u  r+ct|} t|}t|tot j|tjS#t$rYdSwxYw)z Has a numeric measure >>> isnumeric('int32') True >>> isnumeric('3 * ?real') True >>> isnumeric('string') False >>> isnumeric('var * {amount: ?int32}') False F)launderrr$r np issubdtypenumber TypeError)r)npdtypes r*rrsk BJ $$"d##I gry(I(II uusA A"!A"ct|trt|}t|tr|j}t |d|S)Nr#)r$r%rrr:r'r(s r*r=r=sJ"c BZZ"i   Z 2tR  r+c\t|}t|to|tvS)z Has a numeric measure >>> isreal('float32') True >>> isreal('3 * ?real') True >>> isreal('string') False )r=r$r rr(s r*r!r!s( B b$   2B(N2r+c.t|tvS)z Has a boolean measure >>> isboolean('bool') True >>> isboolean('3 * ?bool') True >>> isboolean('int') False )r=rr(s r*rrs 2;;' !!r+cLt|}|tkp |tkS)z Has a date or datetime measure >>> isdatelike('int32') False >>> isdatelike('3 * datetime') True >>> isdatelike('?datetime') True )r=rrr(s r*r r s" B ; )" /)r+))numpyr>utilrrinternal_utilsr coretypesrrr r r r r rrrrrrrtypesetsrr__all__intr.rrrrr3rrrrr=r!rr r+r*rOs!!!!!!!!""""""--------------------------------('''''''   h+BBB"555" + + + E E E!!! 0 9 9 9   JJJ,!!! 3 3 3 " " " * * * * *r+