def test_save_errors():
with assert_raises(ValueError):
# no arrays provided
save_group('data/group.zarr')
with assert_raises(ValueError):
# no arrays provided
save('data/group.zarr')
def test_save_errors():
with pytest.raises(ValueError):
# no arrays provided
save_group('data/group.zarr')
with pytest.raises(ValueError):
# no arrays provided
save('data/group.zarr')
def test_save_errors(zarr_version):
with pytest.raises(ValueError):
# no arrays provided
save_group('data/group.zarr', zarr_version=zarr_version)
with pytest.raises(TypeError):
# no array provided
save_array('data/group.zarr', zarr_version=zarr_version)
with pytest.raises(ValueError):
# no arrays provided
save('data/group.zarr', zarr_version=zarr_version)
def test_zarr_v3_save_errors():
x = np.ones(8)
with pytest.raises(ValueError):
# no path provided
save_group('data/group.zr3', x, zarr_version=3)
with pytest.raises(ValueError):
# no path provided
save_array('data/group.zr3', x, zarr_version=3)
with pytest.raises(ValueError):
# no path provided
save('data/group.zr3', x, zarr_version=3)
def test_lazy_loader():
foo = np.arange(100)
bar = np.arange(100, 0, -1)
store = 'data/group.zarr'
save(store, foo=foo, bar=bar)
loader = load(store)
assert 'foo' in loader
assert 'bar' in loader
assert 'baz' not in loader
assert len(loader) == 2
assert sorted(loader) == ['bar', 'foo']
assert_array_equal(foo, loader['foo'])
assert_array_equal(bar, loader['bar'])
def test_lazy_loader():
foo = np.arange(100)
bar = np.arange(100, 0, -1)
store = 'data/group.zarr'
save(store, foo=foo, bar=bar)
loader = load(store)
assert 'foo' in loader
assert 'bar' in loader
assert 'baz' not in loader
assert len(loader) == 2
assert sorted(loader) == ['bar', 'foo']
assert_array_equal(foo, loader['foo'])
assert_array_equal(bar, loader['bar'])
def test_lazy_loader(zarr_version):
foo = np.arange(100)
bar = np.arange(100, 0, -1)
store = 'data/group.zarr' if zarr_version == 2 else 'data/group.zr3'
kwargs = _init_creation_kwargs(zarr_version)
save(store, foo=foo, bar=bar, **kwargs)
loader = load(store, **kwargs)
assert 'foo' in loader
assert 'bar' in loader
assert 'baz' not in loader
assert len(loader) == 2
assert sorted(loader) == ['bar', 'foo']
assert_array_equal(foo, loader['foo'])
assert_array_equal(bar, loader['bar'])
assert 'LazyLoader: ' in repr(loader)
def test_load_array(zarr_version):
foo = np.arange(100)
bar = np.arange(100, 0, -1)
store = 'data/group.zarr' if zarr_version == 2 else 'data/group.zr3'
kwargs = _init_creation_kwargs(zarr_version)
save(store, foo=foo, bar=bar, **kwargs)
# can also load arrays directly into a numpy array
for array_name in ['foo', 'bar']:
array_path = 'dataset/' + array_name if zarr_version == 3 else array_name
array = load(store, path=array_path, zarr_version=zarr_version)
assert isinstance(array, np.ndarray)
if array_name == 'foo':
assert_array_equal(foo, array)
else:
assert_array_equal(bar, array)
