pandera.strategies.xarray_strategies.dataset_strategy

pandera.strategies.xarray_strategies.dataset_strategy(data_vars=None, coords=None, dims=None, sizes=None, size=None)

Strategy to generate an xarray.Dataset.

Parameters:
  • data_vars (dict[str, dict[str, Any]] | None) – mapping of var name to dict with dtype, dims, nullable, etc.

  • coords (dict | None) – mapping of coord name to coord spec.

  • dims (tuple[str, ...] | None) – dataset-level dimension names.

  • sizes (dict[str, int | None] | None) – mapping of dim name to size.

  • size (int | None) – default size for dimensions.

Return type:

SearchStrategy

Returns:

hypothesis strategy producing xr.Dataset.