Skip to content

Commit ba82a82

Browse files
[DEVX-453]: Modified tests to check the sequence after multiprocessing
1 parent 003ee05 commit ba82a82

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

tests/pipelines/test_pdf_pipelines.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,8 @@ def test_pipeline_run(self,):
3232
])
3333
elements = pipeline.run(files=PDF_FILE_PATH)
3434
assert len(elements) == 3
35-
assert elements[0].text[:9] == 'MAIN GAME'
35+
assert (elements[0].text[:9] == 'MAIN GAME') and (
36+
elements[1].text[1:12] == 'Transcribed') and (elements[2].text[:10] == 'Thankfully')
3637
assert elements[0].metadata['filename'] == 'DA-1p.pdf'
3738
assert elements[0].metadata['page_number'] == 1
3839
assert elements[0].metadata['text_after'] == 'our assault."'

tests/pipelines/test_text_pipelines.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@ def test_pipeline_run(self,):
3636
])
3737
elements = pipeline.run(files=TEXT_FILE_PATH)
3838
assert len(elements) == 4
39-
assert elements[0].text[:9] == 'CHAPTER I'
39+
assert (elements[0].text[:9] == 'CHAPTER I') and (elements[1].text[:3] == 'All') and (
40+
elements[2].text[1:6] == 'First') and (elements[3].text[1:5] == 'What')
4041
assert elements[0].metadata['filename'] == 'book-war-and-peace-1p.txt'
4142
assert elements[0].metadata['text_after'] == 'used only by the elite.'
4243

0 commit comments

Comments
 (0)