test orgs

tests/authoring/__init__.py

@@ -0,0 +1 @@
+"""Focused authoring tests."""

tests/authoring/helpers.py

@@ -0,0 +1,64 @@
+from __future__ import annotations
+
+from typing import Annotated, TypedDict
+
+from pydantic import BaseModel, Field
+
+from wf_authoring import node, state_field
+
+
+class WorkflowInput(BaseModel):
+    text: str
+
+
+class WorkflowState(BaseModel):
+    text: str
+    count: int
+    tags: list[str]
+
+
+class WorkflowOutput(BaseModel):
+    text: str
+
+
+class TypedDictInput(TypedDict):
+    text: str
+
+
+class AutoBindInput(BaseModel):
+    text: str
+    count: int
+
+
+class AutoBindOutput(BaseModel):
+    text: str
+    count: int
+
+
+class AutoBindState(BaseModel):
+    text: str
+    count: int
+
+
+class AppendState(BaseModel):
+    items: Annotated[list[str], state_field(merge_strategy="append")] = Field(
+        default_factory=list
+    )
+
+
+class DefaultedState(BaseModel):
+    items: list[str] = Field(default_factory=list)
+    metadata: dict[str, str] = Field(default_factory=dict)
+    explicit: int = 3
+
+
+@node(name="test.auto_bind")
+def auto_bind_node(input: AutoBindInput) -> AutoBindOutput:
+    """Return updated fields using automatically mapped state input."""
+    return AutoBindOutput(text=input.text.upper(), count=input.count + 1)
+
+
+@node(name="test.branch_router", outcomes=("left", "right"))
+def branch_router(input: AutoBindInput) -> AutoBindOutput:
+    """Route to left or right while preserving state shape."""
+    return AutoBindOutput(text=input.text, count=input.count)

tests/authoring/test_builder.py

