mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-06-26 22:00:13 +00:00
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import pytest
|
|
from haystack.testing.sample_components import FString
|
|
|
|
|
|
def test_fstring_with_one_var():
|
|
fstring = FString(template="Hello, {name}!", variables=["name"])
|
|
output = fstring.run(name="Alice")
|
|
assert output == {"string": "Hello, Alice!"}
|
|
|
|
|
|
def test_fstring_with_no_vars():
|
|
fstring = FString(template="No variables in this template.", variables=[])
|
|
output = fstring.run()
|
|
assert output == {"string": "No variables in this template."}
|
|
|
|
|
|
def test_fstring_with_template_at_runtime():
|
|
fstring = FString(template="Hello {name}", variables=["name"])
|
|
output = fstring.run(template="Goodbye {name}!", name="Alice")
|
|
assert output == {"string": "Goodbye Alice!"}
|
|
|
|
|
|
def test_fstring_with_vars_mismatch():
|
|
fstring = FString(template="Hello {name}", variables=["name"])
|
|
with pytest.raises(KeyError):
|
|
fstring.run(template="Goodbye {person}!", name="Alice")
|
|
|
|
|
|
def test_fstring_with_vars_in_excess():
|
|
fstring = FString(template="Hello {name}", variables=["name"])
|
|
output = fstring.run(template="Goodbye!", name="Alice")
|
|
assert output == {"string": "Goodbye!"}
|
|
|
|
|
|
def test_fstring_with_vars_missing():
|
|
fstring = FString(template="{greeting}, {name}!", variables=["name"])
|
|
with pytest.raises(KeyError):
|
|
fstring.run(greeting="Hello")
|