| 177 | assert_array_equal(actual.data, actual.values) |
| 178 | |
| 179 | def test_indexes(self) -> None: |
| 180 | array = DataArray(np.zeros((2, 3)), [("x", [0, 1]), ("y", ["a", "b", "c"])]) |
| 181 | expected_indexes = {"x": pd.Index([0, 1]), "y": pd.Index(["a", "b", "c"])} |
| 182 | expected_xindexes = { |
| 183 | k: PandasIndex(idx, k) for k, idx in expected_indexes.items() |
| 184 | } |
| 185 | assert array.xindexes.keys() == expected_xindexes.keys() |
| 186 | assert array.indexes.keys() == expected_indexes.keys() |
| 187 | assert all(isinstance(idx, pd.Index) for idx in array.indexes.values()) |
| 188 | assert all(isinstance(idx, Index) for idx in array.xindexes.values()) |
| 189 | for k in expected_indexes: |
| 190 | assert array.xindexes[k].equals(expected_xindexes[k]) |
| 191 | assert array.indexes[k].equals(expected_indexes[k]) |
| 192 | |
| 193 | def test_get_index(self) -> None: |
| 194 | array = DataArray(np.zeros((2, 3)), coords={"x": ["a", "b"]}, dims=["x", "y"]) |