assert type in test_load_zip and test_load_local

Author: brokkoli71

tests/test_api.py

@@ -401,36 +401,35 @@ def test_load_array(memory_store: Store) -> None:
 
 
 @pytest.mark.parametrize("path", ["data", None])
-def test_load_zip(tmp_path: pathlib.Path, path: str | None) -> None:
+@pytest.mark.parametrize("load_read_only", [True, False, None])
+def test_load_zip(tmp_path: pathlib.Path, path: str | None, load_read_only: bool | None) -> None:
     file = tmp_path / "test.zip"
     data = np.arange(100).reshape(10, 10)
 
     with ZipStore(file, mode="w", read_only=False) as zs:
         save(zs, data, path=path)
-    with ZipStore(file, mode="r", read_only=False) as zs:
+    with ZipStore(file, mode="r", read_only=load_read_only) as zs:
         result = zarr.load(store=zs, path=path)
+        assert isinstance(result, np.ndarray)
         assert np.array_equal(result, data)
-    with ZipStore(file, mode="r") as zs:
+    with ZipStore(file, read_only=load_read_only) as zs:
         result = zarr.load(store=zs, path=path)
-        assert_array_equal(result, data)
-    with ZipStore(file, read_only=True) as zs:
-        result = zarr.load(store=zs, path=path)
-        assert_array_equal(result, data)
+        assert isinstance(result, np.ndarray)
+        assert np.array_equal(result, data)
 
 
 @pytest.mark.parametrize("path", ["data", None])
-def test_load_local(tmp_path: pathlib.Path, path: str | None) -> None:
+@pytest.mark.parametrize("load_read_only", [True, False])
+def test_load_local(tmp_path: pathlib.Path, path: str | None, load_read_only: bool) -> None:
     file = tmp_path / "test.zip"
     data = np.arange(100).reshape(10, 10)
 
     with LocalStore(file, read_only=False) as zs:
         save(zs, data, path=path)
-    with LocalStore(file, read_only=False) as zs:
+    with LocalStore(file, read_only=load_read_only) as zs:
         result = zarr.load(store=zs, path=path)
-        assert_array_equal(result, data)
-    with LocalStore(file, read_only=True) as zs:
-        result = zarr.load(store=zs, path=path)
-        assert_array_equal(result, data)
+        assert isinstance(result, np.ndarray)
+        assert np.array_equal(result, data)
 
 
 def test_tree() -> None: