q-Ph3:UdZddlmZddlZddlmZddlmZmZmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZm 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.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?dd l@mAZAdd lBmCZCmDZDe r&dd lmEZEddlmFZFmGZGddlmHZHddlImJZJmKZKmLZLddl$mMZMmNZNdZOdZPejQeeejQeeejQeeejQeeejQeedejQde ejQde!ejQde"ejQde#ddZRdeSd< dedddfd"ZQ dedd#dgd'ZTdhd)ZUdid+ZVdjd-ZWePfdkd0ZXdld2ZYdmd4ZZ dndod;Z[dpdqd=Z\ drdsdCZ]ddddDdEdtdPZ^ddQdudVZ_dvdXZ`dwdZZaie'eUe1eQd[de2eQd\de3eQd]de4eQdde5eQd^de<eQd[dDe=eQd\dDe>eQd]dDe?eQddDe;eYe)eZe9eVe&eWe7e`e8eaZbd_eSd`<dDdQdxddZcdS)yz0Strategies 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_MAX)ArrayBinaryBoolean CategoricalDateDatetimeDecimalDurationEnumFieldFloat32Float64Int8Int16Int32Int64Int128ListNullObjectStringStructTimeUInt8UInt16UInt32UInt64)flexhash)_DEFAULT_ARRAY_WIDTH_LIMIT_DEFAULT_ENUM_CATEGORIES_LIMIT)Sequence)datetime)SearchStrategy)PolarsDataType SchemaDictTimeUnit)DataType DataTypeClass ) @)rErFrGrH)TFz*dict[bool, dict[int, SearchStrategy[int]]]_INTEGER_STRATEGIESrHT)signed bit_widthLiteral[8, 16, 32, 64, 128]rKboolreturnSearchStrategy[int]c(t||S)*Create a strategy for generating integers.)rJ)rLrKs i/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/testing/parametric/strategies/data.pyintegersrT_s v &y 11allow_infinityLiteral[32, 64]rWSearchStrategy[float]c.tj||S)rR)widthrW)stfloats)rLrWs rSr]r]fs 99^ D D DDrUSearchStrategy[bool]c(tjS)z*Create a strategy for generating booleans.)r\booleansrUrSr`r`ms ;==rUSearchStrategy[str]c^tjdddg}tj|dS)z/Create a strategy for generating string values.CsCc) max_codepointexclude_categoriesrE)alphabetmax_size)r\ characterstext)ris rSstringsrmrs/}4T4LQQQH 7Hq 1 1 11rUSearchStrategy[bytes]c(tjS)z'Create a strategy for generating bytes.)r\binaryrarUrSrprpxs 9;;rU n_categoriesintc\dt|D}tj|S)z Create a strategy for generating category strings. Parameters ---------- n_categories The number of categories. cg|]}d|S)cra).0is rS zcategories..s777a'a''777rU)ranger\ sampled_from)rq categoriess rSr{r{}s/875#6#6777J ?: & &&rUSearchStrategy[time]c(tjS)z0Create a strategy for generating `time` objects.)r\timesrarUrSr~r~ 8::rUSearchStrategy[date]c(tjS)z0Create a strategy for generating `date` objects.)r\datesrarUrSrrrrUus time_unitr@ time_zone str | NoneSearchStrategy[datetime]c|dvrtj}tj}n^|dkrDtt t dzdzz}tt t dzz}nd|}t||tj ||St|}|dkr&|t dz }|t dz}tj ||tj |d  d 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. )rmsnsrd microsecondsinvalid time unit: N)daysF) min_value max_value timezonesallow_imaginarych|tjdS)N)tzinfo) astimezonerutcreplace)dts rSzdatetimes..s%R]]8<00888EErU) rminmaxr rrrr r\ datetimesr justmap)rrrrmsgtime_zone_infos rSrrsL  L L d  I7d?Q3FGGGG I7d?CCCC 1I11c"""|Iy111i((NDYA&&&& YA&&&&  <'.))      c E EFF GrUSearchStrategy[timedelta]c|dkr'@'@AAO _--I i((I ;    rUF select_frommin_sizerjunique inner_dtyperArSequence[Any] | Nonerrjrkwargsr SearchStrategy[list[Any]]c ||dkrtn|dz}|)|stj|}nt |f||||d|}tj||||rt ndS)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)elementsrrj unique_by)_DEFAULT_LIST_LEN_LIMIT is_nestedr\rzdatalistsr7)rrrrjrrinner_strategys rSrrsN.6!mm**A{'<'<'>'>55  #       8%/884    rU allow_nullfieldsSequence[Field] | SchemaDictrSearchStrategy[dict[str, Any]]c t|trd|D}fd|D}rtji|Stj|S)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. c4g|]\}}t||Sra)r%)rvnamedtypes rSrxzstructs..Ps&GGGu%e$$GGGrUcDi|]}|jt|jfdiSr)rrr)rvfrrs rS zstructs..Rs4 U U Uafd17DDzDVDD U U UrU)optional) isinstanceritemsr\fixed_dictionaries)rrrstratss `` rSstructsr;s(&'""HGG GGG U U U U Uf U U UF-$R&9999$V,,,rUSearchStrategy[None]c(tjS)z-Create a strategy for generating null values.)r\nonerarUrSnullsrZs 799rUSearchStrategy[object]c4tjtS)z3Create a strategy for generating arbitrary objects.)r\buildsobjectrarUrSobjectsr_s 9V  rUrErFrGrIz(dict[DataTypeClass, SearchStrategy[Any]]_STATIC_STRATEGIESrr>SearchStrategy[Any]c \t|x}|}n|tkr't d|dd}n|t kr't d|dd}n|tkr3tt|ddpdt|d d }nD|tkr#tt|ddpd }n|tkr+t|d t }n|tkrt!|trS|j x}rt%}nt'j|}nmt|d t, }nB|t.kr0t1t|ddt|dd}n|t2kr/t|ddp t5}t7|fd|i|}n|t8kret|ddp t5}t|dt:}d|D}t7|f|||d|}n]|t>kr>t|ddptAdt5g}tC|fd|i|}nd|}tE||rt%|z}|S)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`. NrGrWTrVrHrrr)rr)rrq)rqrrrinnerrsizec"i|] \}}|dv || S))rrjra)rvkvs rSrzdata..s)WWW41aQ>V5V5V!Q5V5V5VrU)rrjrrf0zunsupported data type: )#rget base_typer&r]popr'r!rgetattrr#rrr{_DEFAULT_N_CATEGORIESr$ris_emptyrr\rzto_listr9r"rr-r.rrr8rr1r%rr ) rrrstrategycatsrrrrs rSrrys7 '**5??+<+<===J '  "VZZ8H$-O-OPPP '  "VZZ8H$-O-OPPP (  e[$77?4e[$77    (  wuk4'H'H'PDQQQ +  N4IJJ    $ eT " " ((2244 ; 77?4<<>>::!#ZZ8VWWHH '   E; - -wugq/I/I   $w--7@@:@@@ %w--7uf&@AAWW6<<>>WWW  !       &$//HE$4G4G3H6CCjCFCC///c"""&77X% OrU)rH)rLrMrKrNrOrP)rLrXrWrNrOrY)rOr^)rOrb)rOrn)rqrrrOrb)rOr|)rOr)rN)rr@rrrOr)r)rr@rOr)rr)rrrrrrOr)rrArrrrrrjrrrNrr rOr)rrrrNrr rOr)rOr)rOr)rr>rrNrr rOr)d__doc__ __future__rrcollections.abcrrrrtypingrr r zoneinfor hypothesis.strategies strategiesr\hypothesis.errorsr polars._utils.constantsr rrrrrrrrrrrrrrpolars.datatypesrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6+polars.testing.parametric.strategies._utilsr7*polars.testing.parametric.strategies.dtyper8r9r:r;r<r=polars._typingr>r?r@rArBrrrTrJ__annotations__r]r`rmrpr{r~rrrrrrrrrrrarUrSrs666""""""######2222222222..........""""""------":A@@@@@ 9((((((########444444CCCCCCCCCC88888888 2;vv & & BK ) ) BK ) ) BK ) ) R[8 , ,    2;q& ! ! BK7 # # BK7 # # BK7 # #   CC$.02CG222222"$E?CEEEEEE 2222  $9 ' ' ' ' '  9='G'G'G'G'GT#####B./D)- ;;;;;;B------>  @ XXZZ@((1T " " "@ 88Bt $ $ $@ 88Bt $ $ $ @  88Bt $ $ $ @  HHS & & & @ 88Ae $ $ $@ HHR & & &@ HHR & & &@ HHR & & &@ %%''@ %%''@ GGII@ FFHH@ %%''@  GGII!@*27GGGGGGGGrU