You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
dr-doc-search/tests/workflow_steps_test.py

27 lines
555 B
Python

from doc_search.workflow import (
AskQuestion,
CombineAllText,
ConvertImagesToText,
ConvertPDFToImages,
CreateIndex,
ImageMagickCommand,
LoadIndex,
VerifyInputFile,
workflow_steps,
)
def test_return_expected_workflow() -> None:
expected_workflow_steps = workflow_steps()
assert expected_workflow_steps == [
VerifyInputFile,
ImageMagickCommand,
ConvertPDFToImages,
ConvertImagesToText,
CombineAllText,
CreateIndex,
LoadIndex,
AskQuestion,
]