o i[@sVddlmZddlZddlZddlmZddlmZ ddddddZdddZ dS)) annotationsN)settings)ParametricProfileNamesfastF)set_environmentprofileParametricProfileNames | intrboolreturnNonecCsddd}t|}dD]\}}tjd d||d|q |s(td|r@t|dd }d|}tjd ||d|d|dd }t||rVt |dSdS) ae Load a named (or custom) hypothesis profile for use with the parametric tests. .. warning:: This functionality is currently considered **unstable**. It may be changed at any point without it being considered a breaking change. Parameters ---------- profile : {str, int}, optional Name of the profile to load; one of "fast", "balanced", "expensive", or the integer number of iterations to run (which will create and register a custom profile with that value). set_environment : bool, default False If True, also set the environment variable `POLARS_HYPOTHESIS_PROFILE` to the given profile name/value. Examples -------- >>> # load a custom profile that will run with 1500 iterations >>> from polars.testing.parametric import load_profile >>> load_profile(1500) TN)Z print_blobdeadline))rd)Zbalancedi)Z expensivei'zpolars.)nameZ max_exampleszpolars\.custom\.[\d_]+$zpolars.custom.) strrZregister_profileisdigitrematchintreplace load_profile set_profile)rrZcommon_settings profile_namerZ iterationsZ n_iterationsrrY/home/app/Keep/.python/lib/python3.10/site-packages/polars/testing/parametric/profiles.pyr s.      rcCsrt|dd}|ddrtt|}nddlm}|t}||vr2d|d|}t||t j d <d S) a Set the env var `POLARS_HYPOTHESIS_PROFILE` to the given profile name/value. .. warning:: This functionality is currently considered **unstable**. It may be changed at any point without it being considered a breaking change. Parameters ---------- profile : {str, int}, optional Name of the profile to load; one of "fast", "balanced", "expensive", or the integer number of iterations to run (which will create and register a custom profile with that value). Examples -------- >>> # prefer the 'balanced' profile for running parametric tests >>> from polars.testing.parametric import set_profile >>> set_profile("balanced") ._rr)get_argszinvalid profile name z; expected one of ZPOLARS_HYPOTHESIS_PROFILEN) rsplitrrrtypingrr ValueErrorosenviron)rrrZvalid_profile_namesmsgrrrrJs r)r)rrrr r r )rrr r ) __future__rr"rZ hypothesisrZpolars._typingrrrrrrrs   ?