forked from Archives/langchain
d15f481352
This pull request adds unit tests for various output parsers (BooleanOutputParser, CommaSeparatedListOutputParser, and StructuredOutputParser) to ensure their correct functionality and to increase code reliability and maintainability. The tests cover both valid and invalid input cases. Changes: Added unit tests for BooleanOutputParser. Added unit tests for CommaSeparatedListOutputParser. Added unit tests for StructuredOutputParser. Testing: All new unit tests have been executed, and they pass successfully. The overall test suite has been run, and all tests pass. Notes: These tests cover both successful parsing scenarios and error handling for invalid inputs. If any new output parsers are added in the future, corresponding unit tests should also be created to maintain coverage.
26 lines
931 B
Python
26 lines
931 B
Python
from langchain.output_parsers import ResponseSchema, StructuredOutputParser
|
|
from langchain.schema import OutputParserException
|
|
|
|
|
|
def test_parse() -> None:
|
|
response_schemas = [
|
|
ResponseSchema(name="name", description="desc"),
|
|
ResponseSchema(name="age", description="desc"),
|
|
]
|
|
parser = StructuredOutputParser.from_response_schemas(response_schemas)
|
|
|
|
# Test valid JSON input
|
|
text = '```json\n{"name": "John", "age": 30}\n```'
|
|
expected_result = {"name": "John", "age": 30}
|
|
result = parser.parse(text)
|
|
assert result == expected_result, f"Expected {expected_result}, but got {result}"
|
|
|
|
# Test invalid JSON input
|
|
text = '```json\n{"name": "John"}\n```'
|
|
try:
|
|
parser.parse(text)
|
|
except OutputParserException:
|
|
pass # Test passes if OutputParserException is raised
|
|
else:
|
|
assert False, f"Expected OutputParserException, but got {parser.parse(text)}"
|