pandera.extensions

Extensions module, for backwards compatibility.

class pandera.extensions.CheckType(value)

Bases: Enum

Check types for registered check methods.

VECTORIZED = 1

Check applied to a Series or DataFrame

ELEMENT_WISE = 2

Check applied to an element of a Series or DataFrame

GROUPBY = 3

Check applied to dictionary of Series or DataFrames.

pandera.extensions.register_builtin_check(fn=None, strategy=None, _check_cls=<class 'pandera.api.checks.Check'>, aliases=None, **outer_kwargs)

Register a check method to the Check namespace.

This is the primary way for extending the Check api to define additional built-in checks.

pandera.extensions.register_builtin_hypothesis(**kwargs)

Register a new hypothesis.

pandera.extensions.register_check_method(check_fn=None, *, statistics=None, supported_types=None, check_type='vectorized', strategy=None)

Registers a function as a Check method.

See the user guide for more details.

Parameters:

• check_fn – check function to register. The function should take one positional argument for the object to validate and additional keyword-only arguments for the check statistics.

• statistics (Optional[List[str]]) – list of keyword-only arguments in the check_fn, which serve as the statistics needed to serialize/de-serialize the check and generate data if a strategy function is provided.

• supported_types (Union[type, Tuple, List, None]) – the pandas type(s) supported by the check function. Valid values are pd.DataFrame, pd.Series, ps.DataFrame, or a list/tuple of (pa.DataFrame, pa.Series, ps.DataFrame) if both types are supported. Valid values are pd.DataFrame, pd.Series, ps.DataFrame, or a list/tuple of (pa.DataFrame, pa.Series, ps.DataFrame) if both types are supported.

• check_type (Union[CheckType, str]) –

  the expected input of the check function. Valid values are CheckType enums or {"vectorized", "element_wise", "groupby"}. The input signature of check_fn is determined by this argument:

  • if vectorized, the first positional argument of check_fn should be one of the supported_types.

  • if element_wise, the first positional argument of check_fn should be a single scalar element in the pandas Series or DataFrame.

  • if groupby, the first positional argument of check_fn should be a dictionary mapping group names to subsets of the Series or DataFrame.

• strategy – data-generation strategy associated with the check function.

Returns:

register check function wrapper.

pandera.extensions.register_check_statistics(statistics_args)

Decorator to set statistics based on Check method.