Skip to content

Commit 69172fb

Browse files
committed
fix countingdict tests
1 parent aeecda8 commit 69172fb

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

tests/test_indexing.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
if TYPE_CHECKING:
3535
from collections.abc import AsyncGenerator
3636

37+
from zarr.abc.store import ByteRequest
3738
from zarr.core.buffer import BufferPrototype
3839
from zarr.core.buffer.core import Buffer
3940

@@ -83,6 +84,21 @@ async def set(self, key: str, value: Buffer, byte_range: tuple[int, int] | None
8384
self.counter["__setitem__", key_suffix] += 1
8485
return await super().set(key, value, byte_range)
8586

87+
def get_sync(
88+
self,
89+
key: str,
90+
prototype: BufferPrototype | None = None,
91+
byte_range: ByteRequest | None = None,
92+
) -> Buffer | None:
93+
key_suffix = "/".join(key.split("/")[1:])
94+
self.counter["__getitem__", key_suffix] += 1
95+
return super().get_sync(key, prototype, byte_range)
96+
97+
def set_sync(self, key: str, value: Buffer) -> None:
98+
key_suffix = "/".join(key.split("/")[1:])
99+
self.counter["__setitem__", key_suffix] += 1
100+
return super().set_sync(key, value)
101+
86102

87103
def test_normalize_integer_selection() -> None:
88104
assert 1 == normalize_integer_selection(1, 100)

0 commit comments

Comments
 (0)