luciano/markitdown
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Test for PPTX conversion with OpenAI client

Browse source 
Signed-off-by: Hankyeol Kyung <kghnkl0103@gmail.com>
This commit is contained in:
Hankyeol Kyung 2024-12-27 16:07:33 +09:00
parent 9449d5b959
commit 06ccea5219
No known key found for this signature in database
GPG key ID: 0430C7F42578E222
2 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

BIN
tests/test_files/test.pptx vendored
View file

Binary file not shown.

16
tests/test_markitdown.py
View file

@ -300,6 +300,21 @@ def test_markitdown_llm() -> None:
assert test_string in result.text_content.lower() assert test_string in result.text_content.lower()
@pytest.mark.skipif(
skip_llm,
reason="do not run llm tests without a key",
)
def test_markitdown_pptx_llm() -> None:
client = openai.OpenAI()
markitdown = MarkItDown(llm_client=client, llm_model="gpt-4o-mini")
result = markitdown.convert(os.path.join(TEST_FILES_DIR, "test.pptx"))
# like test_markitdown_llm, this should be improved
for test_string in ["red", "blue"]:
assert test_string in result.text_content.lower()
if __name__ == "__main__": if __name__ == "__main__":
"""Runs this file's tests from the command line.""" """Runs this file's tests from the command line."""
test_markitdown_remote() test_markitdown_remote()
@ -307,3 +322,4 @@ if __name__ == "__main__":
test_markitdown_exiftool() test_markitdown_exiftool()
test_markitdown_deprecation() test_markitdown_deprecation()
test_markitdown_llm() test_markitdown_llm()
test_markitdown_pptx_llm()