Method test_to_dataframe

xarray/tests/test_dataarray.py:3662–3694

(self)

            DataArray(np.random.randn(1, 2, 3, 4, 5)).to_pandas()

    def test_to_dataframe(self) -> None:
        # regression test for #260
        arr_np = np.random.randn(3, 4)

        arr = DataArray(arr_np, [("B", [1, 2, 3]), ("A", list("cdef"))], name="foo")
        expected_s = arr.to_series()
        actual_s = arr.to_dataframe()["foo"]
        assert_array_equal(np.asarray(expected_s.values), np.asarray(actual_s.values))
        assert_array_equal(np.asarray(expected_s.name), np.asarray(actual_s.name))
        assert_array_equal(expected_s.index.values, actual_s.index.values)

        actual_s = arr.to_dataframe(dim_order=["A", "B"])["foo"]
        assert_array_equal(arr_np.transpose().reshape(-1), np.asarray(actual_s.values))

        # regression test for coords with different dimensions
        arr.coords["C"] = ("B", [-1, -2, -3])
        expected_df = arr.to_series().to_frame()
        expected_df["C"] = [-1] * 4 + [-2] * 4 + [-3] * 4
        expected_df = expected_df[["C", "foo"]]
        actual_df = arr.to_dataframe()
        assert_array_equal(np.asarray(expected_df.values), np.asarray(actual_df.values))
        assert_array_equal(expected_df.columns.values, actual_df.columns.values)
        assert_array_equal(expected_df.index.values, actual_df.index.values)

        with pytest.raises(ValueError, match="does not match the set of dimensions"):
            arr.to_dataframe(dim_order=["B", "A", "C"])

        with pytest.raises(ValueError, match=r"cannot convert a scalar"):
            arr.sel(A="c", B=2).to_dataframe()

        arr.name = None  # unnamed
        with pytest.raises(ValueError, match=r"unnamed"):
            arr.to_dataframe()

    def test_to_dataframe_multiindex(self) -> None:
        # regression test for #3008