abb gc test

tests/test_async_background_batcher.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+import asyncio as aio
+import gc
+import weakref
+from typing import Iterable, Tuple
+
+from aiuti.typing import AYields
+from aiuti.asyncio import AsyncBackgroundBatcher
+
+
+async def batch_things() -> weakref.ref[AsyncBackgroundBatcher[int, int]]:
+    @AsyncBackgroundBatcher
+    async def batcher(
+        batch: Iterable[Tuple[str, int]],
+    ) -> AYields[Tuple[str, int]]:
+        for key, value in batch:
+            await aio.sleep(0)
+            yield key, value + 1
+
+    await aio.gather(*map(batcher, range(30)))
+
+    return weakref.ref(batcher)
+
+
+def test_gc() -> None:
+    ref = aio.run(batch_things())
+    gc.collect()
+    assert ref() is None