Method test_region_write

xarray/tests/test_backends.py:4016–4110

(self)

    @requires_dask
    def test_region_write(self) -> None:
        ds = Dataset({"foo": ("x", [1, 2, 3])}, coords={"x": [1, 2, 3]}).chunk()
        with self.create_zarr_target() as store:
            if has_zarr_v3:
                expected = {
                    "set": 5,
                    "get": 2,
                    "list_dir": 2,
                    "list_prefix": 4,
                }
            else:
                expected = {
                    "iter": 1,
                    "contains": 16,
                    "setitem": 9,
                    "getitem": 13,
                    "listdir": 0,
                    "list_prefix": 5,
                }

            patches = self.make_patches(store)
            with patch.multiple(KVStore, **patches):
                ds.to_zarr(store, mode="w", compute=False)
            self.check_requests(expected, patches)

            # v2024.03.0: {'iter': 5, 'contains': 2, 'setitem': 1, 'getitem': 6, 'listdir': 5, 'list_prefix': 0}
            # 6057128b: {'iter': 4, 'contains': 2, 'setitem': 1, 'getitem': 5, 'listdir': 4, 'list_prefix': 0}
            if has_zarr_v3:
                expected = {
                    "set": 1,
                    "get": 3,
                    "list_dir": 0,
                    "list_prefix": 0,
                }
            else:
                expected = {
                    "iter": 1,
                    "contains": 6,
                    "setitem": 1,
                    "getitem": 7,
                    "listdir": 0,
                    "list_prefix": 0,
                }

            patches = self.make_patches(store)
            with patch.multiple(KVStore, **patches):
                ds.to_zarr(store, region={"x": slice(None)})
            self.check_requests(expected, patches)

            # v2024.03.0: {'iter': 6, 'contains': 4, 'setitem': 1, 'getitem': 11, 'listdir': 6, 'list_prefix': 0}
            # 6057128b: {'iter': 4, 'contains': 2, 'setitem': 1, 'getitem': 7, 'listdir': 4, 'list_prefix': 0}
            if has_zarr_v3:
                expected = {
                    "set": 1,
                    "get": 4,
                    "list_dir": 0,
                    "list_prefix": 0,
                }