@@ -0,0 +1,153 @@
+from __future__ import annotations
+
+import pytest
+
+from wf_authoring import WorkflowBuilder, state
+from wf_core import RunStatus, WorkflowExecutionError
+
+from tests.authoring.helpers import (
+    AutoBindInput,
+    AutoBindOutput,
+    AutoBindState,
+    auto_bind_node,
+)
+
+
+def test_builder_auto_binds_matching_node_inputs_and_outputs_to_state() -> None:
+    builder = WorkflowBuilder(
+        name="auto_bind_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+        start="update",
+    )
+    step = builder.use(auto_bind_node, id="update")
+    builder.connect(step, "ok", "__end__")
+
+    run = builder.execute(
+        {"text": "hello", "count": 1},
+    )
+
+    assert step.in_map == {
+        "state.text": "text",
+        "state.count": "count",
+    }
+    assert step.out_map == {
+        "text": "state.text",
+        "count": "state.count",
+    }
+    assert run.status == RunStatus.COMPLETED
+    assert run.state["text"] == "HELLO"
+    assert run.state["count"] == 2
+
+
+def test_builder_can_auto_id_node_uses_from_spec_name() -> None:
+    builder = WorkflowBuilder(
+        name="auto_id_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+        start="test_auto_bind",
+    )
+
+    first = builder.use(auto_bind_node)
+    second = builder.use(auto_bind_node)
+
+    assert first.id == "test_auto_bind"
+    assert second.id == "test_auto_bind_2"
+
+
+def test_builder_can_compile_with_explicit_start_set_later() -> None:
+    builder = WorkflowBuilder(
+        name="optional_start_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+    step = builder.use(auto_bind_node)
+    builder.set_entry_point(step)
+
+    workflow = builder.compile()
+
+    assert workflow.start == "test_auto_bind"
+
+
+def test_builder_requires_explicit_start_before_compile() -> None:
+    builder = WorkflowBuilder(
+        name="missing_start_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+
+    with pytest.raises(WorkflowExecutionError, match="start"):
+        builder.compile()
+
+
+def test_builder_registry_exports_used_node_specs() -> None:
+    builder = WorkflowBuilder(
+        name="registry_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+        start="test_auto_bind",
+    )
+    builder.use(auto_bind_node)
+
+    assert set(builder.registry()) == {"test.auto_bind"}
+
+
+def test_builder_execute_compiles_and_runs_with_used_registry() -> None:
+    builder = WorkflowBuilder(
+        name="execute_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+    step = builder.use(auto_bind_node)
+    builder.set_entry_point(step)
+    builder.connect(step, "ok", "__end__")
+
+    run = builder.execute({"text": "hello", "count": 1})
+
+    assert run.status == RunStatus.COMPLETED
+    assert run.state["text"] == "HELLO"
+    assert run.state["count"] == 2
+
+
+def test_builder_can_auto_id_condition_foreach_and_interrupt() -> None:
+    builder = WorkflowBuilder(
+        name="auto_id_control_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+
+    first_condition = builder.condition(check=state("count").gt(0))
+    second_condition = builder.condition(check=state("count").gt(1))
+    foreach = builder.foreach(over="state.tags", as_="tag")
+    interrupt = builder.interrupt(kind="approval")
+
+    assert first_condition.id == "condition"
+    assert second_condition.id == "condition_2"
+    assert foreach.id == "foreach_tag"
+    assert interrupt.id == "interrupt_approval"
+
+
+def test_builder_connect_can_use_node_specs_and_returns_resolved_refs() -> None:
+    builder = WorkflowBuilder(
+        name="connect_specs_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+
+    source, target = builder.connect(auto_bind_node, "ok", auto_bind_node)
+
+    assert not isinstance(source, str)
+    assert not isinstance(target, str)
+    assert source.id == "test_auto_bind"
+    assert target.id == "test_auto_bind_2"
+    assert builder.edges[0].from_ == "test_auto_bind"
+    assert builder.edges[0].outcome == "ok"
+    assert builder.edges[0].to == "test_auto_bind_2"

tests/authoring/test_builder_branch.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+import pytest
+
+from wf_authoring import WorkflowBuilder
+
+from tests.authoring.helpers import (
+    AutoBindInput,
+    AutoBindOutput,
+    AutoBindState,
+    auto_bind_node,
+    branch_router,
+)
+
+
+def test_builder_branch_connects_existing_steps() -> None:
+    builder = WorkflowBuilder(
+        name="branch_existing_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+    router = builder.use(branch_router)
+    left = builder.use(auto_bind_node, id="left")
+    right = builder.use(auto_bind_node, id="right")
+
+    builder.branch(router, {"left": left, "right": right})
+
+    assert [(edge.from_, edge.outcome, edge.to) for edge in builder.edges] == [
+        ("test_branch_router", "left", "left"),
+        ("test_branch_router", "right", "right"),
+    ]
+
+
+def test_builder_branch_can_use_node_specs_as_targets() -> None:
+    builder = WorkflowBuilder(
+        name="branch_specs_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+    router = builder.use(branch_router)
+
+    targets = builder.branch(router, {"left": auto_bind_node})
+
+    target = targets["left"]
+    assert not isinstance(target, str)
+    assert target.id == "test_auto_bind"
+    assert builder.edges[0].from_ == "test_branch_router"
+    assert builder.edges[0].outcome == "left"
+    assert builder.edges[0].to == "test_auto_bind"
+
+
+def test_builder_branch_warns_on_empty_branch_map() -> None:
+    builder = WorkflowBuilder(
+        name="branch_empty_demo",
+        input_schema=AutoBindInput,
+        state_schema=AutoBindState,
+        output_schema=AutoBindOutput,
+    )
+    router = builder.use(branch_router)
+
+    with pytest.warns(UserWarning, match="no branches"):
+        targets = builder.branch(router, {})
+
+    assert targets == {}

tests/authoring/test_schemas.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+from wf_authoring import WorkflowBuilder
+
+from tests.authoring.helpers import (
+    AppendState,
+    DefaultedState,
+    TypedDictInput,
+    WorkflowInput,
+    WorkflowOutput,
+    WorkflowState,
+)
+
+
+def test_builder_accepts_basemodel_classes_for_workflow_schemas() -> None:
+    builder = WorkflowBuilder(
+        name="model_schema_demo",
+        input_schema=WorkflowInput,
+        state_schema=WorkflowState,
+        output_schema=WorkflowOutput,
+        start="start",
+    )
+
+    workflow = builder.compile()
+
+    assert workflow.input_schema.properties["text"]["type"] == "string"
+    assert workflow.output_schema.properties["text"]["type"] == "string"
+    assert set(workflow.state_schema.fields) == {"text", "count", "tags"}
+    assert workflow.state_schema.fields["text"].type == "string"
+    assert workflow.state_schema.fields["count"].type == "integer"
+    assert workflow.state_schema.fields["tags"].type == "array"
+
+
+def test_builder_accepts_typeddict_for_json_schema_refs() -> None:
+    builder = WorkflowBuilder(
+        name="typed_dict_schema_demo",
+        input_schema=TypedDictInput,
+        state_schema=WorkflowState,
+        output_schema=WorkflowOutput,
+        start="start",
+    )
+
+    workflow = builder.compile()
+
+    assert workflow.input_schema.properties["text"]["type"] == "string"
+
+
+def test_state_basemodel_can_declare_merge_strategy_with_annotated_metadata() -> None:
+    builder = WorkflowBuilder(
+        name="state_metadata_demo",
+        input_schema=WorkflowInput,
+        state_schema=AppendState,
+        output_schema=WorkflowOutput,
+        start="start",
+    )
+
+    workflow = builder.compile()
+
+    assert workflow.state_schema.fields["items"].type == "array"
+    assert workflow.state_schema.fields["items"].merge_strategy == "append"
+
+
+def test_state_basemodel_seeds_safe_initial_defaults() -> None:
+    builder = WorkflowBuilder(
+        name="state_defaults_demo",
+        input_schema=WorkflowInput,
+        state_schema=DefaultedState,
+        output_schema=WorkflowOutput,
+        start="start",
+    )
+
+    workflow = builder.compile()
+
+    assert workflow.state_schema.fields["items"].default == []
+    assert workflow.state_schema.fields["metadata"].default == {}
+    assert workflow.state_schema.fields["explicit"].default == 3

tests/test_authoring_schemas.py

@@ -1,324 +0,0 @@
-from __future__ import annotations
-
-from typing import Annotated, TypedDict
-
-from pydantic import BaseModel, Field
-
-import pytest
-
-from wf_authoring import WorkflowBuilder, node, state, state_field
-from wf_core import RunStatus, WorkflowExecutionError
-
-
-class WorkflowInput(BaseModel):
-    text: str
-
-
-class WorkflowState(BaseModel):
-    text: str
-    count: int
-    tags: list[str]
-
-
-class WorkflowOutput(BaseModel):
-    text: str
-
-
-class TypedDictInput(TypedDict):
-    text: str
-
-
-class AutoBindInput(BaseModel):
-    text: str
-    count: int
-
-
-class AutoBindOutput(BaseModel):
-    text: str
-    count: int
-
-
-class AutoBindState(BaseModel):
-    text: str
-    count: int
-
-
-class AppendState(BaseModel):
-    items: Annotated[list[str], state_field(merge_strategy="append")] = Field(
-        default_factory=list
-    )
-
-
-class DefaultedState(BaseModel):
-    items: list[str] = Field(default_factory=list)
-    metadata: dict[str, str] = Field(default_factory=dict)
-    explicit: int = 3
-
-
-@node(name="test.auto_bind")
-def auto_bind_node(input: AutoBindInput) -> AutoBindOutput:
-    """Return updated fields using automatically mapped state input."""
-    return AutoBindOutput(text=input.text.upper(), count=input.count + 1)
-
-
-@node(name="test.branch_router", outcomes=("left", "right"))
-def branch_router(input: AutoBindInput) -> AutoBindOutput:
-    """Route to left or right while preserving state shape."""
-    return AutoBindOutput(text=input.text, count=input.count)
-
-
-def test_builder_accepts_basemodel_classes_for_workflow_schemas() -> None:
-    builder = WorkflowBuilder(
-        name="model_schema_demo",
-        input_schema=WorkflowInput,
-        state_schema=WorkflowState,
-        output_schema=WorkflowOutput,
-        start="start",
-    )
-
-    workflow = builder.compile()
-
-    assert workflow.input_schema.properties["text"]["type"] == "string"
-    assert workflow.output_schema.properties["text"]["type"] == "string"
-    assert set(workflow.state_schema.fields) == {"text", "count", "tags"}
-    assert workflow.state_schema.fields["text"].type == "string"
-    assert workflow.state_schema.fields["count"].type == "integer"
-    assert workflow.state_schema.fields["tags"].type == "array"
-
-
-def test_builder_accepts_typeddict_for_json_schema_refs() -> None:
-    builder = WorkflowBuilder(
-        name="typed_dict_schema_demo",
-        input_schema=TypedDictInput,
-        state_schema=WorkflowState,
-        output_schema=WorkflowOutput,
-        start="start",
-    )
-
-    workflow = builder.compile()
-
-    assert workflow.input_schema.properties["text"]["type"] == "string"
-
-
-def test_state_basemodel_can_declare_merge_strategy_with_annotated_metadata() -> None:
-    builder = WorkflowBuilder(
-        name="state_metadata_demo",
-        input_schema=WorkflowInput,
-        state_schema=AppendState,
-        output_schema=WorkflowOutput,
-        start="start",
-    )
-
-    workflow = builder.compile()
-
-    assert workflow.state_schema.fields["items"].type == "array"
-    assert workflow.state_schema.fields["items"].merge_strategy == "append"
-
-
-def test_state_basemodel_seeds_safe_initial_defaults() -> None:
-    builder = WorkflowBuilder(
-        name="state_defaults_demo",
-        input_schema=WorkflowInput,
-        state_schema=DefaultedState,
-        output_schema=WorkflowOutput,
-        start="start",
-    )
-
-    workflow = builder.compile()
-
-    assert workflow.state_schema.fields["items"].default == []
-    assert workflow.state_schema.fields["metadata"].default == {}
-    assert workflow.state_schema.fields["explicit"].default == 3
-
-
-def test_builder_auto_binds_matching_node_inputs_and_outputs_to_state() -> None:
-    builder = WorkflowBuilder(
-        name="auto_bind_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-        start="update",
-    )
-    step = builder.use(auto_bind_node, id="update")
-    builder.connect(step, "ok", "__end__")
-
-    run = builder.execute(
-        {"text": "hello", "count": 1},
-    )
-
-    assert step.in_map == {
-        "state.text": "text",
-        "state.count": "count",
-    }
-    assert step.out_map == {
-        "text": "state.text",
-        "count": "state.count",
-    }
-    assert run.status == RunStatus.COMPLETED
-    assert run.state["text"] == "HELLO"
-    assert run.state["count"] == 2
-
-
-def test_builder_can_auto_id_node_uses_from_spec_name() -> None:
-    builder = WorkflowBuilder(
-        name="auto_id_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-        start="test_auto_bind",
-    )
-
-    first = builder.use(auto_bind_node)
-    second = builder.use(auto_bind_node)
-
-    assert first.id == "test_auto_bind"
-    assert second.id == "test_auto_bind_2"
-
-
-def test_builder_can_compile_with_explicit_start_set_later() -> None:
-    builder = WorkflowBuilder(
-        name="optional_start_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-    step = builder.use(auto_bind_node)
-    builder.set_entry_point(step)
-
-    workflow = builder.compile()
-
-    assert workflow.start == "test_auto_bind"
-
-
-def test_builder_requires_explicit_start_before_compile() -> None:
-    builder = WorkflowBuilder(
-        name="missing_start_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-
-    with pytest.raises(WorkflowExecutionError, match="start"):
-        builder.compile()
-
-
-def test_builder_registry_exports_used_node_specs() -> None:
-    builder = WorkflowBuilder(
-        name="registry_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-        start="test_auto_bind",
-    )
-    builder.use(auto_bind_node)
-
-    assert set(builder.registry()) == {"test.auto_bind"}
-
-
-def test_builder_execute_compiles_and_runs_with_used_registry() -> None:
-    builder = WorkflowBuilder(
-        name="execute_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-    step = builder.use(auto_bind_node)
-    builder.set_entry_point(step)
-    builder.connect(step, "ok", "__end__")
-
-    run = builder.execute({"text": "hello", "count": 1})
-
-    assert run.status == RunStatus.COMPLETED
-    assert run.state["text"] == "HELLO"
-    assert run.state["count"] == 2
-
-
-def test_builder_branch_connects_existing_steps() -> None:
-    builder = WorkflowBuilder(
-        name="branch_existing_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-    router = builder.use(branch_router)
-    left = builder.use(auto_bind_node, id="left")
-    right = builder.use(auto_bind_node, id="right")
-
-    builder.branch(router, {"left": left, "right": right})
-
-    assert [(edge.from_, edge.outcome, edge.to) for edge in builder.edges] == [
-        ("test_branch_router", "left", "left"),
-        ("test_branch_router", "right", "right"),
-    ]
-
-
-def test_builder_branch_can_use_node_specs_as_targets() -> None:
-    builder = WorkflowBuilder(
-        name="branch_specs_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-    router = builder.use(branch_router)
-
-    targets = builder.branch(router, {"left": auto_bind_node})
-
-    target = targets["left"]
-    assert not isinstance(target, str)
-    assert target.id == "test_auto_bind"
-    assert builder.edges[0].from_ == "test_branch_router"
-    assert builder.edges[0].outcome == "left"
-    assert builder.edges[0].to == "test_auto_bind"
-
-
-def test_builder_branch_warns_on_empty_branch_map() -> None:
-    builder = WorkflowBuilder(
-        name="branch_empty_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-    router = builder.use(branch_router)
-
-    with pytest.warns(UserWarning, match="no branches"):
-        targets = builder.branch(router, {})
-
-    assert targets == {}
-
-
-def test_builder_can_auto_id_condition_foreach_and_interrupt() -> None:
-    builder = WorkflowBuilder(
-        name="auto_id_control_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-
-    first_condition = builder.condition(check=state("count").gt(0))
-    second_condition = builder.condition(check=state("count").gt(1))
-    foreach = builder.foreach(over="state.tags", as_="tag")
-    interrupt = builder.interrupt(kind="approval")
-
-    assert first_condition.id == "condition"
-    assert second_condition.id == "condition_2"
-    assert foreach.id == "foreach_tag"
-    assert interrupt.id == "interrupt_approval"
-
-
-def test_builder_connect_can_use_node_specs_and_returns_resolved_refs() -> None:
-    builder = WorkflowBuilder(
-        name="connect_specs_demo",
-        input_schema=AutoBindInput,
-        state_schema=AutoBindState,
-        output_schema=AutoBindOutput,
-    )
-
-    source, target = builder.connect(auto_bind_node, "ok", auto_bind_node)
-
-    assert not isinstance(source, str)
-    assert not isinstance(target, str)
-    assert source.id == "test_auto_bind"
-    assert target.id == "test_auto_bind_2"
-    assert builder.edges[0].from_ == "test_auto_bind"
-    assert builder.edges[0].outcome == "ok"
-    assert builder.edges[0].to == "test_auto_bind_2"