tensorstore_zarr_array_create_write_read_slice.py

1import tensorstore as ts
2import numpy as np
3
4# Create a 100x100 zarr array on the local file system
5dataset = ts.open({
6    'driver': 'zarr',
7    'kvstore': 'file:///tmp/my_dataset/',
8    'metadata': {
9        'shape': [100, 100],
10        'dtype': 'int32',
11        'chunks': [10, 10],
12    },
13    'create': True,
14    'delete_existing': True,
15}).result()
16
17# Create a 10x10 numpy array
18data = np.arange(100, dtype='int32').reshape((10, 10))
19
20# Write the data to a slice of the tensorstore
21dataset[20:30, 40:50] = data
22
23# Read the data back from the tensorstore
24read_data = dataset[20:30, 40:50].read().result()
25
26# Verify the data
27np.testing.assert_array_equal(data, read_data)