test_trim() — langchain Function Reference

Architecture documentation for the test_trim() function in test_graph.py from the langchain codebase.

Function python CoreAbstractions Serialization

Entity Profile

CoreAbstractions→ Serialization→ test_trim() — langchain Function Reference

Dependency Diagram

graph TD
  d7e7f7a7_2702_d481_22db_658c269abec1["test_trim()"]
  972d89eb_abd0_b940_67b4_eff75c775ace["test_graph.py"]
  d7e7f7a7_2702_d481_22db_658c269abec1 -->|defined in| 972d89eb_abd0_b940_67b4_eff75c775ace
  style d7e7f7a7_2702_d481_22db_658c269abec1 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_graph.py lines 54–77

def test_trim(snapshot: SnapshotAssertion) -> None:
    runnable = StrOutputParser()

    class Schema(BaseModel):
        a: int

    graph = Graph()
    start = graph.add_node(Schema, id="__start__")
    ask = graph.add_node(runnable, id="ask_question")
    answer = graph.add_node(runnable, id="answer_question")
    end = graph.add_node(Schema, id="__end__")
    graph.add_edge(start, ask)
    graph.add_edge(ask, answer)
    graph.add_edge(answer, ask, conditional=True)
    graph.add_edge(answer, end, conditional=True)

    assert _normalize_schema(graph.to_json()) == snapshot
    assert graph.first_node() is start
    assert graph.last_node() is end
    # can't trim start or end node
    graph.trim_first_node()
    assert graph.first_node() is start
    graph.trim_last_node()
    assert graph.last_node() is end