john/hooked-containers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reorg

Browse Source
This commit is contained in:
John Lancaster
2026-02-21 23:24:09 -06:00
parent 7c3e073c54
commit 071c3bd342
2 changed files with 23 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/hooked_containers/sequence.py
View File

@@ -12,27 +12,30 @@ class HookedList(HookedContainer[T], MutableSequence[T]):
_data: MutableSequence[T] _data: MutableSequence[T]
path: MutableSequence[int] path: MutableSequence[int]
def __init__(self, hook: HookFunction, existing: MutableSequence[T], path: MutableSequence[int] | None = None): def __init__(
self,
existing: MutableSequence[T],
hook: HookFunction | None = None,
path: MutableSequence[int] | None = None,
) -> None:
self.hook = hook self.hook = hook
match existing: match existing:
case HookedContainer(_data=seq):
self._data = seq
case MutableSequence() as seq: case MutableSequence() as seq:
self._data = seq self._data = seq
case HookedContainer(_data=seq):
self._data = seq
case _ as it: case _ as it:
self._data = list(it) self._data = list(it)
self._path = list(path) if path is not None else [] self._path = list(path) if path is not None else []
def __getitem__(self, s): def __getitem__(self, s):
new_path = copy(self._path)
new_path.append(s)
match self._data[s]: match self._data[s]:
case MutableSequence() as seq: case MutableSequence() as seq:
new_path = copy(self._path) return type(self)(seq, self.hook, path=new_path)
new_path.append(s)
return type(self)(self.hook, existing=seq, path=new_path)
case HookedContainer(_data=seq): case HookedContainer(_data=seq):
new_path = copy(self._path) return type(self)(seq, self.hook, path=new_path)
new_path.append(s)
return type(self)(self.hook, existing=seq, path=new_path)
case _ as item: case _ as item:
return item return item

22
tests/test_sequence.py
View File

@@ -12,48 +12,48 @@ def get_id(a: Any):
class TestHookedList: class TestHookedList:
class TestConstruction: class TestConstruction:
def test_empty(self): def test_empty(self):
lst = HookedList(None, existing=[]) lst = HookedList(existing=[])
assert list(lst) == [] assert list(lst) == []
def test_with_items(self): def test_with_items(self):
lst = HookedList(None, existing=[1, 2, 3]) lst = HookedList(existing=[1, 2, 3])
assert list(lst) == [1, 2, 3] assert list(lst) == [1, 2, 3]
def test_recreation(self): def test_recreation(self):
initial = [1, 2, 3] initial = [1, 2, 3]
initial_id = id(initial) initial_id = id(initial)
lst = HookedList(None, existing=initial) lst = HookedList(existing=initial)
assert id(lst._data) == initial_id assert id(lst._data) == initial_id
lst2 = HookedList(None, existing=lst) lst2 = HookedList(existing=lst)
assert id(lst2._data) == initial_id assert id(lst2._data) == initial_id
class TestSeqOps: class TestSeqOps:
def test_len(self): def test_len(self):
lst = HookedList(None, existing=[1, 2, 3]) lst = HookedList(existing=[1, 2, 3])
assert len(lst) == 3 assert len(lst) == 3
def test_getitem(self): def test_getitem(self):
lst = HookedList(None, existing=[1, 2, 3]) lst = HookedList(existing=[1, 2, 3])
assert lst[0] == 1 assert lst[0] == 1
assert lst[1] == 2 assert lst[1] == 2
assert lst[-1] == 3 assert lst[-1] == 3
def test_contains(self): def test_contains(self):
lst = HookedList(None, existing=[1, 2, 3]) lst = HookedList(existing=[1, 2, 3])
assert 2 in lst assert 2 in lst
assert 4 not in lst assert 4 not in lst
def test_iter(self): def test_iter(self):
lst = HookedList(None, existing=[1, 2, 3]) lst = HookedList(existing=[1, 2, 3])
for i, item in enumerate(lst, start=1): for i, item in enumerate(lst, start=1):
assert item == i assert item == i
class TestMutableOps: class TestMutableOps:
def test_setitem(self): def test_setitem(self):
added = [] added = []
lst = HookedList(lambda e: added.append(e.item), existing=[1, 2, [4, 5, [6, 7]]]) lst = HookedList([1, 2, [4, 5, [6, 7]]], lambda e: added.append(e.item))
lst[0] = 10 lst[0] = 10
lst.append(20) lst.append(20)
assert added == [10, 20] assert added == [10, 20]
@@ -62,11 +62,11 @@ class TestHookedList:
assert added == [10, 20, 8] assert added == [10, 20, 8]
def test_delitem(self): def test_delitem(self):
lst = HookedList(None, existing=[1, 2, 3]) lst = HookedList(existing=[1, 2, 3])
del lst[1] del lst[1]
assert list(lst) == [1, 3] assert list(lst) == [1, 3]
def test_insert(self): def test_insert(self):
lst = HookedList(None, existing=[1, 3]) lst = HookedList(existing=[1, 3])
lst.insert(1, 2) lst.insert(1, 2)
assert list(lst) == [1, 2, 3] assert list(lst) == [1, 2, 3]