i5 UdZddlmZddlZddlmZddlmZmZmZddl m Z m Z m Z m Z ddlmZddlmZddlmZdd lmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDdd lEmFZFdd lGmHZHmIZIe r(dd lmJZJddlmKZKmLZLddlmMZMddlNmOZOmPZPmQZQddl&mRZRmSZSmTZTdZUdZVejeeejeeejeeejeeeje edejde!ejde"ejde#ejde$ejde%ddZXdeYd< d7dd d8dZW d7ddd d9dZZd:dZ[d;dZ\dd"Z_d?d#Z` d@ dAd$ZadBdCd%Zb dD dEd&Zcdddd'd( dFd)Zddd* dGd+ZedHd,ZfdId-Zgie*e[e5eWd.de6eWd/de7eWd0de8eWdde9eWd1de@eWd.d'eAeWd/d'eBeWd0d'eCeWdd'eDeWd1d'e?e_e,e`e=e\e)e]e;efe<egZhd2eYd3<e2d/e3d0e4diZid4eYd5<d'd* dJd6Zjy)Kz0Strategies for generating various forms of data.) annotationsN)Mapping)datetime timedeltatimezone) TYPE_CHECKINGAnyLiteralcast)ZoneInfo)InvalidArgument)EPOCHI8_MAXI8_MINI16_MAXI16_MINI32_MAXI32_MINI64_MAXI64_MINI128_MAXI128_MINU8_MAXU16_MAXU32_MAXU64_MAXU128_MAX)Array BaseExtensionBinaryBoolean CategoricalDateDatetimeDecimalDurationEnumFieldFloat16Float32Float64Int8Int16Int32Int64Int128ListNullObjectStringStructTimeUInt8UInt16UInt32UInt64UInt128)flexhash)_DEFAULT_ARRAY_WIDTH_LIMIT_DEFAULT_ENUM_CATEGORIES_LIMIT)Sequence)datetime)SearchStrategy)PolarsDataType SchemaDictTimeUnit)DataType DataTypeClass FloatType ) @)TFz*dict[bool, dict[int, SearchStrategy[int]]]_INTEGER_STRATEGIESrNT)signedct||S)*Create a strategy for generating integers.)rP) bit_widthrQs k/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/testing/parametric/strategies/data.pyintegersrVds v &y 11) allow_nanallow_infinityc2tj|||S)rS)widthrXrY)stfloatsrTrXrYs rUr]r]ks 999^ rWc*tjS)z*Create a strategy for generating booleans.)r\booleansrWrUr`r`ws ;;=rWcbtjdddg}tj|dS)z/Create a strategy for generating string values.CsCc) max_codepointexclude_categoriesrK)alphabetmax_size)r\ characterstext)rhs rUstringsrl|s'}}4T4LQH 77Hq 11rWc*tjS)z'Create a strategy for generating bytes.)r\binaryrarWrUrnrns 99;rWclt|Dcgc]}d| }}tj|Scc}w)z Create a strategy for generating category strings. Parameters ---------- n_categories The number of categories. c)ranger\ sampled_from) n_categoriesi categoriess rUrurus6$)#67aAaS'7J7 ??: &&8s 1c*tjS)z0Create a strategy for generating `time` objects.)r\timesrarWrUrwrw 88:rWc*tjS)z0Create a strategy for generating `date` objects.)r\datesrarWrUrzrzrxrWc|dvr!tj}tj}nM|dk(r8tt t dzdzz}tt t dzz}nd|}t||tj||St|}|dk(r|t dz }|t dz}tj||tj|d jd S) a  Create a strategy for generating `datetime` objects in the time unit's range. Parameters ---------- time_unit Time unit for which the datetime objects are valid. time_zone Time zone for which the datetime objects are valid. )usmsnsrc microsecondsinvalid time unit: )daysF) min_value max_value timezonesallow_imaginaryc`|jtjjdS)N)tzinfo) astimezonerutcreplace)dts rUzdatetimes..s!R]]8<<0888ErW) rminmaxrrrrr r\ datetimesr justmap) time_unit time_zonerrmsgtime_zone_infos rUrrsL LL LL d I7d?Q3FGG I7d?CC #I=1c""||Iy11i(NDYA&& YA&&  <<''.)    c EF GrWc~|dk(r3tjttttS|dk(r9tjttdzttdzS|dk(r3tjttttSd|}t |)z Create a strategy for generating `timedelta` objects in the time unit's range. Parameters ---------- time_unit Time unit for which the timedelta objects are valid. r|r)rrr~rcr}r)r\ timedeltasrrrr )rrs rU durationsrsD}}W5W5   d }}W_=W_=   d }}W5W5  $I=1c""rWc|d}tj|}|jd||z }|j|}|j |}t j ||dd|S)aH Create a strategy for generating `Decimal` objects. Parameters ---------- precision Maximum number of digits in each number. If set to `None`, the precision is set to 38 (the maximum supported by Polars). scale Number of digits to the right of the decimal point in each number. &)precz1E+F)rrrXrYplaces)decimalContextcreate_decimal next_minus copy_negater\decimals) precisionscalerpexclusive_limitrrs rUrrsv Y'A&&Y->,?'@AO _-I i(I ;;  rWF select_frommin_sizeriuniquec ||dk(rtn|dz}|&|jstj|}nt |f||||d|}tj ||||r t SdS)aD Create a strategy for generating lists of the given data type. .. warning:: This functionality is currently considered **unstable**. It may be changed at any point without it being considered a breaking change. Parameters ---------- inner_dtype Data type of the list elements. If the data type is not fully instantiated, defaults will be used, e.g. `Datetime` will become `Datetime('us')`. select_from The values to use for the innermost lists. If set to `None` (default), the default strategy associated with the innermost data type is used. min_size The minimum length of the generated lists. max_size The maximum length of the generated lists. If set to `None` (default), the maximum is set based on `min_size`: `3` if `min_size` is zero, otherwise `2 * min_size`. unique Ensure that the generated lists contain unique values. **kwargs Additional arguments that are passed to nested data generation strategies. Examples -------- ... Nrr)elementsrri unique_by)_DEFAULT_LIST_LEN_LIMIT is_nestedr\rrdatalistsr<) inner_dtyperrrirkwargsinner_strategys rUrrsN.6!m*A{'<'<'>5  #     88%8  ,0  rW) allow_nullc Lt|tr,|jDcgc]\}}t||}}}|Dcic]%}|jt |j fd|i|'}}|rtji|Stj|Scc}}wcc}w)a Create a strategy for generating structs with the given fields. Parameters ---------- fields The fields that make up the struct. Can be either a sequence of Field objects or a mapping of column names to data types. allow_null Allow nulls as possible values. If set to True, the returned dictionaries may miss certain fields and are in random order. **kwargs Additional arguments that are passed to nested data generation strategies. r)optional) isinstanceritemsr(namerdtyper\fixed_dictionaries)fieldsrrrrfstratss rUstructsrEs(&'"8> Gu%e$GGNT Uaffd177DzDVDD UF U$$R&99$$V,,H Us B*B!c*tjS)z-Create a strategy for generating null values.)r\nonerarWrUnullsrds 779rWc4tjtS)z3Create a strategy for generating arbitrary objects.)r\buildsobjectrarWrUobjectsris 99V rWrKrLrMrOz(dict[DataTypeClass, SearchStrategy[Any]]_STATIC_STRATEGIESzMapping[type[FloatType], int]_DTYPE_BIT_WIDTHSc tj|jx}|}n|jrXt d|}t t |}tt d||jdd|jdd}nN|tk(r)tt|ddxsd t|d d }n|tk(rtt|ddxsd }n|tk(r"t|jd t }n|t"k(rt%|t"rM|jx}j'r t)}nt+j,|j/}net|jd t0}nC|t2k(r$t5t|ddt|dd}n|t6k(r(t|ddxs t9}t;|fd|i|}n|t<k(rct|ddxs t9}t|dt>}|jAD cic] \}} |dvs || }}} t;|f|||d|}ny|tBk(r3t|ddxstEdt9g} tG| fd|i|}n=t%|tHrtK|jMfd|i|}nd|} tO| |r t)|z}|Scc} }w)a Create a strategy for generating data for the given data type. Parameters ---------- dtype A Polars data type. If the data type is not fully instantiated, defaults will be used, e.g. `Datetime` will become `Datetime('us')`. allow_null Allow nulls as possible values. **kwargs Additional parameters for the strategy associated with the given `dtype`. NrHLiteral[16, 32, 64]rXTrYr^rr|r)rr)rrs)rsrrrinnerrsize)rri)rrirrf0zunsupported data type: )(rget base_typeis_floatr rtyper]popr$rgetattrr&rr"ru_DEFAULT_N_CATEGORIESr'ris_emptyrr\rrto_listr>r%rr1r2rrr=rr5r(rrr ext_storager ) rrrstrategyrTcatsrrkvrrs rUrrs '**5??+<==J  [%(%d5k2 0)<jjd3!::&6=  ( e[$7?4e[$7  ( wuk4'H'PDQ + N4IJ  $ eT "(((224 7??4<<>:!#ZZ8VWH '  E; -wugq/I  $w-7@:@@ %w-7uf&@A#)<<>W41aQ>V5V!Q$WW  !     &$/HE$4G3H6CjCFC E= )))+M MfM'w/c""7X% O)Xs  K1K1)rN)rTzLiteral[8, 16, 32, 64, 128]rQboolreturnzSearchStrategy[int])rTrrXrrYrrzSearchStrategy[float])rzSearchStrategy[bool])rSearchStrategy[str])rzSearchStrategy[bytes])rsintrr)rzSearchStrategy[time])rzSearchStrategy[date])r|N)rrErz str | NonerzSearchStrategy[datetime])r|)rrErzSearchStrategy[timedelta])rr)r int | NonerrrzSearchStrategy[decimal.Decimal])rrFrzSequence[Any] | Nonerrrirrrrr rzSearchStrategy[list[Any]])rzSequence[Field] | SchemaDictrrrr rzSearchStrategy[dict[str, Any]])rzSearchStrategy[None])rzSearchStrategy[object])rrCrrrr rzSearchStrategy[Any])k__doc__ __future__rrcollections.abcrrrrtypingrr r r zoneinfor hypothesis.strategies strategiesr\hypothesis.errorsr polars._utils.constantsrrrrrrrrrrrrrrrrpolars.datatypesrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;+polars.testing.parametric.strategies._utilsr<*polars.testing.parametric.strategies.dtyper=r>r?r@rArBpolars._typingrCrDrErFrGrHrrrVrP__annotations__r]r`rlrnrurwrzrrrrrrrrrrrarWrUrsR6"#2244"-$@A (#4CCCC 2;;vv & BKK ) BKK ) BKK ) R[[8 ,   2;;q& ! BKK7 # BKK7 # BKK7 # R[[H %  C?&.02CG2*2<@22&(  "      2  $9 '  9='G'G+5'G'GT#B./'*$D)- ;;&; ;  ;  ;;;B- (-- - $ ->  @ XZ@(1T "@ 8Bt $@ 8Bt $ @  8Bt $ @  HS & @ 8Ae $@ HR &@ HR &@ HR &@ Xc% (@ %'@ %'@ GI@ FH@  %'!@" GI#@<* R R R4027M M*.MBEMMrW