haystack/test/components/builders/test_prompt_builder.py

41 lines
1.1 KiB
Python
Raw Normal View History

import pytest
from haystack.preview.components.builders.prompt_builder import PromptBuilder
@pytest.mark.unit
def test_init():
builder = PromptBuilder(template="This is a {{ variable }}")
assert builder._template_string == "This is a {{ variable }}"
@pytest.mark.unit
def test_to_dict():
builder = PromptBuilder(template="This is a {{ variable }}")
res = builder.to_dict()
assert res == {
"type": "haystack.preview.components.builders.prompt_builder.PromptBuilder",
"init_parameters": {"template": "This is a {{ variable }}"},
}
@pytest.mark.unit
def test_run():
builder = PromptBuilder(template="This is a {{ variable }}")
res = builder.run(variable="test")
assert res == {"prompt": "This is a test"}
@pytest.mark.unit
def test_run_without_input():
builder = PromptBuilder(template="This is a template without input")
res = builder.run()
assert res == {"prompt": "This is a template without input"}
@pytest.mark.unit
def test_run_with_missing_input():
builder = PromptBuilder(template="This is a {{ variable }}")
res = builder.run()
assert res == {"prompt": "This is a "}