mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-06-26 22:00:13 +00:00
65 lines
2.6 KiB
Python
65 lines
2.6 KiB
Python
|
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
import logging
|
||
|
|
|
||
|
|
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"] == str(files[0])
|
||
|
|
assert docs[1].meta == bytestream.meta
|
||
|
|
|
||
|
|
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": str(test_files_path / "pptx" / "sample_pptx.pptx"),
|
||
|
|
"key": "value",
|
||
|
|
"language": "it",
|
||
|
|
}
|