Skip to content

Commit

Permalink
Pass dimension_separator param when creating fixtures
Browse files Browse the repository at this point in the history
  • Loading branch information
shlomnissan committed Jun 6, 2024
1 parent d6466a0 commit e6a0c46
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions fixtures/generate_fixtures.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,14 @@
FIXTURES_FOLDER = "./"


def create_simple_array(store, dtype, compression=None, order="C", ndim=2, write_chunks=False):
def create_simple_array(store, dtype, compression=None, order="C", ndim=2, write_chunks=False, dimension_separator=None):
arr = zarr.open(
store=store,
shape=(8, 8) if ndim == 2 else (8, 8, 8),
chunks=(2, None) if ndim == 2 else (2, None, None),
dtype=dtype,
fill_value=0,
dimension_separator=dimension_separator,
mode="w",
compression=compression,
order=order,
Expand Down Expand Up @@ -80,19 +81,18 @@ def generate_fixtures():
)

# nested
# TODO: Use latest zarr-python once https://github.com/zarr-developers/zarr-python/pull/716 is merged
store = zarr.storage.FSStore(
os.path.join(FIXTURES_FOLDER, "simple_nested.zarr"),
key_separator="/",
auto_mkdir=True,
)
create_simple_array(
store=store, dtype=">i4", compression="blosc", write_chunks=True
store=store,
dtype=">i4",
compression="blosc",
write_chunks=True,
dimension_separator="/",
)
# Manually add dimension separator to array meta
meta = json_loads(store[".zarray"])
meta["dimension_separator"] = "/"
store[".zarray"] = json_dumps(meta)

# Float 16
path = os.path.join(FIXTURES_FOLDER, "simple_float16_LE.zarr")
Expand Down

0 comments on commit e6a0c46

Please sign in to comment.