o i@sddlmZddlmZddlmZmZddlmZ ddl m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZerhddlZdd lmZdd lmZmZejd krbddlm Z nddlm Z e d d+ddeddd,ddZ e d d+ddddddd-d)d*Z!dS).) 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_colsuniquecolsint | Sequence[str] | NonerOneOrMoreDataTypes | Nonerintrrboolreturn list[column]c s|dur tj||d}t|trddt|D}n|}t|}|dur,dg|}n0t|r6|g|}n&t|trSt|}|krPd|d|d} t | |}n |d} t | fd dt ||d d DS) 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)Z min_valueZ max_valuecSsg|]}d|qS)col).0nrrb/home/app/Keep/.python/lib/python3.10/site-packages/polars/testing/parametric/strategies/legacy.py Zszcolumns..zgiven z dtypes for z namesz is not a valid polars datatypecsg|] \}}t||dqS))namerr)r )r nmtprrr"r#ms T)strict) stZintegersexample isinstancerrangelenrrrzip) rrrrrnamesZn_colsr Zn_dtypesmsgrr'r"columnss(;       r1z:`create_list_strategy` is deprecated; use `lists` instead.) select_fromsizemin_sizemax_sizer inner_dtypePolarsDataType | Noner2Sequence[Any] | Noner3 int | Noner4r5SearchStrategy[list[Any]]cCsB|dur|}}|durt}nt|}t|||||dS)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] N)r2r4r5r)r r*r r )r6r2r3r4r5rrrr"create_list_strategyss+  r;)N) rrrrrrrrrrrr)r6r7r2r8r3r9r4rr5r9rrrr:)" __future__rcollections.abcrtypingrrZhypothesis.strategies strategiesr)Zhypothesis.errorsrZpolars._utils.deprecationrZpolars.datatypesrZ)polars.testing.parametric.strategies.corer r Z)polars.testing.parametric.strategies.datar Z*polars.testing.parametric.strategies.dtyper r sysrZpolars._typingrr version_infowarningsZtyping_extensionsr1r;rrrr"sH         S