tests_06/test_flow_graph.py

from engine_03.flow.graph_models import table_to_flowgraph
from engine_03.flow.mermaid_renderer import build_mermaid
from engine_03.flow.simulator import simulate_flowgraph


def test_basic_flow():
    nodes = [
        ["a", "Start", "event"],
        ["b", "Middle", "function"],
        ["c", "End", "output"],
    ]
    edges = [
        ["a", "b", ""],
        ["b", "c", ""],
    ]

    flow = table_to_flowgraph(nodes, edges)
    mermaid = build_mermaid(flow)
    sim = simulate_flowgraph(flow)

    assert "a[Start]" in mermaid
    assert "b[Middle]" in mermaid
    assert "c[End]" in mermaid

    assert "Node `a`" in sim
    assert "Node `b`" in sim
    assert "Node `c`" in sim