MCPcopy Index your code
hub / github.com/pydata/xarray / test_rename

Method test_rename

xarray/tests/test_dataarray.py:2009–2089  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

2007 assert (filled.values == srs.values[1:]).all()
2008
2009 def test_rename(self) -> None:
2010 da = xr.DataArray(
2011 [1, 2, 3], dims="dim", name="name", coords={"coord": ("dim", [5, 6, 7])}
2012 )
2013
2014 # change name
2015 renamed_name = da.rename("name_new")
2016 assert renamed_name.name == "name_new"
2017 expected_name = da.copy()
2018 expected_name.name = "name_new"
2019 assert_identical(renamed_name, expected_name)
2020
2021 # change name to None?
2022 renamed_noname = da.rename(None)
2023 assert renamed_noname.name is None
2024 expected_noname = da.copy()
2025 expected_noname.name = None
2026 assert_identical(renamed_noname, expected_noname)
2027 renamed_noname = da.rename()
2028 assert renamed_noname.name is None
2029 assert_identical(renamed_noname, expected_noname)
2030
2031 # change dim
2032 renamed_dim = da.rename({"dim": "dim_new"})
2033 assert renamed_dim.dims == ("dim_new",)
2034 expected_dim = xr.DataArray(
2035 [1, 2, 3],
2036 dims="dim_new",
2037 name="name",
2038 coords={"coord": ("dim_new", [5, 6, 7])},
2039 )
2040 assert_identical(renamed_dim, expected_dim)
2041
2042 # change dim with kwargs
2043 renamed_dimkw = da.rename(dim="dim_new")
2044 assert renamed_dimkw.dims == ("dim_new",)
2045 assert_identical(renamed_dimkw, expected_dim)
2046
2047 # change coords
2048 renamed_coord = da.rename({"coord": "coord_new"})
2049 assert "coord_new" in renamed_coord.coords
2050 expected_coord = xr.DataArray(
2051 [1, 2, 3], dims="dim", name="name", coords={"coord_new": ("dim", [5, 6, 7])}
2052 )
2053 assert_identical(renamed_coord, expected_coord)
2054
2055 # change coords with kwargs
2056 renamed_coordkw = da.rename(coord="coord_new")
2057 assert "coord_new" in renamed_coordkw.coords
2058 assert_identical(renamed_coordkw, expected_coord)
2059
2060 # change coord and dim
2061 renamed_both = da.rename({"dim": "dim_new", "coord": "coord_new"})
2062 assert renamed_both.dims == ("dim_new",)
2063 assert "coord_new" in renamed_both.coords
2064 expected_both = xr.DataArray(
2065 [1, 2, 3],
2066 dims="dim_new",

Callers

nothing calls this directly

Calls 3

renameMethod · 0.95
copyMethod · 0.95
assert_identicalFunction · 0.90

Tested by

no test coverage detected