haystack/test/dataclasses/test_streaming_chunk.py
Massimiliano Pippi 8adb8bbab8
Remove preview folder in test/
---------

Co-authored-by: Silvano Cerza <silvanocerza@gmail.com>
2023-11-24 11:52:55 +01:00

33 lines
834 B
Python

import pytest
from haystack.preview.dataclasses import StreamingChunk
@pytest.mark.unit
def test_create_chunk_with_content_and_metadata():
chunk = StreamingChunk(content="Test content", metadata={"key": "value"})
assert chunk.content == "Test content"
assert chunk.metadata == {"key": "value"}
@pytest.mark.unit
def test_create_chunk_with_only_content():
chunk = StreamingChunk(content="Test content")
assert chunk.content == "Test content"
assert chunk.metadata == {}
@pytest.mark.unit
def test_access_content():
chunk = StreamingChunk(content="Test content", metadata={"key": "value"})
assert chunk.content == "Test content"
@pytest.mark.unit
def test_create_chunk_with_empty_content():
chunk = StreamingChunk(content="")
assert chunk.content == ""
assert chunk.metadata == {}