.h4 UdZddlmZddlZddlmZddlmZmZmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZm 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.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAdd lBmCZCdd lDmEZEmFZFe r&dd lmGZGddlmHZHmIZIddlmJZJddlKmLZLmMZMmNZNddl%mOZOmPZPdZQdZRejeeejeeejeeejeeejeedejde ejde!ejde"ejde#ejde$ddZTdeUd< d5dd d6dZS d5ddd d7dZVd8dZWd9dZXd:d ZYeRfd;d!ZZd d?d$Z]d@dAd%Z^ dB dCd&Z_dddd'd( dDd)Z`dd* dEd+ZadFd,ZbdGd-Zcie(eWe2eSd.de3eSd/de4eSd0de5eSdde6eSd1de=eSd.d'e>eSd/d'e?eSd0d'e@eSdd'eAeSd1d'e<e[e*e\e:eXe'eYe8ebe9ecZdd2eUd3<d'd* dHd4Zey)Iz0Strategies for generating various forms of data.) annotationsN)Mapping)datetime timedeltatimezone) TYPE_CHECKINGAnyLiteral)ZoneInfo)InvalidArgument)EPOCHI8_MAXI8_MINI16_MAXI16_MINI32_MAXI32_MINI64_MAXI64_MINI128_MAXI128_MINU8_MAXU16_MAXU32_MAXU64_MAXU128_MAX)ArrayBinaryBoolean CategoricalDateDatetimeDecimalDurationEnumFieldFloat32Float64Int8Int16Int32Int64Int128ListNullObjectStringStructTimeUInt8UInt16UInt32UInt64UInt128)flexhash)_DEFAULT_ARRAY_WIDTH_LIMIT_DEFAULT_ENUM_CATEGORIES_LIMIT)Sequence)datetime)SearchStrategy)PolarsDataType SchemaDictTimeUnit)DataType DataTypeClass ) @)TFz*dict[bool, dict[int, SearchStrategy[int]]]_INTEGER_STRATEGIESrJT)signedct||S)*Create a strategy for generating integers.)rL) bit_widthrMs g/var/www/html/ai-service/venv/lib/python3.12/site-packages/polars/testing/parametric/strategies/data.pyintegersrRbs v &y 11 allow_nanallow_infinityc2tj|||S)rO)widthrUrV)stfloats)rPrUrVs rQrZrZis 999^ rSc*tjS)z*Create a strategy for generating booleans.)rYbooleansrSrQr\r\us ;;=rScbtjdddg}tj|dS)z/Create a strategy for generating string values.CsCc) max_codepointexclude_categoriesrG)alphabetmax_size)rY characterstext)rds rQstringsrhzs'}}4T4LQH 77Hq 11rSc*tjS)z'Create a strategy for generating bytes.)rYbinaryr]rSrQrjrjs 99;rSclt|Dcgc]}d| }}tj|Scc}w)z Create a strategy for generating category strings. Parameters ---------- n_categories The number of categories. c)rangerY sampled_from) n_categoriesi categoriess rQrqrqs6$)#67aAaS'7J7 ??: &&8s 1c*tjS)z0Create a strategy for generating `time` objects.)rYtimesr]rSrQrsrs 88:rSc*tjS)z0Create a strategy for generating `date` objects.)rYdatesr]rSrQrvrvrtrSc|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. )usmsnsr_ microsecondsinvalid time unit: )daysF) min_value max_value timezonesallow_imaginaryc`|jtjjdS)N)tzinfo) astimezonerutcreplace)dts rQzdatetimes..s!R]]8<<0888ErS) rminmaxr rrrr rY datetimesr justmap) time_unit time_zonerrmsgtime_zone_infos rQrrsL LL LL d I7d?Q3FGG I7d?CC #I=1c""||Iy11i(NDYA&& YA&&  <<''.)    c EF GrSc~|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. rxr|)rrrzr_ryr~)rY timedeltasrrrr )rrs rQ durationsrsD}}W5W5   d }}W_=W_=   d }}W5W5  $I=1c""rSc|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)rrrUrVplaces)decimalContextcreate_decimal next_minus copy_negaterYdecimals) precisionscalerlexclusive_limitrrs rQrrsv Y'A&&Y->,?'@AO _-I i(I ;;  rSF select_frommin_sizereuniquec ||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)elementsrre unique_by)_DEFAULT_LIST_LEN_LIMIT is_nestedrYrndatalistsr9) inner_dtyperrrerkwargsinner_strategys rQrrsN.6!m*A{'<'<'>5  #     88%8  ,0  rS) 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&namerdtyperYfixed_dictionaries)fieldsrrrrfstratss rQstructsrCs(&'"8> Gu%e$GGNT Uaffd177DzDVDD UF U$$R&99$$V,,H Us B*B!c*tjS)z-Create a strategy for generating null values.)rYnoner]rSrQnullsrbs 779rSc4tjtS)z3Create a strategy for generating arbitrary objects.)rYbuildsobjectr]rSrQobjectsrgs 99V rSrGrHrIrKz(dict[DataTypeClass, SearchStrategy[Any]]_STATIC_STRATEGIESc tj|jx}|}n|tk(r0t d|j dd|j dd}nY|t k(r0t d|j dd|j dd}n |tk(r)tt|ddxsd t|d d }n|tk(rtt|ddxsd }n|tk(r"t|j d t}n|tk(rt!|trM|jx}j#r t%}n\t'j(|j+}n7t|j d t,}n|t.k(r#t1t|ddt|dd}n|t2k(r(t|ddxs t5}t7|fd|i|}n|t8k(rct|ddxs t5}t|dt:}|j=Dcic] \}}|dvs ||}}}t7|f|||d|}nL|t>k(r3t|ddxstAdt5g} tC| fd|i|}nd|} tE| |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`. NrIrUTrVrTrJrrxr)rr)rro)rorrrinnerrsize)rre)rrerrf0zunsupported data type: )#rget base_typer'rZpopr(r"rgetattrr$rr rq_DEFAULT_N_CATEGORIESr%ris_emptyrrYrnto_listr;r#rr.r/rrr:rr2r&rr ) rrrstrategycatsrrkvrrs rQrrs '**5??+<==J '  jjd3!::&6=  '  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'w/c""7X% O%Xs K K )rJ)rPzLiteral[8, 16, 32, 64, 128]rMboolreturnzSearchStrategy[int])rPzLiteral[32, 64]rUrrVrrzSearchStrategy[float])rzSearchStrategy[bool])rSearchStrategy[str])rzSearchStrategy[bytes])rointrr)rzSearchStrategy[time])rzSearchStrategy[date])rxN)rrBrz str | NonerzSearchStrategy[datetime])rx)rrBrzSearchStrategy[timedelta])rr)r int | NonerrrzSearchStrategy[decimal.Decimal])rrCrzSequence[Any] | Nonerrrerrrrr rzSearchStrategy[list[Any]])rzSequence[Field] | SchemaDictrrrr rzSearchStrategy[dict[str, Any]])rzSearchStrategy[None])rzSearchStrategy[object])rr@rrrr rzSearchStrategy[Any])f__doc__ __future__rrcollections.abcrrrrtypingrr r zoneinfor hypothesis.strategies strategiesrYhypothesis.errorsr polars._utils.constantsr rrrrrrrrrrrrrrrpolars.datatypesrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8+polars.testing.parametric.strategies._utilsr9*polars.testing.parametric.strategies.dtyper:r;r<r=r>r?polars._typingr@rArBrCrDrrrRrL__annotations__rZr\rhrjrqrsrvrrrrrrrrrr]rSrQrs-6"#22.."-$<A (#4CC8 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#@<,27O O*.OBEOOrS