github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-19 17:45:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply lint using uv

Browse Source
This commit is contained in:
Sadra Barikbin 2025-07-27 02:44:07 +03:30
parent 4c256fb5b4
commit 5a1440b9f1
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
libs/core/tests/unit_tests/prompts/test_prompt.py
View File

@ -1,7 +1,7 @@
"""Test functionality related to prompts.""" """Test functionality related to prompts."""
import re import re
from typing import Any, Union from typing import Any, Literal, Union
from unittest import mock from unittest import mock
import pytest import pytest
@ -686,8 +686,8 @@ def test_prompt_with_template_variable_name_jinja2() -> None:
def test_prompt_template_add_with_with_another_format() -> None: def test_prompt_template_add_with_with_another_format() -> None:
with pytest.raises(ValueError, match=r"Cannot add templates"): with pytest.raises(ValueError, match=r"Cannot add templates"):
( (
PromptTemplate.from_template("This is a {template}") + PromptTemplate.from_template("This is a {template}")
PromptTemplate.from_template("So {{this}} is", template_format="mustache") + PromptTemplate.from_template("So {{this}} is", template_format="mustache")
) )
@ -699,12 +699,16 @@ def test_prompt_template_add_with_with_another_format() -> None:
"jinja2", "jinja2",
"This is a {{variable}}", "This is a {{variable}}",
". This is {{another_variable}}", ". This is {{another_variable}}",
marks=[pytest.mark.requires("jinja2")] marks=[pytest.mark.requires("jinja2")],
), ),
("mustache", "This is a {{variable}}", ". This is {{another_variable}}"), ("mustache", "This is a {{variable}}", ". This is {{another_variable}}"),
] ],
) )
def test_prompt_template_add(template_format: str, prompt1: str, prompt2: str) -> None: def test_prompt_template_add(
template_format: Literal["f-string", "mustache", "jinja2"],
prompt1: str,
prompt2: str,
) -> None:
first_prompt = PromptTemplate.from_template( first_prompt = PromptTemplate.from_template(
prompt1, prompt1,
template_format=template_format, template_format=template_format,
@ -721,13 +725,10 @@ def test_prompt_template_add(template_format: str, prompt1: str, prompt2: str) -
) )
assert concated_prompt.input_variables == prompt_of_concated.input_variables assert concated_prompt.input_variables == prompt_of_concated.input_variables
assert ( assert concated_prompt.format(
concated_prompt.format( variable="template",
variable="template", another_variable="other_template",
another_variable="other_template", ) == prompt_of_concated.format(
) == variable="template",
prompt_of_concated.format( another_variable="other_template",
variable="template",
another_variable="other_template",
)
) )