ivddlmZddlmZddlmZmZddlmZ ddl m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZer.ddlZdd lmZdd lmZmZej:d k\rddlm Z nddlm Z e d dddedd ddZ e d ddddddd ddZ!y)) annotations)Sequence) TYPE_CHECKINGAnyN)InvalidArgument) deprecated)is_polars_dtype) _COL_LIMITcolumn)lists)_instantiate_dtypedtypes)SearchStrategy)OneOrMoreDataTypesPolarsDataType) zX`columns` is deprecated; use `column` instead, in conjunction with a list comprehension.F)dtypemin_colsmax_colsuniquec |%tj||j}t|trt |Dcgc]}d| }}n|}t |}|dg|z}nYt|r|g|z}nGt|tr't |x} |k7rd| d|d} t| |}n|d} t| t||d D cgc]\} } t| | | c} } Scc}wcc} } w) aG Define multiple columns for use with the @dataframes strategy. .. deprecated:: 0.20.26 Use :class:`column` instead, in conjunction with a list comprehension. .. warning:: This functionality is currently considered **unstable**. It may be changed at any point without it being considered a breaking change. Generate a fixed sequence of `column` objects suitable for passing to the @dataframes strategy, or using standalone (note that this function is not itself a strategy). Notes ----- Additional control is available by creating a sequence of columns explicitly, using the `column` class (an especially useful option is to override the default data-generating strategy for a given col/dtype). Parameters ---------- cols : {int, [str]}, optional integer number of cols to create, or explicit list of column names. if omitted a random number of columns (between mincol and max_cols) are created. dtype : PolarsDataType, optional a single dtype for all cols, or list of dtypes (the same length as `cols`). if omitted, each generated column is assigned a random dtype. min_cols : int, optional if not passing an exact size, can set a minimum here (defaults to 0). max_cols : int, optional if not passing an exact size, can set a maximum value here (defaults to MAX_COLS). unique : bool, optional indicate if the values generated for these columns should be unique (per-column). Examples -------- >>> from polars.testing.parametric import columns, dataframes >>> from hypothesis import given >>> @given(dataframes(columns(["x", "y", "z"], unique=True))) # doctest: +SKIP ... def test_unique_xyz(df: pl.DataFrame) -> None: ... assert_something(df) N) min_value max_valuecolzgiven z dtypes for z namesz is not a valid polars datatypeT)strict)namerr) stintegersexample isinstanceintrangelenr rrzipr ) colsrrrrnnamesn_colsrn_dtypesmsgnmtps m/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/testing/parametric/strategies/legacy.pycolumnsr/sv |{{XBJJL$38;?a#aS ?? ZF }37&6/  6! E8 $E "Hv -8*L?C!#& & 89c"" %5  B Bb0 ' @& s  C0C5z:`create_list_strategy` is deprecated; use `lists` instead.) select_fromsizemin_sizemax_sizerc||x}}|tj}nt|j}t|||||S)a Create a strategy for generating Polars :class:`List` data. .. deprecated:: 0.20.26 Use :func:`lists` instead. Parameters ---------- inner_dtype : PolarsDataType type of the inner list elements (can also be another List). select_from : list, optional randomly select the innermost values from this list (otherwise the default strategy associated with the innermost dtype is used). size : int, optional if set, generated lists will be of exactly this size (and ignore the min_size/max_size params). min_size : int, optional set the minimum size of the generated lists (default: 0 if unset). max_size : int, optional set the maximum size of the generated lists (default: 3 if min_size is unset or zero, otherwise 2x min_size). unique : bool, optional ensure that the generated lists contain unique values. Examples -------- Create a strategy that generates a list of i32 values: >>> from polars.testing.parametric import create_list_strategy >>> lst = create_list_strategy(inner_dtype=pl.Int32) # doctest: +SKIP >>> lst.example() # doctest: +SKIP [-11330, 24030, 116] )r0r2r3r)rr r r ) inner_dtyper0r1r2r3rs r.create_list_strategyr6ssZV ""8h&&( (5==?   )N) r&zint | Sequence[str] | NonerzOneOrMoreDataTypes | Nonerr"rr"rboolreturnz list[column])r5zPolarsDataType | Noner0zSequence[Any] | Noner1 int | Noner2r"r3r:rr8r9zSearchStrategy[list[Any]])" __future__rcollections.abcrtypingrrhypothesis.strategies strategiesrhypothesis.errorsrpolars._utils.deprecationrpolars.datatypesr )polars.testing.parametric.strategies.corer r )polars.testing.parametric.strategies.datar *polars.testing.parametric.strategies.dtyper rsysrpolars._typingrr version_infowarningstyping_extensionsr/r6r7r.rLs5"$%"-0,H;Q4A 7"'0 0 (,P(, P $P %P P  P  PP Pf HI)-8)-8&8&8  8  8  8 88J8r7