move for ZipStore

Author: brokkoli71

src/zarr/storage/_zip.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import os
+import shutil
 import threading
 import time
 import zipfile
@@ -288,3 +289,16 @@ async def list_dir(self, prefix: str) -> AsyncIterator[str]:
                     if k not in seen:
                         seen.add(k)
                         yield k
+
+    async def move(self, path: Path | str) -> None:
+        # docstring inherited
+
+        if isinstance(path, str):
+            path = Path(path)
+        if not isinstance(path, Path):
+            raise TypeError(
+                f"'path' must be a string or Path instance. Got an instance of {type(path)} instead."
+            )
+        os.makedirs(path, exist_ok=True)
+        shutil.move(self.path, path)
+        self.path = path

tests/test_store/test_zip.py

@@ -10,6 +10,7 @@
 import pytest
 
 import zarr
+from zarr import create_array
 from zarr.core.buffer import Buffer, cpu, default_buffer_prototype
 from zarr.storage import ZipStore
 from zarr.testing.store import StoreTests
@@ -135,3 +136,17 @@ def test_externally_zipped_store(self, tmp_path: Path) -> None:
         zipped = zarr.open_group(ZipStore(zip_path, mode="r"), mode="r")
         assert list(zipped.keys()) == list(root.keys())
         assert list(zipped["foo"].keys()) == list(root["foo"].keys())
+
+    async def test_move(self, tmp_path: Path):
+        origin = tmp_path / "origin"
+        destination = tmp_path / "destintion"
+
+        store = await ZipStore.open(path=origin, mode="w")
+        array = create_array(store, data=np.arange(10))
+
+        await store.move(destination)
+
+        assert store.path == destination
+        assert destination.exists()
+        assert not origin.exists()
+        assert np.array_equal(array[...], np.arange(10))