def _assert_indexes_invariants_checks(indexes,
possible_coord_variables,
dims,
check_default=True):
assert isinstance(indexes, dict), indexes
assert all(isinstance(v, Index) for v in indexes.values()), {
k: type(v)
for k, v in indexes.items()
}
index_vars = {
k
for k, v in possible_coord_variables.items()
if isinstance(v, IndexVariable)
}
assert indexes.keys() <= index_vars, (set(indexes), index_vars)
# check pandas index wrappers vs. coordinate data adapters
for k, index in indexes.items():
if isinstance(index, PandasIndex):
pd_index = index.index
var = possible_coord_variables[k]
assert (index.dim, ) == var.dims, (pd_index, var)
if k == index.dim:
# skip multi-index levels here (checked below)
assert index.coord_dtype == var.dtype, (index.coord_dtype,
var.dtype)
assert isinstance(var._data.array, pd.Index), var._data.array
# TODO: check identity instead of equality?
assert pd_index.equals(var._data.array), (pd_index, var)
if isinstance(index, PandasMultiIndex):
pd_index = index.index
for name in index.index.names:
assert name in possible_coord_variables, (pd_index, index_vars)
var = possible_coord_variables[name]
assert (index.dim, ) == var.dims, (pd_index, var)
assert index.level_coords_dtype[name] == var.dtype, (
index.level_coords_dtype[name],
var.dtype,
)
assert isinstance(var._data.array,
pd.MultiIndex), var._data.array
assert pd_index.equals(var._data.array), (pd_index, var)
# check all all levels are in `indexes`
assert name in indexes, (name, set(indexes))
# index identity is used to find unique indexes in `indexes`
assert index is indexes[name], (pd_index, indexes[name].index)
if check_default:
defaults = default_indexes(possible_coord_variables, dims)
assert indexes.keys() == defaults.keys(), (set(indexes), set(defaults))
assert all(v.equals(defaults[k]) for k, v in indexes.items()), (
indexes,
defaults,
)
def _assert_indexes_invariants_checks(indexes, possible_coord_variables, dims):
assert isinstance(indexes, OrderedDict), indexes
assert all(isinstance(v, pd.Index) for v in indexes.values()), {
k: type(v) for k, v in indexes.items()
}
index_vars = {
k for k, v in possible_coord_variables.items() if isinstance(v, IndexVariable)
}
assert indexes.keys() <= index_vars, (set(indexes), index_vars)
# Note: when we support non-default indexes, these checks should be opt-in
# only!
defaults = default_indexes(possible_coord_variables, dims)
assert indexes.keys() == defaults.keys(), (set(indexes), set(defaults))
assert all(v.equals(defaults[k]) for k, v in indexes.items()), (indexes, defaults)
def _assert_indexes_invariants_checks(indexes, possible_coord_variables, dims):
import xarray as xr
assert isinstance(indexes, OrderedDict), indexes
assert all(isinstance(v, pd.Index) for v in indexes.values()), \
{k: type(v) for k, v in indexes.items()}
index_vars = {k for k, v in possible_coord_variables.items()
if isinstance(v, xr.IndexVariable)}
assert indexes.keys() <= index_vars, (set(indexes), index_vars)
# Note: when we support non-default indexes, these checks should be opt-in
# only!
defaults = default_indexes(possible_coord_variables, dims)
assert indexes.keys() == defaults.keys(), \
(set(indexes), set(defaults))
assert all(v.equals(defaults[k]) for k, v in indexes.items()), \
(indexes, defaults)