Use store prefix

Author: kylebarron

pyproject.toml

@@ -64,8 +64,7 @@ keywords = ["Python", "compressed", "ndimensional-arrays", "zarr"]
 # User extras
 remote = [
     "fsspec>=2023.10.0",
-    "obstore==0.4.0b2",
-
+    "obstore==0.4.0b3",
 ]
 gpu = [
     "cupy-cuda12x",

tests/test_store/test_object.py

@@ -1,41 +1,38 @@
 # ruff: noqa: E402
+import pickle
+from typing import Any
+
 import pytest
 
 obstore = pytest.importorskip("obstore")
-
-import pickle
-import re
+from obstore.store import LocalStore
 
 from zarr.core.buffer import Buffer, cpu
 from zarr.storage import ObjectStore
 from zarr.testing.store import StoreTests
 
-PATTERN = r'LocalStore\("([^"]+)"\)'
-
 
 class TestObjectStore(StoreTests[ObjectStore, cpu.Buffer]):
     store_cls = ObjectStore
     buffer_cls = cpu.Buffer
 
     @pytest.fixture
-    def store_kwargs(self, tmpdir) -> dict[str, str | bool]:
-        store = obstore.store.LocalStore(prefix=tmpdir)
+    def store_kwargs(self, tmpdir) -> dict[str, Any]:
+        store = LocalStore(prefix=tmpdir)
         return {"store": store, "read_only": False}
 
     @pytest.fixture
     def store(self, store_kwargs: dict[str, str | bool]) -> ObjectStore:
         return self.store_cls(**store_kwargs)
 
     async def get(self, store: ObjectStore, key: str) -> Buffer:
-        # TODO: Use new store.prefix in obstore 0.4.0
-        store_path = re.search(PATTERN, str(store)).group(1)
-        new_local_store = obstore.store.LocalStore(prefix=store_path)
+        assert isinstance(store.store, LocalStore)
+        new_local_store = LocalStore(prefix=store.store.prefix)
         return self.buffer_cls.from_bytes(obstore.get(new_local_store, key).bytes())
 
     async def set(self, store: ObjectStore, key: str, value: Buffer) -> None:
-        # TODO: Use new store.prefix in obstore 0.4.0
-        store_path = re.search(PATTERN, str(store)).group(1)
-        new_local_store = obstore.store.LocalStore(prefix=store_path)
+        assert isinstance(store.store, LocalStore)
+        new_local_store = LocalStore(prefix=store.store.prefix)
         obstore.put(new_local_store, key, value.to_bytes())
 
     def test_store_repr(self, store: ObjectStore) -> None: