Skip to content

Commit d4261e8

Browse files
author
Martin Durant
committed
Extens s3 test
Also: remove delitems (only rmdir ever gets called)
1 parent 78d7b7b commit d4261e8

File tree

2 files changed

+16
-12
lines changed

2 files changed

+16
-12
lines changed

zarr/storage.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1078,12 +1078,6 @@ def __delitem__(self, key):
10781078
else:
10791079
del self.map[key]
10801080

1081-
def delitems(self, keys):
1082-
if self.mode == 'r':
1083-
raise ReadOnlyError()
1084-
keys = [self._normalize_key(k) for k in keys]
1085-
self.map.delitems(keys)
1086-
10871081
def __contains__(self, key):
10881082
key = self._normalize_key(key)
10891083
return key in self.map

zarr/tests/test_storage.py

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -968,15 +968,25 @@ def test_s3_complex(self):
968968
expected[3] = 3
969969
a[:4] = expected[:4]
970970

971-
a = g.create_dataset("data_f", shape=(8, ), chunks=(1,),
971+
b = g.create_dataset("data_f", shape=(8, ), chunks=(1,),
972972
dtype=[('foo', 'S3'), ('bar', 'i4')],
973973
fill_value=(b"b", 1))
974-
a[:4] = (b"aaa", 2)
975-
g = zarr.open_group("s3://test/out.zarr", mode='r',
976-
storage_options=self.s3so)
974+
b[:4] = (b"aaa", 2)
975+
g2 = zarr.open_group("s3://test/out.zarr", mode='r',
976+
storage_options=self.s3so)
977+
978+
assert (g2.data[:] == expected).all()
979+
assert g2.data_f['foo'].tolist() == [b"aaa"] * 4 + [b"b"] * 4
980+
with pytest.raises(PermissionError):
981+
g2.data[:] = 5
982+
with pytest.raises(PermissionError):
983+
# even though overwrite=True, store is read-only, so fails
984+
g2.create_dataset("data", shape=(8, 8, 8), mode='w',
985+
fill_value=-1, chunks=(1, 1, 1), overwrite=True)
977986

978-
assert (g.data[:] == expected).all()
979-
assert g.data_f['foo'].tolist() == [b"aaa"] * 4 + [b"b"] * 4
987+
a = g.create_dataset("data", shape=(8, 8, 8), mode='w',
988+
fill_value=-1, chunks=(1, 1, 1), overwrite=True)
989+
assert (a[:] == -np.ones((8, 8, 8))).all()
980990

981991

982992
@pytest.fixture()

0 commit comments

Comments
 (0)