Be explicit when opening Zarr arrays (#296)

Author: tomwhite

cubed/tests/primitive/test_blockwise.py

@@ -61,7 +61,7 @@ def test_blockwise(tmp_path, executor, reserved_mem):
 
     execute_pipeline(pipeline, executor=executor)
 
-    res = zarr.open(target_store)
+    res = zarr.open_array(target_store)
     assert_array_equal(res[:], np.outer([0, 1, 2], [10, 50, 100]))
 
 
@@ -124,7 +124,7 @@ def test_blockwise_with_args(tmp_path, executor):
 
     execute_pipeline(pipeline, executor=executor)
 
-    res = zarr.open(target_store)
+    res = zarr.open_array(target_store)
     assert_array_equal(
         res[:], np.transpose(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), axes=(1, 0))
     )

cubed/tests/primitive/test_rechunk.py

@@ -92,7 +92,7 @@ def test_rechunk(
     for pipeline in pipelines:
         execute_pipeline(pipeline, executor=executor)
 
-    res = zarr.open(target_store)
+    res = zarr.open_array(target_store)
     assert_array_equal(res[:], np.ones(shape))
     assert res.chunks == target_chunks
 

cubed/tests/test_core.py

@@ -141,7 +141,7 @@ def test_to_zarr(tmp_path, spec, executor):
     a = xp.asarray([[1, 2, 3], [4, 5, 6], [7, 8, 9]], chunks=(2, 2), spec=spec)
     output = tmp_path / "output.zarr"
     cubed.to_zarr(a, output, executor=executor)
-    res = zarr.open(output)
+    res = zarr.open_array(output)
     assert_array_equal(res[:], np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]))