Skip to content

Commit d8760b7

Browse files
committed
Add test that fails on multi-index DataArrays
1 parent aee5b29 commit d8760b7

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

xarray/tests/test_dataarray.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3510,6 +3510,18 @@ def test_to_dataframe_multiindex(self) -> None:
35103510
assert_array_equal(index_pd.levels[1], ["a", "b"])
35113511
assert_array_equal(index_pd.levels[2], [5, 6, 7])
35123512

3513+
# test converting a dataframe MultiIndexed along a single dimension
3514+
mindex_single = pd.MultiIndex.from_product(
3515+
[list(range(6)), list("ab")], names=["A", "B"]
3516+
)
3517+
3518+
arr_multi_single = DataArray(
3519+
arr_np.flatten(), [("MI", mindex_single)], dims="MI", name="test"
3520+
)
3521+
actual_df = arr_multi_single.to_dataframe()
3522+
expected_df = arr_multi_single.to_series().to_frame()
3523+
assert expected_df.equals(actual_df)
3524+
35133525
def test_to_dataframe_0length(self) -> None:
35143526
# regression test for #3008
35153527
arr_np = np.random.randn(4, 0)

0 commit comments

Comments
 (0)