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)