diff --git a/xarray/tests/test_backends.py b/xarray/tests/test_backends.py index a8722d59659..86986cedb06 100644 --- a/xarray/tests/test_backends.py +++ b/xarray/tests/test_backends.py @@ -2232,10 +2232,10 @@ def test_chunk_encoding_with_dask(self) -> None: pass def test_drop_encoding(self): - ds = open_example_dataset("example_1.nc") - encodings = {v: {**ds[v].encoding} for v in ds.data_vars} - with self.create_zarr_target() as store: - ds.to_zarr(store, encoding=encodings) + with open_example_dataset("example_1.nc") as ds: + encodings = {v: {**ds[v].encoding} for v in ds.data_vars} + with self.create_zarr_target() as store: + ds.to_zarr(store, encoding=encodings) def test_hidden_zarr_keys(self) -> None: expected = create_test_data() @@ -4417,13 +4417,19 @@ def test_inline_array(self) -> None: def num_graph_nodes(obj): return len(obj.__dask_graph__()) - not_inlined_ds = open_dataset(tmp, inline_array=False, chunks=chunks) - inlined_ds = open_dataset(tmp, inline_array=True, chunks=chunks) - assert num_graph_nodes(inlined_ds) < num_graph_nodes(not_inlined_ds) + with open_dataset( + tmp, inline_array=False, chunks=chunks + ) as not_inlined_ds, open_dataset( + tmp, inline_array=True, chunks=chunks + ) as inlined_ds: + assert num_graph_nodes(inlined_ds) < num_graph_nodes(not_inlined_ds) - not_inlined_da = open_dataarray(tmp, inline_array=False, chunks=chunks) - inlined_da = open_dataarray(tmp, inline_array=True, chunks=chunks) - assert num_graph_nodes(inlined_da) < num_graph_nodes(not_inlined_da) + with open_dataarray( + tmp, inline_array=False, chunks=chunks + ) as not_inlined_da, open_dataarray( + tmp, inline_array=True, chunks=chunks + ) as inlined_da: + assert num_graph_nodes(inlined_da) < num_graph_nodes(not_inlined_da) @requires_scipy_or_netCDF4 @@ -5335,8 +5341,8 @@ def test_raise_writing_to_nczarr(self, mode) -> None: @requires_netCDF4 @requires_dask def test_pickle_open_mfdataset_dataset(): - ds = open_example_mfdataset(["bears.nc"]) - assert_identical(ds, pickle.loads(pickle.dumps(ds))) + with open_example_mfdataset(["bears.nc"]) as ds: + assert_identical(ds, pickle.loads(pickle.dumps(ds))) @requires_zarr