h4 %SrSSKJr SSKrSSKJr SSKJrJrJr SSK J r J r J r SSK Jr SSKJr SSKJr SS KJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$ SS K%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;Jr>J?r?J@r@JArA SS KBJCrC SS KDJErEJFrF \ (a&SS KJGrG SSKJHrHJIrI SSKJJrJ SSKKJLrLJMrMJNrN SSK%JOrOJPrP SrQSrR\R"\\5\R"\\5\R"\\5\R"\\5\R"\\5S.\R"S\ 5\R"S\!5\R"S\"5\R"S\#5\R"S\$5S.S.rTS\US'S5SS.S6SjjjrSS5SSS.S7SjjjrVS8SjrWS9SjrXS:S jrY\R4S;S!jjrZSS?S$jjr]S@SAS%jjr^SBSCS&jjr_SSSS'S(.SDS)jjr`SS*.SES+jjraSFS,jrbSGS-jrc0\(\W"5_\2\S"S.SS9_\3\S"S/SS9_\4\S"S0SS9_\5\S"SSS9_\6\S"S1SS9_\=\S"S.S'S9_\>\S"S/S'S9_\?\S"S0S'S9_\@\S"SS'S9_\A\S"S1S'S9_\<\["5_\*\\"5_\:\X"5_\'\Y"5_\8\b"5_\9\c"5_rdS2\US3'S'S*.SHS4jjreg)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)signedc[UU$)*Create a strategy for generating integers.)rL) bit_widthrMs c/home/james-whalen/.local/lib/python3.13/site-packages/polars/testing/parametric/strategies/data.pyintegersrRbs v &y 11 allow_nanallow_infinityc,[R"XUS9$)rO)widthrUrV)stfloats)rPrUrVs rQrZrZis 99^ rSc,[R"5$)z*Create a strategy for generating booleans.)rYbooleansrSrQr\r\us ;;=rScZ[R"SSS/S9n[R"USS9$)z/Create a strategy for generating string values.CsCc) max_codepointexclude_categoriesrG)alphabetmax_size)rY characterstext)rds rQstringsrhzs'}}4T4LQH 77Hq 11rSc,[R"5$)z'Create a strategy for generating bytes.)rYbinaryr]rSrQrjrjs 99;rSct[U5Vs/sHnSU3PM nn[R"U5$s snf)zv Create a strategy for generating category strings. Parameters ---------- n_categories The number of categories. c)rangerY sampled_from) n_categoriesi categoriess rQrqrqs9$)#67#6aAaS'#6J7 ??: &&8s5c,[R"5$)z0Create a strategy for generating `time` objects.)rYtimesr]rSrQrsrs 88:rSc,[R"5$)z0Create a strategy for generating `date` objects.)rYdatesr]rSrQrvrvrtrScUS;a![Rn[RnOIUS:Xa2[[ [ S-S-S9-n[[ [ S-S9-nOSU<3n[U5eUc[R"X#5$[U5nUS:XaU[ SS9- nU[ SS9-n[R"UU[R"U5SS 9RS 5$) z 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_imaginarycZUR[R5RSS9$)N)tzinfo) astimezonerutcreplace)dts rQdatetimes..sR]]8<<0888ErS) rminmaxr rrrr rY datetimesr justmap) time_unit time_zonerrmsgtime_zone_infos rQrrsL LL LL d I7d?Q3FGG I7d?CC #I=1c""||I11i(NDYA&& YA&&  <<''.)    c EF GrScVUS:Xa+[R"[[S9[[S9S9$US:Xa1[R"[[S-S9[[S-S9S9$US:Xa+[R"[[S9[[S9S9$SU<3n[ U5e)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""rScUcSn[R"US9nURSX- 35nURU5nUR U5n[ R "UUSSUS9$)a$ 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 rQrrst Y'A&&Y->,?'@AO _-I i(I ;;  rSF select_frommin_sizereuniquec UcUS:Xa[OUS-nUb,UR5(d[R"U5nO[ U4UUUUS.UD6n[R "UUUU(a[ S9$SS9$)a 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 d[U[5(a-UR5VVs/sHup4[X45PM nnnUVs0sH%oUR[ UR 4SU0UD6_M' nnU(a[R"0US9$[R"U5$s snnfs snf)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> G%$GNT Ufffd177DzDVDDfF U$$R&99$$V,,H Us B',B-c,[R"5$)z-Create a strategy for generating null values.)rYnoner]rSrQnullsrbs 779rSc6[R"[5$)z3Create a strategy for generating arbitrary objects.)rYbuildsobjectr]rSrQobjectsrgs 99V rSrGrHrIrKz(dict[DataTypeClass, SearchStrategy[Any]]_STATIC_STRATEGIESc [RUR55=nbUnGOU[:Xa-[ SUR SS5UR SS5S9nGO{U[ :Xa-[ SUR SS5UR SS5S9nGODU[:Xa+[[USS5=(d S [US S5S 9nGOU[:Xa[[USS5=(d S S 9nGOU[:Xa[UR S [5S9nGOU[:Xa[!U[5(aSUR=nR#5(a [%5nGOq[&R("UR+55nGOK[UR S [,5S9nGO,U[.:Xa#[1[USS5[USS55nOU[2:Xa-[USS5=(d [55n[7U4SU0UD6nOU[8:Xal[USS5=(d [55n[US[:5nUR=5VVs0sHupxUS;dM Xx_M nnn[7U4UUUS.UD6nORU[>:Xa8[USS5=(d [AS[555/n [CU 4SU0UD6nOSU3n [EU 5eU(a [%5U-nU$s snnf)ap 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 " "(((2244 7??4<<>:!#ZZ8VWH '  E; -wugq/I  $w-7@:@@ %w-7uf&@A#)<<>W>41Q>V5V$!$>W  !     &$/HE$4G3H6CjCFC'w/c""7X% O%Xs ! K31K3)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]rSrQrs6"#22.."-$<A (#4CC8 ;;vv & KK ) KK ) KK ) [[8 ,   ;;q& ! KK7 # KK7 # KK7 # [[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