mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-06-26 22:00:13 +00:00
81 lines
3.3 KiB
Python
81 lines
3.3 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import logging
|
|
import os
|
|
|
|
from haystack.dataclasses import ByteStream
|
|
from haystack.components.converters.pptx import PPTXToDocument
|
|
|
|
|
|
class TestPPTXToDocument:
|
|
def test_run(self, test_files_path):
|
|
"""
|
|
Test if the component runs correctly.
|
|
"""
|
|
bytestream = ByteStream.from_file_path(test_files_path / "pptx" / "sample_pptx.pptx")
|
|
bytestream.meta["file_path"] = str(test_files_path / "pptx" / "sample_pptx.pptx")
|
|
bytestream.meta["key"] = "value"
|
|
files = [str(test_files_path / "pptx" / "sample_pptx.pptx"), bytestream]
|
|
converter = PPTXToDocument()
|
|
output = converter.run(sources=files)
|
|
docs = output["documents"]
|
|
|
|
assert len(docs) == 2
|
|
assert (
|
|
"Sample Title Slide\nJane Doe\fTitle of First Slide\nThis is a bullet point\nThis is another bullet point"
|
|
in docs[0].content
|
|
)
|
|
assert (
|
|
"Sample Title Slide\nJane Doe\fTitle of First Slide\nThis is a bullet point\nThis is another bullet point"
|
|
in docs[0].content
|
|
)
|
|
assert docs[0].meta["file_path"] == os.path.basename(files[0])
|
|
assert docs[1].meta == {"file_path": os.path.basename(bytestream.meta["file_path"]), "key": "value"}
|
|
|
|
def test_run_error_non_existent_file(self, caplog):
|
|
sources = ["non_existing_file.pptx"]
|
|
converter = PPTXToDocument()
|
|
with caplog.at_level(logging.WARNING):
|
|
results = converter.run(sources=sources)
|
|
assert "Could not read non_existing_file.pptx" in caplog.text
|
|
assert results["documents"] == []
|
|
|
|
def test_run_error_wrong_file_type(self, caplog, test_files_path):
|
|
sources = [str(test_files_path / "txt" / "doc_1.txt")]
|
|
converter = PPTXToDocument()
|
|
with caplog.at_level(logging.WARNING):
|
|
results = converter.run(sources=sources)
|
|
assert "doc_1.txt and convert it" in caplog.text
|
|
assert results["documents"] == []
|
|
|
|
def test_run_with_meta(self, test_files_path):
|
|
bytestream = ByteStream.from_file_path(test_files_path / "pptx" / "sample_pptx.pptx")
|
|
bytestream.meta["file_path"] = str(test_files_path / "pptx" / "sample_pptx.pptx")
|
|
bytestream.meta["key"] = "value"
|
|
|
|
converter = PPTXToDocument()
|
|
output = converter.run(sources=[bytestream], meta=[{"language": "it"}])
|
|
document = output["documents"][0]
|
|
|
|
assert document.meta == {
|
|
"file_path": os.path.basename(test_files_path / "pptx" / "sample_pptx.pptx"),
|
|
"key": "value",
|
|
"language": "it",
|
|
}
|
|
|
|
def test_run_with_store_full_path_false(self, test_files_path):
|
|
"""
|
|
Test if the component runs correctly with store_full_path=False
|
|
"""
|
|
bytestream = ByteStream.from_file_path(test_files_path / "pptx" / "sample_pptx.pptx")
|
|
bytestream.meta["file_path"] = str(test_files_path / "pptx" / "sample_pptx.pptx")
|
|
bytestream.meta["key"] = "value"
|
|
|
|
converter = PPTXToDocument(store_full_path=False)
|
|
output = converter.run(sources=[bytestream], meta=[{"language": "it"}])
|
|
document = output["documents"][0]
|
|
|
|
assert document.meta == {"file_path": "sample_pptx.pptx", "key": "value", "language": "it"}
|