Skip to content

Commit 52ff3e3

Browse files
committed
fix
1 parent 639ac89 commit 52ff3e3

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

integrations/docling/tests/test_converter.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -26,14 +26,14 @@ def test_run_doc_chunks_minimal() -> None:
2626
meta_extractor_mock = MagicMock()
2727

2828
converter_mock.convert.side_effect = [
29-
SimpleNamespace(document="dl-doc-for-file-a.pdf"),
30-
SimpleNamespace(document="dl-doc-for-file-b.pdf"),
29+
SimpleNamespace(document=SimpleNamespace(texts=[], name="dl-doc-for-file-a.pdf")),
30+
SimpleNamespace(document=SimpleNamespace(texts=[], name="dl-doc-for-file-b.pdf")),
3131
]
3232

3333
def chunk_side_effect(dl_doc: Any) -> list[SimpleNamespace]:
3434
return [
35-
SimpleNamespace(text=f"chunk-1-of-{dl_doc}"),
36-
SimpleNamespace(text=f"chunk-2-of-{dl_doc}"),
35+
SimpleNamespace(text=f"chunk-1-of-{dl_doc.name}"),
36+
SimpleNamespace(text=f"chunk-2-of-{dl_doc.name}"),
3737
]
3838

3939
chunker_mock.chunk.side_effect = chunk_side_effect
@@ -231,7 +231,7 @@ def test_run_with_sources_parameter() -> None:
231231
chunker_mock = MagicMock()
232232
meta_extractor_mock = MagicMock()
233233

234-
converter_mock.convert.return_value = SimpleNamespace(document="dl-doc")
234+
converter_mock.convert.return_value = SimpleNamespace(document=SimpleNamespace(texts=[]))
235235
chunker_mock.chunk.return_value = [SimpleNamespace(text="chunk-1")]
236236
chunker_mock.contextualize.return_value = "contextualized-chunk-1"
237237
meta_extractor_mock.extract_chunk_meta.return_value = {}
@@ -252,7 +252,7 @@ def test_run_paths_deprecated() -> None:
252252
chunker_mock = MagicMock()
253253
meta_extractor_mock = MagicMock()
254254

255-
converter_mock.convert.return_value = SimpleNamespace(document="dl-doc")
255+
converter_mock.convert.return_value = SimpleNamespace(document=SimpleNamespace(texts=[]))
256256
chunker_mock.chunk.return_value = [SimpleNamespace(text="chunk-1")]
257257
chunker_mock.contextualize.return_value = "contextualized-chunk-1"
258258
meta_extractor_mock.extract_chunk_meta.return_value = {}
@@ -278,10 +278,10 @@ def test_run_meta_single_dict_doc_chunks() -> None:
278278
meta_extractor_mock = MagicMock()
279279

280280
converter_mock.convert.side_effect = [
281-
SimpleNamespace(document="dl-doc-a"),
282-
SimpleNamespace(document="dl-doc-b"),
281+
SimpleNamespace(document=SimpleNamespace(texts=[], name="dl-doc-a")),
282+
SimpleNamespace(document=SimpleNamespace(texts=[], name="dl-doc-b")),
283283
]
284-
chunker_mock.chunk.side_effect = lambda dl_doc: [SimpleNamespace(text=f"chunk-of-{dl_doc}")]
284+
chunker_mock.chunk.side_effect = lambda dl_doc: [SimpleNamespace(text=f"chunk-of-{dl_doc.name}")]
285285
chunker_mock.contextualize.side_effect = lambda chunk: chunk.text
286286
meta_extractor_mock.extract_chunk_meta.return_value = {"extractor_key": "extractor_val"}
287287

0 commit comments

Comments
 (0)