(self)
| 961 | data.coords["level_1"] = range(4) |
| 962 | |
| 963 | def test_coords_set(self) -> None: |
| 964 | one_coord = Dataset({"x": ("x", [0]), "yy": ("x", [1]), "zzz": ("x", [2])}) |
| 965 | two_coords = Dataset({"zzz": ("x", [2])}, {"x": ("x", [0]), "yy": ("x", [1])}) |
| 966 | all_coords = Dataset( |
| 967 | coords={"x": ("x", [0]), "yy": ("x", [1]), "zzz": ("x", [2])} |
| 968 | ) |
| 969 | |
| 970 | actual = one_coord.set_coords("x") |
| 971 | assert_identical(one_coord, actual) |
| 972 | actual = one_coord.set_coords(["x"]) |
| 973 | assert_identical(one_coord, actual) |
| 974 | |
| 975 | actual = one_coord.set_coords("yy") |
| 976 | assert_identical(two_coords, actual) |
| 977 | |
| 978 | actual = one_coord.set_coords(["yy", "zzz"]) |
| 979 | assert_identical(all_coords, actual) |
| 980 | |
| 981 | actual = one_coord.reset_coords() |
| 982 | assert_identical(one_coord, actual) |
| 983 | actual = two_coords.reset_coords() |
| 984 | assert_identical(one_coord, actual) |
| 985 | actual = all_coords.reset_coords() |
| 986 | assert_identical(one_coord, actual) |
| 987 | |
| 988 | actual = all_coords.reset_coords(["yy", "zzz"]) |
| 989 | assert_identical(one_coord, actual) |
| 990 | actual = all_coords.reset_coords("zzz") |
| 991 | assert_identical(two_coords, actual) |
| 992 | |
| 993 | with pytest.raises(ValueError, match=r"cannot remove index"): |
| 994 | one_coord.reset_coords("x") |
| 995 | |
| 996 | actual = all_coords.reset_coords("zzz", drop=True) |
| 997 | expected = all_coords.drop_vars("zzz") |
| 998 | assert_identical(expected, actual) |
| 999 | expected = two_coords.drop_vars("zzz") |
| 1000 | assert_identical(expected, actual) |
| 1001 | |
| 1002 | def test_coords_to_dataset(self) -> None: |
| 1003 | orig = Dataset({"foo": ("y", [-1, 0, 1])}, {"x": 10, "y": [2, 3, 4]}) |
nothing calls this directly
no test coverage detected