langchain Architecture Docs
Home / File/ test_enum_parser.py — langchain Source File

test_enum_parser.py — langchain Source File

Architecture documentation for test_enum_parser.py, a python file in the langchain codebase. 4 imports, 0 dependents.

File python CoreAbstractions MessageSchema 4 imports 2 functions 1 classes

Entity Profile

Dependency Diagram

graph LR
  9b6ecfda_f551_6cba_c335_c423ce4fefef["test_enum_parser.py"]
  b188e880_71c6_b93e_127d_c22666293d37["enum"]
  9b6ecfda_f551_6cba_c335_c423ce4fefef --> b188e880_71c6_b93e_127d_c22666293d37
  120e2591_3e15_b895_72b6_cb26195e40a6["pytest"]
  9b6ecfda_f551_6cba_c335_c423ce4fefef --> 120e2591_3e15_b895_72b6_cb26195e40a6
  75137834_4ba7_dc43_7ec5_182c05eceedf["langchain_core.exceptions"]
  9b6ecfda_f551_6cba_c335_c423ce4fefef --> 75137834_4ba7_dc43_7ec5_182c05eceedf
  9c18407b_b338_17e8_c321_b67251a632df["langchain_classic.output_parsers.enum"]
  9b6ecfda_f551_6cba_c335_c423ce4fefef --> 9c18407b_b338_17e8_c321_b67251a632df
  style 9b6ecfda_f551_6cba_c335_c423ce4fefef fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

from enum import Enum

import pytest
from langchain_core.exceptions import OutputParserException

from langchain_classic.output_parsers.enum import EnumOutputParser


class Colors(Enum):
    RED = "red"
    GREEN = "green"
    BLUE = "blue"


def test_enum_output_parser_parse() -> None:
    parser = EnumOutputParser(enum=Colors)

    # Test valid inputs
    result = parser.parse("red")
    assert result == Colors.RED

    result = parser.parse("green")
    assert result == Colors.GREEN

    result = parser.parse("blue")
    assert result == Colors.BLUE

    # Test invalid input
    with pytest.raises(OutputParserException):
        parser.parse("INVALID")


def test_enum_output_parser_output_type() -> None:
    """Test the output type of the enum output parser is the expected enum."""
    assert EnumOutputParser(enum=Colors).OutputType is Colors

Domain

Subdomains

Functions

Classes

Dependencies

  • enum
  • langchain_classic.output_parsers.enum
  • langchain_core.exceptions
  • pytest

Source

Frequently Asked Questions

What does test_enum_parser.py do?
test_enum_parser.py is a source file in the langchain codebase, written in python. It belongs to the CoreAbstractions domain, MessageSchema subdomain.
What functions are defined in test_enum_parser.py?
test_enum_parser.py defines 2 function(s): test_enum_output_parser_output_type, test_enum_output_parser_parse.
What does test_enum_parser.py depend on?
test_enum_parser.py imports 4 module(s): enum, langchain_classic.output_parsers.enum, langchain_core.exceptions, pytest.
Where is test_enum_parser.py in the architecture?
test_enum_parser.py is located at libs/langchain/tests/unit_tests/output_parsers/test_enum_parser.py (domain: CoreAbstractions, subdomain: MessageSchema, directory: libs/langchain/tests/unit_tests/output_parsers).

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free