"""Test formatting functionality.""" import pytest from langchain_core.utils import formatter def test_valid_formatting() -> None: """Test formatting works as expected.""" template = "This is a {foo} test." output = formatter.format(template, foo="good") expected_output = "This is a good test." assert output == expected_output def test_does_not_allow_args() -> None: """Test formatting raises error when args are provided.""" template = "This is a {} test." with pytest.raises( ValueError, match="No arguments should be provided, " "everything should be passed as keyword arguments", ): formatter.format(template, "good") def test_allows_extra_kwargs() -> None: """Test formatting allows extra keyword arguments.""" template = "This is a {foo} test." output = formatter.format(template, foo="good", bar="oops") expected_output = "This is a good test." assert output == expected_output