q-PhXddlmZddlmZddlmZmZddlmZ ddl m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZer*ddlZdd lmZdd lmZmZejd krddlm Z nddlm Z e d d)ddeddd*dZ e d d)ddddddd+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|(tj||}t|trdt |D}n|}t |}|dg|z}not|r|g|z}nYt|tr0t |x}|krd|d|d} t| |}n|d} t| fdt||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) min_value max_valuecg|]}d|S)col).0ns k/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/testing/parametric/strategies/legacy.py zcolumns..Zs???a a ???zgiven z dtypes for z namesz is not a valid polars datatypec:g|]\}}t||S))namerr)r )r%nmtprs r'r(zcolumns..ms, V V VRF"V 4 4 4 V V Vr)) stintegersexample isinstancerrangelenr rrzip) rrrrrnamesn_colsrn_dtypesmsgs ` r'columnsr9s0v |{XBBBJJLL$??5;;??? ZZF }37&6/    #6! E8 $ $#E "Hv - -?8?????C!#&& &999c""" W V V V3ufCUCU V V VVr)z:`create_list_strategy` is deprecated; use `lists` instead.) select_fromsizemin_sizemax_sizer inner_dtypePolarsDataType | Noner:Sequence[Any] | Noner; int | Noner<r=SearchStrategy[list[Any]]c||x}}|!t}n!t|}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] N)r:r<r=r)rr0r r )r>r:r;r<r=rs r'create_list_strategyrDpsqV ""8hh&&(( (55==??     r))N) rrrrrrrrrrrr)r>r?r:r@r;rAr<rr=rArrrrB)" __future__rcollections.abcrtypingrrhypothesis.strategies strategiesr.hypothesis.errorsrpolars._utils.deprecationrpolars.datatypesr )polars.testing.parametric.strategies.corer r )polars.testing.parametric.strategies.datar *polars.testing.parametric.strategies.dtyper rsysrpolars._typingrr version_infowarningstyping_extensionsr9rDr$r)r'rUs""""""$$$$$$%%%%%%%%""""""------000000,,,,,,HHHHHHHH;;;;;;QQQQQQQQ 1JJJ444444AAAAAAAA 7""'''''''000000 0 (,MW(, MWMWMWMWMW MW`  HII)-8)-88888JI888r)