Skip to content

Commit 96e3626

Browse files
committed
nit refactor
1 parent ad56e7b commit 96e3626

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

xarray/testing/assertions.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -264,13 +264,15 @@ def _assert_indexes_invariants_checks(
264264
}
265265

266266
index_vars = {
267-
k for k, v in possible_coord_variables.items() if isinstance(v, IndexVariable)
267+
k: v
268+
for k, v in possible_coord_variables.items()
269+
if isinstance(v, IndexVariable)
268270
}
269-
assert indexes.keys() <= index_vars, (set(indexes), index_vars)
271+
index_var_names = set(index_vars)
272+
assert indexes.keys() <= index_var_names, (set(indexes), index_var_names)
270273

271-
for k, v in possible_coord_variables.items():
272-
if isinstance(v, IndexVariable):
273-
assert k == v.name, (k, v.name)
274+
for k, v in index_vars.items():
275+
assert k == v.name, (k, v.name)
274276

275277
# check pandas index wrappers vs. coordinate data adapters
276278
for k, index in indexes.items():
@@ -287,7 +289,7 @@ def _assert_indexes_invariants_checks(
287289
if isinstance(index, PandasMultiIndex):
288290
pd_index = index.index
289291
for name in index.index.names:
290-
assert name in possible_coord_variables, (pd_index, index_vars)
292+
assert name in possible_coord_variables, (pd_index, index_var_names)
291293
var = possible_coord_variables[name]
292294
assert (index.dim,) == var.dims, (pd_index, var)
293295
assert index.level_coords_dtype[name] == var.dtype, (

0 commit comments

Comments
 (0)