danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use patch.object to satisfy type checker

Browse source
This commit is contained in:
Chris McDonough 2026-06-22 14:09:01 -04:00
parent 99a200c9d2
commit 21a8f52893
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_chunker.py
View file

@ -59,6 +59,7 @@ async def test_local_chunker(qa_corpus: list[dict[str, str]]):
async def test_local_chunker_runs_off_event_loop_thread(): async def test_local_chunker_runs_off_event_loop_thread():
"""Chunking is CPU-bound; verify it runs in a worker thread.""" """Chunking is CPU-bound; verify it runs in a worker thread."""
import threading import threading
from unittest.mock import patch
chunker = DoclingLocalChunker() chunker = DoclingLocalChunker()
event_loop_thread = threading.current_thread() event_loop_thread = threading.current_thread()
@ -66,15 +67,15 @@ async def test_local_chunker_runs_off_event_loop_thread():
original = chunker._chunk_sync original = chunker._chunk_sync
def recording_chunk_sync(document): def recording_chunk_sync(self, document):
called_from.append(threading.current_thread()) called_from.append(threading.current_thread())
return original(document) return original(document)
chunker._chunk_sync = recording_chunk_sync
converter = get_converter(Config) converter = get_converter(Config)
doc = await converter.convert_text("# Hello\n\nWorld", name="test.md") doc = await converter.convert_text("# Hello\n\nWorld", name="test.md")
await chunker.chunk(doc)
with patch.object(DoclingLocalChunker, "_chunk_sync", recording_chunk_sync):
await chunker.chunk(doc)
assert called_from, "_chunk_sync was never called" assert called_from, "_chunk_sync was never called"
assert called_from[0] is not event_loop_thread, ( assert called_from[0] is not event_loop_thread, (