diff --git a/xarray/tests/test_backends.py b/xarray/tests/test_backends.py index 0f47b36e50b..86986cedb06 100644 --- a/xarray/tests/test_backends.py +++ b/xarray/tests/test_backends.py @@ -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) - - 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_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) + + 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