Test all datasets

ratt-ru · Mar 19, 2024 · f93211e · f93211e
1 parent 5c7eff5
commit f93211e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/daskms/experimental/zarr/tests/test_zarr.py b/daskms/experimental/zarr/tests/test_zarr.py
@@ -429,5 +429,6 @@ def test_xarray_reading_daskms_written_dataset(ms, tmp_path_factory):
     path = store / "test.zarr"
     dask.compute(xds_to_zarr(datasets, path, consolidated=True))
 
-    ds = xarray.open_zarr(path / "MAIN" / "MAIN_0")
-    assert ds == datasets[0]
+    for i, mem_ds in enumerate(datasets):
+        ds = xarray.open_zarr(path / "MAIN" / f"MAIN_{i}")
+        assert ds == mem_ds