def test_stack_non_unique(self) -> None:
prod_vars = {
"x": xr.Variable("x", pd.Index(["b", "a"]), attrs={"foo": "bar"}),
"y": xr.Variable("y", pd.Index([1, 1, 2])),
}
index = PandasMultiIndex.stack(prod_vars, "z")
np.testing.assert_array_equal(index.index.codes,
[[0, 0, 0, 1, 1, 1], [0, 0, 1, 0, 0, 1]])
np.testing.assert_array_equal(index.index.levels[0], ["b", "a"])
np.testing.assert_array_equal(index.index.levels[1], [1, 2])
def test_stack(self) -> None:
prod_vars = {
"x": xr.Variable("x", pd.Index(["b", "a"]), attrs={"foo": "bar"}),
"y": xr.Variable("y", pd.Index([1, 3, 2])),
}
index = PandasMultiIndex.stack(prod_vars, "z")
assert index.dim == "z"
assert index.index.names == ["x", "y"]
np.testing.assert_array_equal(
index.index.codes, [[0, 0, 0, 1, 1, 1], [0, 1, 2, 0, 1, 2]]
)
with pytest.raises(
ValueError, match=r"conflicting dimensions for multi-index product.*"
):
PandasMultiIndex.stack(
{"x": xr.Variable("x", ["a", "b"]), "x2": xr.Variable("x", [1, 2])},
"z",
)