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

Function test_dataset

xarray/tests/test_interp.py:606–626  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

604
605@requires_scipy
606def test_dataset() -> None:
607 ds = create_test_data()
608 ds.attrs["foo"] = "var"
609 ds["var1"].attrs["buz"] = "var2"
610 new_dim2 = xr.DataArray([0.11, 0.21, 0.31], dims="z")
611 interpolated = ds.interp(dim2=new_dim2)
612
613 assert_allclose(interpolated["var1"], ds["var1"].interp(dim2=new_dim2))
614 assert interpolated["var3"].equals(ds["var3"])
615
616 # make sure modifying interpolated does not affect the original dataset
617 interpolated["var1"][:, 1] = 1.0
618 interpolated["var2"][:, 1] = 1.0
619 interpolated["var3"][:, 1] = 1.0
620
621 assert not interpolated["var1"].equals(ds["var1"])
622 assert not interpolated["var2"].equals(ds["var2"])
623 assert not interpolated["var3"].equals(ds["var3"])
624 # attrs should be kept
625 assert interpolated.attrs["foo"] == "var"
626 assert interpolated["var1"].attrs["buz"] == "var2"
627
628
629@pytest.mark.parametrize("case", [pytest.param(0, id="2D"), pytest.param(3, id="3D")])

Callers

nothing calls this directly

Calls 4

assert_allcloseFunction · 0.90
create_test_dataFunction · 0.85
interpMethod · 0.45
equalsMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…