test zarr.save with multiple arrays

Author: brokkoli71

tests/test_api.py

@@ -132,6 +132,31 @@ async def test_open_group_unspecified_version(
         assert g2.metadata.zarr_format == zarr_format
 
 
+@pytest.mark.parametrize("store", ["local", "memory", "zip"], indirect=["store"])
+@pytest.mark.parametrize("n_args", [2, 1, 0])
+@pytest.mark.parametrize("n_kwargs", [2, 1, 0])
+def test_save(store: Store, n_args: int, n_kwargs: int) -> None:
+    data = np.arange(10)
+    if n_kwargs == 0 and n_args == 0:
+        with pytest.raises(ValueError):
+            save(store)
+        return
+    args = [np.arange(10) for _ in range(n_args)]
+    kwargs = {f"arg_{i}": data for i in range(n_kwargs)}
+    save(store, *args, **kwargs)
+    # open arrays
+    if n_args == 1 and n_kwargs == 0:
+        a = open(store)
+        assert isinstance(a, Array)
+        assert_array_equal(a, data)
+    else:
+        g = open(store)
+        assert isinstance(g, Group)
+        for a in g.array_values():
+            assert_array_equal(a, data)
+        assert g.nmembers() == n_args + n_kwargs
+
+
 def test_save_errors() -> None:
     with pytest.raises(ValueError):
         # no arrays provided
@@ -145,7 +170,7 @@ def test_save_errors() -> None:
     with pytest.raises(TypeError):
         # mode is no valid argument and would get handled as an array
         a = np.arange(10)
-        zarr.save('data/example.zarr', a, mode='w')
+        zarr.save("data/example.zarr", a, mode="w")
 
 
 def test_open_with_mode_r(tmp_path: pathlib.Path) -> None: