Massimiliano Pippi 8adb8bbab8
Remove preview folder in test/
---------

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

38 lines
1.1 KiB
Python

import pytest
from haystack.preview.components.generators.utils import default_streaming_callback
from haystack.preview.components.generators.utils import serialize_callback_handler, deserialize_callback_handler
# streaming callback needs to be on module level
def streaming_callback(chunk):
pass
@pytest.mark.unit
def test_callback_handler_serialization():
result = serialize_callback_handler(streaming_callback)
assert result == "test_utils.streaming_callback"
@pytest.mark.unit
def test_callback_handler_serialization_non_local():
result = serialize_callback_handler(default_streaming_callback)
assert result == "haystack.preview.components.generators.utils.default_streaming_callback"
@pytest.mark.unit
def test_callback_handler_deserialization():
result = serialize_callback_handler(streaming_callback)
fn = deserialize_callback_handler(result)
assert fn is streaming_callback
@pytest.mark.unit
def test_callback_handler_deserialization_non_local():
result = serialize_callback_handler(default_streaming_callback)
fn = deserialize_callback_handler(result)
assert fn is default_streaming_callback