add a test.

Co-authored-by: Zachary Blackwood <[email protected]>
Co-authored-by: Nathan Lis <[email protected]>

Author: 3 people

xarray/backends/api.py

@@ -1337,8 +1337,15 @@ def to_zarr(
     else:
         from fsspec import get_mapper
 
+        if not isinstance(store, str):
+            raise ValueError(
+                f"store must be a string to use storage_options. Got {type(store)}"
+            )
         mapper = get_mapper(store, **storage_options)
-        chunk_mapper = get_mapper(chunk_store, **storage_options)
+        if chunk_store is not None:
+            chunk_mapper = get_mapper(chunk_store, **storage_options)
+        else:
+            chunk_mapper = chunk_store
 
     if encoding is None:
         encoding = {}

xarray/backends/zarr.py

@@ -713,6 +713,9 @@ def open_zarr(
         falling back to read non-consolidated metadata if that fails.
     chunk_store : MutableMapping, optional
         A separate Zarr store only for chunk data.
+    storage_options : dict, optional
+        Any additional parameters for the storage backend (ignored for local
+        paths).
     decode_timedelta : bool, optional
         If True, decode variables and coordinates with time units in
         {'days', 'hours', 'minutes', 'seconds', 'milliseconds', 'microseconds'}

xarray/tests/test_backends.py

@@ -2388,6 +2388,15 @@ def create_zarr_target(self):
             yield tmp
 
 
+@requires_fsspec
+def test_zarr_storage_options():
+    ds = create_test_data()
+    store_target = "memory://test.zarr"
+    ds.to_zarr(store_target, storage_options={"test": "zarr_write"})
+    ds_a = xr.open_zarr(store_target, storage_options={"test": "zarr_read"})
+    assert_identical(ds, ds_a)
+
+
 @requires_scipy
 class TestScipyInMemoryData(CFEncodedBase, NetCDF3Only):
     engine = "scipy"