github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-19 01:21:50 +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."""
import re
from typing import Any, Union
from typing import Any, Literal, Union
from unittest import mock
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:
with pytest.raises(ValueError, match=r"Cannot add templates"):
(
PromptTemplate.from_template("This is a {template}") +
PromptTemplate.from_template("So {{this}} is", template_format="mustache")
PromptTemplate.from_template("This is a {template}")
+ PromptTemplate.from_template("So {{this}} is", template_format="mustache")
)
@ -699,12 +699,16 @@ def test_prompt_template_add_with_with_another_format() -> None:
"jinja2",
"This is a {{variable}}",
". This is {{another_variable}}",
marks=[pytest.mark.requires("jinja2")]
marks=[pytest.mark.requires("jinja2")],
),
("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(
prompt1,
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.format(
variable="template",
another_variable="other_template",
) ==
prompt_of_concated.format(
variable="template",
another_variable="other_template",
)
assert concated_prompt.format(
variable="template",
another_variable="other_template",
) == prompt_of_concated.format(
variable="template",
another_variable="other_template",
)