@@ -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